装饰器
装饰器的作用是,对于有切面需求的场景,例如:调用函数的日志等,采用其他方式显得繁琐。而使用装饰器可以美观有快捷的实现代码复用。以下是对装饰器关键部分的笔记。来源:www.zhihu.com/question/26930016
图1 简单装饰器简单装饰器:就是将函数作为参数放到装饰器函数中。装饰器函数返回的是个wrapper函数,这个函数内执行了作为参数的函数的操作以及其他一些跟这个函数属性相关联的操作。本例中,bar函数作为参数传入use_logging函数,然后返回wrapper函数赋值给bar,在调用bar时,实际上调用的是wrapper函数。简单装饰器的一个问题是:在使用时,需要有语句:bar=use_logging(bar),繁琐不美观且容易引起错误。使用@装饰器名 的语法糖可以解决这一问题。
图2 @装饰器名 语法糖@装饰器名语法糖:可以解决简单装饰器带来的每次调用前要使用bar=sue_logging(bar)引起的混乱繁琐不美观。其与简单装饰器在语法上的不同只在于函数前加一行@装饰器名语法糖。
现在假设,有一个标志位,只在这个标志位为1时对函数调用进行记录。上述装饰器不能实现这个功能,可以使用带参数的装饰器。
图3 带参数的装饰器这种装饰器的运行逻辑是,将参数传入到decorator中,再把函数传入,在wrapper中执行操作。在使用时,只需要在定义函数前加带参数的语法糖即可。调用直接使用参数名。
图4 类装饰器除了函数装饰器外,还有类装饰器,这种装饰器具有灵活度大、高内聚、封装性等优点。在构造函数中,将要执行的函数赋值给装饰器,如果是带参数的装饰器,同样可以在构造函数中赋值。在使用是,@语法糖将定义的函数名传入构造函数中,调用时只需要调用函数名即可。
以上装饰器虽然带来了减少重复代码的好处,但是,导致了原始函数的信息丢失。可以使用functools模块的wraps进行装饰。wraps同样是一个装饰器,实现的就是将原来函数的信息复制到装饰器中。
图5 functools.wraps装饰器对于多个装饰器,先用内层的装饰,再用外层的进行装饰。