Python 的装饰器

记下学习的过程中的一些理解。

装饰器（Decorators）是这样一种函数：接收一个函数作为参数，并返回一个函数。函数通常被用来操作变量，而装饰器是用来改变函数的函数。

注意 foo() 表示调用一个函数，而 foo 才表示该函数。

定义一个装饰器

任何装饰器的定义将包含以下结构，简单装饰器 simple_decorator 接收一个待装饰函数 function_to_decorate 作为参数，将返回已装饰函数 decorated_function，而 decorated_function 将在 simple_decorator 内部定义。

def simple_decorator(function_to_decorate):
    def decorated_function():
        """something"""
    return decorated_function

"""something""" 中可以调用 function_to_decorate，例如：

def simple_decorator(function_to_decorate):
    def decorated_function():
        print('I was called.')
        function_to_decorate()
    return decorated_function

这样一个装饰器将返回一个 function_to_decorate 几乎相同的 decorated_function，唯一不同是 decorated_function 顶部多了一行 print('I was called.')。

在定义函数时使用装饰器

def foo():
    """something"""
foo = simple_decorator(foo)

定义完了一个函数 foo 以后，我们以 foo 作为参数调用 simple_decorator 返回一个 decorated_function，再用返回的 decorated_function 赋值给 foo。最终结果是，我们用装饰器改变了 foo，就好像在 i = foo(i) 里我们将改变变量 i。

为了使这种丑陋的写法更加 Pythonic，Python 允许在定义 foo 的上一行用 @simple_decorator 来等价地在最后调用 foo = simple_decorator(foo)。因此上面也可以写成这样：

@simple_decorator
def foo():
    """something"""

第二种装饰器

定义

def decorator(how_to_decorate):
    def real_decorator(function_to_decorate):
        def decorated_function():
            """something"""
        return decorated_function
    return real_decorator

装饰器 decorator 并不接收一个函数作为参数，相反它接收普通参数 how_to_decorate，并返回一个真装饰器 real_decorator，观察 real_decorator 会发现与上面的 simple_decorator 如出一辙。

使用

当我们调用 decorator 的时候，方法也会略有不同：

def foo():
    """something"""
foo = decorator('something')(foo)

我们先以 'something' 作为参数调用 decorator 将返回 real_decorator，所以

• decorator('something') 等价于 real_decorator
• decorator('something')(foo) 等价于 real_decorator(foo)

简化的写法是：

@decorator('something')
def foo():
    """something"""