标点符(钱魏 Way)

Python装饰器的学习笔记

装饰器(decorator)是一种高级Python语法。可以对一个函数、方法或者类进行加工。在Python中,我们有多种方法对函数和类进行加工,相对于其它方式,装饰器语法简单,代码可读性高。因此,装饰器在Python项目中有广泛的应用。修饰器经常被用于有切面需求的场景,较为经典的有插入日志、性能测试、事务处理, Web权限校验, Cache等。很有名的例子,就是咖啡,加糖的咖啡,加牛奶的咖啡。本质上,还是咖啡,只是在原有的东西上,做了“装饰”,使之附加一些功能或特性。

装饰器的优点是能够抽离出大量函数中与函数功能本身无关的雷同代码并继续重用。即,可以将函数“修饰”为完全不同的行为,可以有效的将业务逻辑正交分解。概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能。例如记录日志,需要对某些函数进行记录。笨的办法,每个函数加入代码,如果代码变了,就悲催了。装饰器的办法,定义一个专门日志记录的装饰器,对需要的函数进行装饰。

Python 的 Decorator在使用上和Java/C#的Annotation很相似,都是在方法名前面加一个@XXX注解来为这个方法装饰一些东西。但是,Java/C#的Annotation也很让人望而却步,在使用它之前你需要了解一堆Annotation的类库文档,让人感觉就是在学另外一门语言。而Python使用了一种相对于Decorator Pattern和Annotation来说非常优雅的方法,这种方法不需要你去掌握什么复杂的OO模型或是Annotation的各种类库规定,完全就是语言层面的玩法:一种函数式编程的技巧。

在Python中,装饰器实现是十分方便。原因是:函数可以被扔来扔去。

python的函数就是对象

要理解装饰器,就必须先知道,在python里,函数也是对象(functions are objects)。明白这一点非常重要,让我们通过一个例子来看看为什么。

python 函数的另一个有趣的特性是,它们可以在另一个函数体内定义。

函数引用(Functions references)

你刚刚已经知道了,python的函数也是对象,因此:

  • 可以被赋值给变量
  • 可以在另一个函数体内定义

那么,这样就意味着一个函数可以返回另一个函数 :-),来看个例子:

既然可以返回一个函数,那么也就可以像参数一样传递:

现在已经具备了理解装饰器的所有基础知识了。装饰器也就是一种包装材料,它们可以让你在执行被装饰的函数之前或之后执行其他代码,而且不需要修改函数本身。

手工制作装饰器(Handcrafted decorators)

你可以像这样来定制:

现在你大概希望,每次调用 a_stand_alone_function 时,实际调用的是a_stand_alone_function_decorated 。这很容易,只要把 my_shiny_new_decorator 返回的函数覆盖 a_stand_alone_function 就可以了:

揭秘装饰器(Decorators demystified)

我们用装饰器的语法来重写一下前面的例子:

是的,这就完了,就这么简单。@decorator 只是下面这条语句的简写(shortcut):

装饰器其实就是装饰器模式的一个python化的变体(pythonic variant)。为了方便开发,python已经内置了好几种经典的设计模式,比如迭代器(iterators)。 当然,你还可以堆积使用装饰器(you can cumulate decorators):

用python的装饰器语法表示:

装饰器放置的顺序 很重要:

给装饰器函数传参(Passing arguments to the decorated function)

含参的装饰器

在上面的装饰器调用中,比如@decorator,该装饰器默认它后面的函数是唯一的参数。装饰器的语法允许我们调用decorator时,提供其它参数,比如@decorator(a)。这样,就为装饰器的编写和使用提供了更大的灵活性。

上面的pre_str是允许参数的装饰器。它实际上是对原有装饰器的一个函数封装,并返回一个装饰器。我们可以将它理解为一个含有环境参量的闭包。当我们使用@pre_str(‘^_^’)调用的时候,Python能够发现这一层的封装,并把参数传递到装饰器的环境中。该调用相当于:

装饰方法(Decorating methods)

Python的一个伟大之处在于:方法和函数几乎是一样的(methods and functions are really the same),除了方法的第一个参数应该是当前对象的引用(也就是 self)。这也就意味着只要记住把 self 考虑在内,你就可以用同样的方法给方法创建装饰器了:

当然,如果你想编写一个非常通用的装饰器,可以用来装饰任意函数和方法,你就可以无视具体参数了,直接使用 *args, **kwargs 就行:

装饰类

