Python-装饰器

Posted 2021-02-15

装饰器

函数也是一个对象，而且函数对象可以被赋值给变量，所以，通过变量也能调用该函数。

创建一个 now 函数 然后赋值给变量 f 通过调用 f() 来运行

函数对象有一个 name属性，可以拿到函数的名字：

如果要增强 now() 函数的功能，比如now()函数调用前先说明正在使用哪个函数，但又不希望修改now 函数的定义，这种在代码运行期间动态增加功能的方式，称之为 “装饰器”（Decorator）。

定义如下：

运行：

Python的@语法，把decorator置于函数的定义处：

把@log放到now()函数的定义处，相当于执行了语句：

now = log(now)

由于log()是一个decorator，返回一个函数，所以，原来的now()函数仍然存在，只是现在同名的now变量指向了新的函数，于是调用now()将执行新函数，即在log()函数中返回的Pg()函数。

Pg()函数的参数定义是(*x, **kw)，因此，Pg()函数可以接受任意参数的调用。在wrapper()函数内，首先打印日志，再紧接着调用原始函数。

如果decorator本身需要传入参数，那就需要编写一个返回decorator的高阶函数，比如，要自定义log的文本：

执行结果如下：

和两层嵌套的decorator相比，3层嵌套的效果是这样的：

now = log(‘君子终日乾乾‘) (now)

也就是首先 执行 log(‘君子终日乾乾‘) 返回 fnc 函数 ，然后 fnc执行 参数是 now ，再返回 pg 函数 。。 慢慢执行下去。

以上两种decorator的定义都没有问题，但还差最后一步。函数也是对象，它有name等属性，但你去看经过decorator装饰之后的函数，它们的name已经从原来的 ‘now‘ 变成了 ‘Pg‘：

不需要编写Pg.name = func.name这样的代码，Python内置的functools.wraps就是干这个事的，所以，一个完整的decorator的写法如下：