Python中的参数化装饰器

Question

我一直在尝试使用 Python 中的不同类型的装饰器，但我发现很难理解参数化装饰器。 装饰器接受可调用对象并返回可调用对象（在我的情况下可调用函数是一个函数） 在以下代码中：-

def check_non_negative(index):
    def validator(f):
        def wrap(*args):
            if args[index]<0:
                raise ValueError(
                    'Argument {} must be non negative.'.format(index))
            return f(*args)
        return wrap
    return validator

@check_non_negative(1)
def create_list(value,size):
    return [value]*size


create_list('a',3)

在这里，我看到 check_non_negative 不是根据定义的装饰器，而是表现得像一个（在运行期间验证器是实际的装饰器）。 check_non_negative 接受一个整数而不是可调用的，但它的行为类似于装饰器。谁能解释一下原因？

Answer 1

check_non_negative 按照您的定义从技术上讲并不是装饰器，人们可能会称其为“装饰器工厂”。它返回validator，是一个装饰器。

基本上只要你有：

@<expression>
def ...

然后<expression> 必须评估为装饰器，即接受单个参数的可调用对象，该参数也是可调用对象。

Answer 2

当您将参数传递给装饰器函数（即@check_non_negative(1)）时，会使用这些参数调用该函数，然后它会返回一个接受并返回一个函数的装饰器（就像您使用 @validator 时没有参数）。

使用类型注释 IMO 更容易理解：

import functools
from typing import cast, Callable, List, TypeVar


_Elem = TypeVar('_Elem')
_Func = TypeVar('_Func', bound=Callable)


def check_non_negative(index: int) -> Callable[[_Func], _Func]:
    def validator(f: _Func) -> _Func:
        @functools.wraps(f)
        def wrap(*args, **kwargs):
            if args[index] < 0:
                raise ValueError(
                    'Argument {} must be non negative.'.format(index))
            return f(*args, **kwargs)
        return cast(_Func, wrap)
    return validator


@check_non_negative(1)
def create_list(value: _Elem, size: int) -> List[_Elem]:
    return [value]*size

所以check_non_negative 接受int 参数，并返回一个函数(validator)，该函数接受特定类型的函数(_Func，这里指的是修饰函数的类型，例如create_list ) 并返回相同类型的函数。