python - Python 装饰器理论

Question

我对 Python 中的干净理论有疑问。什么时候：

@decorator_func
def func(bla, alba):
    pass

相当于：

def func(bla, alba):
    pass
func = decorator_func(func)

所以：

@decorator_func(aaa, bar)
def func(bla, alba):
    pass

相当于...？

Answer 1

它相当于：

def func(bla, alba):
    pass
func = decorator_func(aaa, bar)(func)

或者：

def func(bla, alba):
    pass
decorator = decorator_func(aaa, bar)
func = decorator(func)

因此，在您的第二个示例中，decorator_func应该是一个返回可调用对象的可调用对象。

这是这种结构的一个例子：

class prepend_two_arguments:
    def __init__(self, a, b):
        self.a = a
        self.b = b

    def __call__(self, f):
        def wrapped_function(*args, **kwargs):
            return f(self.a, self.b, *args, **kwargs)
        return wrapped_function

@prepend_two_arguments(1,2)
def f(a, b, c):
    return a+b+c
print(f(3)) # 6

还有一个，只使用函数：

def add_to_result(x):
    def decorator(fn):
        def wrapped_function(*args, **kwargs):
            return fn(*args, **kwargs)+x
        return wrapped_function
    return decorator

@add_to_result(3)
def my_func(a, b):
    return a+b
print(my_func(1,2)) # 6

Answer 2

下面是一个使用闭包的装饰器函数示例：

def print_string_before(string):
    def decorator_fn(fn):
        def wrapped_fn(*args, **kwargs):
            print string
            return fn(*args, **kwargs)
        return wrapped_fn
    return decorator_fn

请注意，装饰器同样可以返回被装饰的函数（或类），并以某种方式对其进行修改（例如设置属性）。