在上面的例子中,装饰器接收一个函数,并返回一个函数,从而起到加工函数的效果。在Python 2.6以后,装饰器被拓展到类。一个装饰器可以接收一个类,并返回一个类,从而起到加工类的效果。

在decorator中,我们返回了一个新类newClass。在新类中,我们记录了原来类生成的对象(self.wrapped),并附加了新的属性total_display,用于记录调用display的次数。我们也同时更改了display方法。通过修改,我们的Bird类可以显示调用display的次数了。

给装饰器传参(Passing arguments to the decorator)

现在对于给装饰器本身传参数,你有什么看法呢?好吧,这样说有点绕,因为装饰器必须接受一个函数作为参数,所以就不能把被装饰的函数的参数,直接传给装饰器(you cannot pass the decorated function arguments directly to the decorator.)

在直奔答案之前,我们先写一个小提示:

这完全一样,都是 my_decorator 被调用。所以当你使用 @my_decorator 时,你在告诉 python 去调用 “被变量 my_decorator 标记的” 函数(the function ‘labeled by the variable “my_decorator”‘)。这很重要,因为你给的这个标签能直接指向装饰器或者其他!

不要感到惊讶,让我们做一件完全一样的事情,只不过跳过了中间变量:

再做一次,代码甚至更短:

我们在用 @ 语法调用了函数 , 那么回到带参数的装饰器。如果我们能够使用一个函数动态(on the fly)的生成装饰器,那么我们就能把参数传递给那个函数,对吗?

这就是了,带参数的装饰器。参数也可以设置为变量:

如你所见,你可以给装饰器传递参数,就好像其他任意一个使用了这种把戏的函数一样(。如果你愿意,甚至可以使用 *args, **kwargs。但是,记住,装置器只调用一次,仅当python导入这个脚本时。你不能在之后动态的设置参数。当你执行 import x 时,这个函数已经被装饰了,因此你不能修改任何东西。

实践:装饰器装饰一个装饰器(Let’s practice: a decorator to decorate a decorator)

我将展示一段能用来创建能接受通用的任意参数的装饰器的代码。毕竟,为了能接受参数,我们用了另一个函数来创建我们的装饰器。我们包装了装饰器。在我们刚刚看到的东西里,还有用来包装函数的吗?是的,就是装饰器。让我们给装饰器写一个装饰器来玩玩:

它可以像这样使用:

我知道,你上一次有这种感觉,是在听一个人说“在理解递归之前,你必须先理解递归”之后。但是现在,掌握之后,你不觉得很爽吗?

装饰器最佳实践(Best practices while using decorators)

  • 装饰器是在 python 2.4 之后才有的,所以先确定你的代码运行时;
  • 记住这点:装饰器降低了函数调用效率;
  • 你不能“解装饰”一个函数。有一些能用来创建可以移除的装饰器的方法,但没人用它们。所以一个函数一旦被装饰了,就结束了(不能改变了)。
  • 装饰器包装了函数,这使得会难以调试。

Python 2.5 通过提供了一个 functools 模块解决了最后一个问题。functools.wraps 把任意被包装函数的函数名、模块名和 docstring 拷贝给了 wrapper. 有趣的事是,functools.wraps 也是一个装饰器:-)

装饰器如何才能有用(How can the decorators be useful?)

现在问题来了:我能用装饰器来干嘛?看起来很酷也很强大,但是来一个实际例子才更好。一个典型的用途是,用来扩展一个外部导入的函数(你不能修改)的行为,或者为了调试(你不想修改这个函数,因为只是暂时的)。你也可以用装饰器实现只用一段相同的代码来扩展成几个不同的函数,而且你不需要每次都重写这段代码。这样就是常说的 DRY。比如:

当然,装饰器的好处就是你可以几乎用来装饰所有东西,而且不要重写。也就是我说的 DRY:

Python 语言本身也提供了一些装饰器:property、staticmethod 等。Django 用装饰器来管理换成和视图权限。Twisted 用来伪装 内联异步函数调用。

总结

装饰器的核心作用是name binding。这种语法是Python多编程范式的又一个体现。大部分Python用户都不怎么需要定义装饰器,但有可能会使用装饰器。鉴于装饰器在Python项目中的广泛使用,了解这一语法是非常有益的。

原文链接:http://stackoverflow.com/questions/739654/how-can-i-make-a-chain-of-function-decorators-in-python#answer-1594484

码字很辛苦,转载请注明来自标点符《Python装饰器的学习笔记》

评论