跨过这道装饰器的坎,或许你就升级了

作者:Toby

原文:https://betacat.online/posts/python-decorator/index.html

Python 中的装饰器是你进入 Python 大门的一道坎,不管你跨不跨过去它都在那里。

为什么需要装饰器

我们假设你的程序实现了 和 两个函数。

但是在实际调用中,我们发现程序出错了,上面的代码打印了两个 hello 。经过调试你发现是 出错了。老板要求调用每个方法前都要记录进入函数的时间和名称,比如这样:

好,小 A 是个毕业生,他是这样实现的。

很 low 吧? 嗯是的。小 B 工作有一段时间了,他告诉小 A 应该这样写。

是不是好一点?那当然,但是每个业务函数里都要调用一下 函数,是不是很难受?万一老板说 say 相关的函数不用 debug , do 相关的才需要呢?

那么装饰器这时候应该登场了。

装饰器本质上是一个 Python 函数,它可以让其他函数在不需要做任何代码变动的前提下增加额外功能,装饰器的返回值也是一个函数对象。它经常用于有切面需求的场景,比如:插入日志、性能测试、事务处理、缓存、权限校验等场景。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

概括的讲,装饰器的作用就是为已经存在的函数或对象添加额外的功能

怎么写一个装饰器

在早些时候 \ ( Python Version

上面的 debug 函数其实已经是一个装饰器了,它对原函数做了包装并返回了另外一个函数,额外添加了一些功能。因为这样写实在不太优雅, [email

protected],下面代码等同于早期的写法。

这是最简单的装饰器,但是有一个问题,如果被装饰的函数需要传入参数,那么这个装饰器就坏了。因为返回的函数并不能接受参数,你可以指定装饰器函数 接受和原函数一样的参数,比如:

这样你就解决了一个问题,但又多了 N 个问题。因为函数有千千万,你只管你自己的函数,别人的函数参数是什么样子,鬼知道?还好 Python 提供了可变参数 和关键字参数 ,有了这两个参数,装饰器就可以用于任意目标函数了。

至此,你已完全掌握初级的装饰器写法。

高级一点的装饰器

带参数的装饰器和类装饰器属于进阶的内容。在理解这些装饰器之前,最好对函数的闭包和装饰器的接口约定有一定了解。(参见http://betacat.online/posts/python-closure/)

带参数的装饰器

假设我们前文的装饰器需要完成的功能不仅仅是能在进入某个函数后打出 log 信息,而且还需指定 log 的级别,那么装饰器就会是这样的。

是不是有一些晕?你可以这么理解,当带参数的装饰器被打在某个函数上时,比如 ,它其实是一个函数,会马上被执行,只要这个它返回的结果是一个装饰器时,那就没问题。细细再体会一下。

基于类实现的装饰器

装饰器函数其实是这样一个接口约束,它必须接受一个 callable 对象作为参数,然后返回一个 callable 对象。在 Python 中一般 callable 对象都是函数,但也有例外。只要某个对象重载了 方法,那么这个对象就是 callable 的。

像 这样前后都带下划线的方法在 Python 中被称为内置方法,有时候也被称为魔法方法。重载这些魔法方法一般会改变对象的内部行为。上面这个例子就让一个类对象拥有了被调用的行为。

回到装饰器上的概念上来,装饰器要求接受一个 callable 对象,并返回一个 callable 对象(不太严谨,详见后文)。那么用类来实现也是也可以的。我们可以让类的构造函数 接受一个函数,然后重载 并返回一个函数,也可以达到装饰器函数的效果。

带参数的类装饰器

如果需要通过类形式实现带参数的装饰器,那么会比前面的例子稍微复杂一点。那么在构造函数里接受的就不是一个函数,而是传入的参数。通过类把这些参数保存起来。然后在重载 方法是就需要接受一个函数并返回一个函数。

内置的装饰器

内置的装饰器和普通的装饰器原理是一样的,只不过返回的不是函数,而是类对象,所以更难理解一些。

@ property

在了解这个装饰器前,你需要知道在不使用装饰器怎么写一个属性。

以上就是一个Python属性的标准写法,其实和Java挺像的,能达到一样的效果但看起来更简单。

属性有三个装饰器: , ,

,都是在 的基础上做了一些封装,因为 和 是 的第二和第三个参数, 装饰器和不带 的属性装饰器效果是一样的,估计只是为了凑数,本身没有任何存在的意义。经过 装饰过的函数返回的不再是一个函数,而是一个 对象。

@ classmethod

有了 装饰器的了解,这两个装饰器的原理是差不多的。 返回的是一个 类对象,而 返回的是一个 类对象。他们都是调用的是各自的 构造函数。

至此,我们上文提到的装饰器接口定义可以更加明确一些,装饰器必须接受一个 callable 对象,其实它并不关心你返回什么,可以是另外一个 callable 对象(大部分情况),也可以是其他类对象,比如 property 。

装饰器里的那些坑

装饰器可以让你代码更加优雅,减少重复,但也不全是优点,也会带来一些问题。

位置错误的代码

让我们直接看示例代码。

在装饰器中我在各个可能的位置都加上了 print 语句,用于记录被调用的情况。你知道他们最后打印出来的顺序吗?如果你心里没底,那么最好不要在装饰器函数之外添加逻辑功能,否则这个装饰器就不受你控制了。以下是输出结果:

错误的函数签名和文档

装饰器装饰过的函数看上去名字没变,其实已经变了。

为什么会这样呢?@等同于这样的写法。

其实返回的函数名字刚好是 ,那么上面的这个语句刚好就是把这个结果赋值给 , 的 自然也就是 了,不仅仅是 ,其他属性也都是来自 ,比如 , 等等。

使用标准库里的 ,可以基本解决这个问题。

看上去不错!主要问题解决了,但其实还不太完美。因为函数的签名和源码还是拿不到的。

如果要彻底解决这个问题可以借用第三方包,比如 。后文有介绍。

不能装饰@staticmethod 或者 @classmethod”

当你想把装饰器用在一个静态方法或者类方法时,不好意思,报错了。

前面已经解释了 这个装饰器,其实它返回的并不是一个 callable 对象,而是一个 对象,那么它是不符合装饰器要求的(比如传入一个 callable 对象),你自然不能在它之上再加别的装饰器。要解决这个问题很简单,只要把你的装饰器放在 之前就好了,因为你的装饰器返回的还是一个正常的函数,然后再加上一个 是不会出问题的。

如何优化你的装饰器

嵌套的装饰函数不太直观,我们可以使用第三方包类改进这样的情况,让装饰器函数可读性更好。

decorator.py

decorator.py是一个非常简单的装饰器加强包。你可以很直观的先定义包装函数 ,再使用 方法就可以完成一个装饰器。

你也可以使用它自带的 装饰器来完成你的装饰器。

实现的装饰器能完整保留原函数的 , 和 ,唯一有问题的就是 返回的还是装饰器的源代码,你需要改成 。

wrapt

wrapt是一个功能非常完善的包,用于实现各种你想到或者你没想到的装饰器。使用 wrapt 实现的装饰器你不需要担心之前 inspect 中遇到的所有问题,因为它都帮你处理了,甚至 也准确无误。

使用 wrapt 你只需要定义一个装饰器函数,但是函数签名是固定的,必须是 ,注意第二个参数 是必须的,就算你不用它。当装饰器装饰在不同位置时它将得到不同的值,比如装饰在类实例方法时你可以拿到这个类实例。根据 的值你能够更加灵活的调整你的装饰器。另外, 和 也是固定的,注意前面没有星号。在装饰器内部调用原函数时才带星号。

如果你需要使用 wrapt 写一个带参数的装饰器,可以这样写。

关于 wrapt 的使用,建议查阅官方文档,在此不在赘述。

http://wrapt.readthedocs.io/en/latest/quick-start.html

小结

Python 的装饰器和 Java 的注解( Annotation )并不是同一回事,和 C# 中的特性( Attribute )也不一样,完全是两个概念。

装饰器的理念是对原函数、对象的加强,相当于重新封装,所以一般装饰器函数都被命名为 ,意义在于包装。函数只有在被调用时才会发挥其作用。比如 装饰器可以在函数执行时额外输出日志, 装饰过的函数可以缓存计算结果等等。

而注解和特性则是对目标函数或对象添加一些属性,相当于将其分类。这些属性可以通过反射拿到,在程序运行时对不同的特性函数或对象加以干预。比如带有 的函数就当成准备步骤执行,或者找到所有带有 的函数依次执行等等。

至此我所了解的装饰器已经讲完,但是还有一些内容没有提到,比如装饰类的装饰器。有机会再补充。谢谢观看。

本文源码

转载声明:本文转载自「编程派」,搜索「codingpy」即可关注。

- 结束 -

  • 发表于:
  • 原文链接https://kuaibao.qq.com/s/20190131B18D9E00?refer=cp_1026
  • 腾讯「云+社区」是腾讯内容开放平台帐号(企鹅号)传播渠道之一,根据《腾讯内容开放平台服务协议》转载发布内容。
  • 如有侵权,请联系 yunjia_community@tencent.com 删除。

扫码关注云+社区

领取腾讯云代金券