Python 的 装饰器 (decorator) 是一个可以让代码达到精简又漂亮的写法，用起来不但轻松简单，又可以提升代码的可读性，这篇文章将会介绍 Python 的装饰器。

什么是装饰器 decorator？

装饰器 decorator 是 Python 的一种程序设计模式，装饰器本质上是一个 Python 函数或类 (class)，它可以让其他函数或类，在不需要做任何代码修改的前提下增加额外功能，装饰器的返回值也是一个函数或类对象，有了装饰器，就可以抽离与函数功能本身无关的代码，放到装饰器中并继续重复使用。

在 Python 中，使用「@」作为装饰器使用的语法糖符号（语法糖指的是将复杂的代码包装起来的糖衣，也就是简化写法）。

创建一个装饰器

下方的代码，定义了一个装饰器函数 a 和一个被装饰的函数 b，当 b 函数执行后，会看到 a 运算后的结果，套用在 b 函数上。

简单来说，当某个函数加上装饰器后，执行该函数之前会先执行「装饰」的内容，就如同要离家出门，必须先装饰身体（化妆、衣服、裤子、鞋子... 等），完成后就执行出门的动作。

def a(func):
    print('makeup...')
    return func

def b():
    print('go!!!!')

b = a(b)
b()

# makeup...
# go!!!!

在 Python 里，函数 function 可以当成参数传递并执行，所以上面的代码将 b 传入 a 作为参数，所以执行 b 时效果等同于执行 a 里的 func，执行流程如下所示。

接着使用语法糖「@」包装，就能达到简化的效果。

def a(func):
    print('makeup...')
    return func

装饰器写在 b 的前面，表示装饰 b

@a
def b():
    print('go!!!!')

b()

makeup...
go!!!!

多个装饰器

如果有多个装饰器，执行的顺序将会「由下而上」触发（函数一层层往上），下方的代码，会先装饰 a3，接着装饰 a2，最后装饰 a1。

def a1(func):
    print('1')
    return func

def a2(func):
    print('2')
    return func

def a3(func):
    print('3')
    return func

@a1
@a2
@a3
def b():
    print('go!!!!')

b()
# 3
# 2
# 1
# go!!!!

单一参数处理

如果装饰器遇到带有参数的函数，同样能将参数一并带入处理，实现的方式如下代码，在装饰器函数 a 里，新增一个函数内的函数 c，并透过函数 c 来获取被装饰函数 b 的参数 msg。

def a(func):
    def c(m):             # 新增一个内部函数，待有一个参数
        print('makeup...')
        return func(m)      # 返回 func(m)
    return c

@a
def b(msg):
    print(msg)

b('go!!!!')

makeup...
go!!!!

多个参数处理

如果遇到被装饰的函数有多个参数，可以使用 *args 和 **kwargs 运算符来取得所有的参数，下方的例子，b 函数传入的参数有序列以及带有关键字参数的参数，就能将这些参数传递给装饰器函数 a。

def a(func):
    def c(*args, **kwargs):
        print(args)
        print(kwargs)
        print('ok...')
        return func(*args, **kwargs)
    return c

@a
def b(*args, **kwargs):
    print('go!!!!')

b([123, 456], x=1, y=2, z=3)

# ([123, 456],)
# {'x': 1, 'y': 2, 'z': 3}
# ok...
# go!!!!

python33

python教程31

python · 目录

上一篇Python教程 - 闭包