Python 装饰器

简介:

Python 装饰器

文章先由stackoverflow上面的一个问题引起吧,如果使用如下的代码:

@makebold

@makeitalic

def say():

   return "Hello"

打印出如下的输出:

<b><i>Hello<i></b>

你会怎么做?最后给出的答案是:

def makebold(fn):

    def wrapped():

        return "<b>" + fn() +"</b>"

    return wrapped

 

def makeitalic(fn):

    def wrapped():

        return "<i>" + fn() +"</i>"

    return wrapped

 

@makebold

@makeitalic

def hello():

    return "hello world"

 

print hello() ## 返回 <b><i>helloworld</i></b>

现在我们来看看如何从一些最基础的方式来理解Python的装饰器。英文讨论参考Here

装饰器是一个很著名的设计模式,经常被用于有切面需求的场景,较为经典的有插入日志、性能测试、事务处理等。装饰器是解决这类问题的绝佳设计,有了装饰器,我们就可以抽离出大量函数中与函数功能本身无关的雷同代码并继续重用。概括的讲,装饰器的作用就是为已经存在的对象添加额外的功能。

1.1. 需求是怎么来的?

装饰器的定义很是抽象,我们来看一个小例子。

 

def foo():

    print 'in foo()'

foo()

这是一个很无聊的函数没错。但是突然有一个更无聊的人,我们称呼他为B君,说我想看看执行这个函数用了多长时间,好吧,那么我们可以这样做:

import time

def foo():

    start = time.clock()

    print 'in foo()'

    end = time.clock()

    print 'used:', end - start

 

foo()

很好,功能看起来无懈可击。可是蛋疼的B君此刻突然不想看这个函数了,他对另一个叫foo2的函数产生了更浓厚的兴趣。

怎么办呢?如果把以上新增加的代码复制到foo2里,这就犯了大忌了~复制什么的难道不是最讨厌了么!而且,如果B君继续看了其他的函数呢?

1.2. 以不变应万变,是变也

还记得吗,函数在Python中是一等公民,那么我们可以考虑重新定义一个函数timeit,将foo的引用传递给他,然后在timeit中调用foo并进行计时,这样,我们就达到了不改动foo定义的目的,而且,不论B君看了多少个函数,我们都不用去修改函数定义了!

import time

 

def foo():

    print 'in foo()'

 

def timeit(func):

    start = time.clock()

    func()

    end =time.clock()

    print 'used:', end - start

 

timeit(foo)

看起来逻辑上并没有问题,一切都很美好并且运作正常!……等等,我们似乎修改了调用部分的代码。原本我们是这样调用的:foo(),修改以后变成了:timeit(foo)。这样的话,如果fooN处都被调用了,你就不得不去修改这N处的代码。或者更极端的,考虑其中某处调用的代码无法修改这个情况,比如:这个函数是你交给别人使用的。

1.3. 最大限度地少改动!

既然如此,我们就来想想办法不修改调用的代码;如果不修改调用代码,也就意味着调用foo()需要产生调用timeit(foo)的效果。我们可以想到将timeit赋值给foo,但是timeit似乎带有一个参数……想办法把参数统一吧!如果timeit(foo)不是直接产生调用效果,而是返回一个与foo参数列表一致的函数的话……就很好办了,将timeit(foo)的返回值赋值给foo,然后,调用foo()的代码完全不用修改!

#-*- coding: UTF-8-*-

import time

 

def foo():

    print 'in foo()'

 

定义一个计时器,传入一个,并返回另一个附加了计时功能的方法

def timeit(func):

    

    # 定义一个内嵌的包装函数,给传入的函数加上计时功能的包装

    def wrapper():

        start = time.clock()

        func()

        end =time.clock()

        print 'used:', end - start

    

    # 将包装后的函数返回

    return wrapper

 

foo = timeit(foo)

foo()

这样,一个简易的计时器就做好了!我们只需要在定义foo以后调用foo之前,加上foo = timeit(foo),就可以达到计时的目的,这也就是装饰器的概念,看起来像是footimeit装饰了。在在这个例子中,函数进入和退出时需要计时,这被称为一个横切面(Aspect),这种编程方式被称为面向切面的编程(Aspect-Oriented Programming)。与传统编程习惯的从上往下执行方式相比较而言,像是在函数执行的流程中横向地插入了一段逻辑。在特定的业务领域里,能减少大量重复代码。面向切面编程还有相当多的术语,这里就不多做介绍,感兴趣的话可以去找找相关的资料。

这个例子仅用于演示,并没有考虑foo带有参数和有返回值的情况,完善它的重任就交给你了:)

上面这段代码看起来似乎已经不能再精简了,Python于是提供了一个语法糖来降低字符输入量。

import time

 

def timeit(func):

    def wrapper():

        start = time.clock()

        func()

        end =time.clock()

        print 'used:', end - start

    return wrapper

 

@timeit

def foo():

    print 'in foo()'

 

foo()

重点关注第11行的@timeit,在定义上加上这一行与另外写foo = timeit(foo)完全等价,千万不要以为@有另外的魔力。除了字符输入少了一些,还有一个额外的好处:这样看上去更有装饰器的感觉。

-------------------

要理解python的装饰器,我们首先必须明白在Python中函数也是被视为对象。这一点很重要。先看一个例子:

defshout(word="yes") :

    return word.capitalize()+" !"

 

print shout()

输出 : 'Yes !'

 

作为一个对象,你可以把函数赋给任何其他对象变量

 

scream = shout

 

注意我们没有使用圆括号,因为我们不是在调用函数

我们把函数shout赋给scream,也就是说你可以通过scream调用shout

 

print scream()

输出 : 'Yes !'

 

还有,你可以删除旧的名字shout,但是你仍然可以通过scream来访问该函数

 

del shout

try :

    print shout()

except NameError, e :

    print e

    #输出 : "name 'shout' is not defined"

 

print scream()

输出 : 'Yes !'

我们暂且把这个话题放旁边,我们先看看python另外一个很有意思的属性:可以在函数中定义函数:

def talk() :

 

    # 你可以在talk中定义另外一个函数

    def whisper(word="yes") :

        return word.lower()+"...";

 

    # ... 并且立马使用它

 

    print whisper()

 

你每次调用'talk',定义在talk里面的whisper同样也会被调用

talk()

输出 :

# yes...

 

但是"whisper" 不会单独存在:

 

try :

    print whisper()

except NameError, e :

    print e

    #输出 : "name 'whisper' is not defined"*

函数引用

从以上两个例子我们可以得出,函数既然作为一个对象,因此:

1. 其可以被赋给其他变量

2. 其可以被定义在另外一个函数内

这也就是说,函数可以返回一个函数,看下面的例子:

defgetTalk(type="shout") :

 

    # 我们定义另外一个函数

    def shout(word="yes") :

        return word.capitalize()+" !"

 

    def whisper(word="yes") :

        return word.lower()+"...";

 

    # 然后我们返回其中一个

    if type == "shout" :

        # 我们没有使用(),因为我们不是在调用该函数

        # 我们是在返回该函数

        return shout

    else :

        return whisper

 

然后怎么使用呢 ?

 

把该函数赋予某个变量

talk = getTalk()    

 

这里你可以看到talk其实是一个函数对象:

print talk

#输出 : <function shout at0xb7ea817c>

 

该对象由函数返回的其中一个对象:

print talk()

 

或者你可以直接如下调用 :

printgetTalk("whisper")()

#输出 : yes...

还有,既然可以返回一个函数,我们可以把它作为参数传递给函数:

defdoSomethingBefore(func) :

    print "I do something before then Icall the function you gave me"

    printfunc()

 

doSomethingBefore(scream)

#输出 :

#I do somethingbefore then I call the function you gave me

#Yes !

这里你已经足够能理解装饰器了,其他它可被视为封装器。也就是说,它能够让你在装饰前后执行代码而无须改变函数本身内容。

手工装饰

那么如何进行手动装饰呢?

装饰器是一个函数,而其参数为另外一个函数

defmy_shiny_new_decorator(a_function_to_decorate) :

 

    # 在内部定义了另外一个函数:一个封装器。

    # 这个函数将原始函数进行封装,所以你可以在它之前或者之后执行一些代码

    defthe_wrapper_around_the_original_function() :

 

        # 放一些你希望在真正函数执行前的一些代码

        print "Before the functionruns"

 

        #执行原始函数

        a_function_to_decorate()

 

        # 放一些你希望在原始函数执行后的一些代码

        print "After the functionruns"

 

    #在此刻,"a_function_to_decrorate"还没有被执行,我们返回了创建的封装函数

    #封装器包含了函数以及其前后执行的代码,其已经准备完毕

    returnthe_wrapper_around_the_original_function

 

现在想象下,你创建了一个你永远也不远再次接触的函数

defa_stand_alone_function() :

    print "I am a stand alone function,don't you dare modify me"

 

a_stand_alone_function()

#输出: I am a stand alone function, don'tyou dare modify me

 

好了,你可以封装它实现行为的扩展。可以简单的把它丢给装饰器

装饰器将动态地把它和你要的代码封装起来,并且返回一个新的可用的函数。

a_stand_alone_function_decorated= my_shiny_new_decorator(a_stand_alone_function)

a_stand_alone_function_decorated()

#输出 :

#Before the functionruns

#I am a stand alonefunction, don't you dare modify me

#After the functionruns

现在你也许要求当每次调用a_stand_alone_function时,实际调用却是a_stand_alone_function_decorated。实现也很简单,可以用my_shiny_new_decorator来给a_stand_alone_function重新赋值。

a_stand_alone_function= my_shiny_new_decorator(a_stand_alone_function)

a_stand_alone_function()

#输出 :

#Before the functionruns

#I am a stand alonefunction, don't you dare modify me

#After the functionruns

 

# And guess what,that's EXACTLY what decorators do !

装饰器揭秘

前面的例子,我们可以使用装饰器的语法:

@my_shiny_new_decorator

defanother_stand_alone_function() :

    print "Leave me alone"

 

another_stand_alone_function()

#输出 :

#Before the functionruns

#Leave me alone

#After the functionruns

当然你也可以累积装饰:

def bread(func) :

    def wrapper() :

        print "</''''''\>"

        func()

        print "<\______/>"

    return wrapper

 

def ingredients(func):

    def wrapper() :

        print "#tomatoes#"

        func()

        print "~salad~"

    return wrapper

 

defsandwich(food="--ham--") :

    print food

 

sandwich()

#输出 : --ham--

sandwich =bread(ingredients(sandwich))

sandwich()

#outputs :

#</''''''\>

# #tomatoes#

# --ham--

# ~salad~

#<\______/>

使用python装饰器语法:

@bread

@ingredients

defsandwich(food="--ham--") :

    print food

 

sandwich()

#输出 :

#</''''''\>

# #tomatoes#

# --ham--

# ~salad~

#<\______/>

装饰器的顺序很重要,需要注意:

@ingredients

@bread

defstrange_sandwich(food="--ham--") :

    print food

 

strange_sandwich()

#输出 :

##tomatoes#

#</''''''\>

# --ham--

#<\______/>

# ~salad~

最后回答前面提到的问题:

装饰器makebold用于转换为粗体

def makebold(fn):

    # 结果返回该函数

    def wrapper():

        # 插入一些执行前后的代码

        return "<b>" + fn() +"</b>"

    return wrapper

 

装饰器makeitalic用于转换为斜体

def makeitalic(fn):

    # 结果返回该函数

    def wrapper():

        # 插入一些执行前后的代码

        return "<i>" + fn() +"</i>"

    return wrapper

 

@makebold

@makeitalic

def say():

    return "hello"

 

print say()

#输出:<b><i>hello</i></b>

 

等同于

def say():

    return "hello"

say =makebold(makeitalic(say))

 

print say()

#输出:<b><i>hello</i></b>

内置的装饰器

内置的装饰器有三个,分别是staticmethodclassmethodproperty,作用分别是把类中定义的实例方法变成静态方法、类方法和类属性。由于模块里可以定义函数,所以静态方法和类方法的用处并不是太多,除非你想要完全的面向对象编程。而属性也不是不可或缺的,Java没有属性也一样活得很滋润。从我个人的Python经验来看,我没有使用过property,使用staticmethodclassmethod的频率也非常低。

class Rabbit(object):

    

    def __init__(self, name):

        self._name = name

    

    @staticmethod

    defnewRabbit(name):

        return Rabbit(name)

    

    @classmethod

    def newRabbit2(cls):

        return Rabbit('')

    

    @property

    def name(self):

        return self._name

这里定义的属性是一个只读属性,如果需要可写,则需要再定义一个setter

@name.setter

def name(self, name):

    self._name = name

functools模块

functools模块提供了两个装饰器。这个模块是Python 2.5后新增的,一般来说大家用的应该都高于这个版本。但我平时的工作环境是2.4 T-T

2.3.1. wraps(wrapped[, assigned][, updated]): 
这是一个很有用的装饰器。看过前一篇反射的朋友应该知道,函数是有几个特殊属性比如函数名,在被装饰后,上例中的函数名foo会变成包装函数的名字wrapper,如果你希望使用反射,可能会导致意外的结果。这个装饰器可以解决这个问题,它能将装饰过的函数的特殊属性保留。

import time

import functools

 

def timeit(func):

    @functools.wraps(func)

    def wrapper():

        start = time.clock()

        func()

        end =time.clock()

        print 'used:', end - start

    return wrapper

 

@timeit

def foo():

    print 'in foo()'

 

foo()

print foo.__name__

首先注意第5行,如果注释这一行,foo.__name__将是'wrapper'。另外相信你也注意到了,这个装饰器竟然带有一个参数。实际上,他还有另外两个可选的参数,assigned中的属性名将使用赋值的方式替换,而updated中的属性名将使用update的方式合并,你可以通过查看functools的源代码获得它们的默认值。对于这个装饰器,相当于wrapper =functools.wraps(func)(wrapper)

2.3.2. total_ordering(cls): 
这个装饰器在特定的场合有一定用处,但是它是在Python 2.7后新增的。它的作用是为实现了至少__lt____le____gt____ge__其中一个的类加上其他的比较方法,这是一个类装饰器。如果觉得不好理解,不妨仔细看看这个装饰器的源代码:

 def total_ordering(cls):

54      """Class decorator thatfills in missing ordering methods"""

55      convert = {

56          '__lt__': [('__gt__', lambda self,other: other < self),

57                     ('__le__', lambda self,other: not other < self),

58                     ('__ge__', lambda self,other: not self < other)],

59          '__le__': [('__ge__', lambda self,other: other <= self),

60                     ('__lt__', lambda self,other: not other <= self),

61                     ('__gt__', lambda self, other: not self<= other)],

62          '__gt__': [('__lt__', lambda self,other: other > self),

63                     ('__ge__', lambda self,other: not other > self),

64                     ('__le__', lambda self,other: not self > other)],

65          '__ge__': [('__le__', lambda self,other: other >= self),

66                     ('__gt__', lambda self,other: not other >= self),

67                     ('__lt__', lambda self,other: not self >= other)]

68      }

69      roots = set(dir(cls)) & set(convert)

70      if not roots:

71          raise ValueError('must define atleast one ordering operation: < > <= >=')

72      root = max(roots)       # prefer __lt__ to __le__ to __gt__ to__ge__

73      for opname, opfunc in convert[root]:

74          if opname not in roots:

75              opfunc.__name__ = opname

76              opfunc.__doc__ = getattr(int,opname).__doc__

77              setattr(cls, opname, opfunc)

78      return cls

 


本文转自 周新宇1991 51CTO博客,原文链接:http://blog.51cto.com/zhouxinyu1991/1836479,如需转载请自行联系原作者


相关文章
|
8天前
|
开发者 Python
探索Python中的装饰器:从基础到高级应用
本文将带你深入了解Python中的装饰器,这一强大而灵活的工具。我们将一起探讨装饰器的基本概念,它们如何工作,以及如何使用它们来增强函数和类的功能,同时不改变其核心逻辑。通过具体代码示例,我们将展示装饰器的创建和使用,并探索一些高级应用,比如装饰器堆栈和装饰带参数的装饰器。无论你是初学者还是有经验的开发者,这篇文章都将为你提供新的视角,帮助你更有效地使用装饰器来简化和优化你的代码。
|
9天前
|
测试技术 数据安全/隐私保护 开发者
探索Python中的装饰器:从基础到高级应用
装饰器在Python中是一个强大且令人兴奋的功能,它允许开发者在不修改原有函数代码的前提下增加额外的功能。本文将通过具体代码示例,带领读者从装饰器的基础概念入手,逐步深入到高级用法,如带参数的装饰器和装饰器嵌套等。无论你是初学者还是有经验的开发者,这篇文章都将为你提供有价值的见解和技巧。
|
9天前
|
开发框架 数据建模 中间件
Python中的装饰器:简化代码,增强功能
在Python的世界里,装饰器是那些静悄悄的幕后英雄。它们不张扬,却能默默地为函数或类增添强大的功能。本文将带你了解装饰器的魅力所在,从基础概念到实际应用,我们一步步揭开装饰器的神秘面纱。准备好了吗?让我们开始这段简洁而富有启发性的旅程吧!
21 6
|
11天前
|
测试技术 Python
探索Python中的装饰器:简化代码,增强功能
在Python的世界中,装饰器是那些能够为我们的代码增添魔力的小精灵。它们不仅让代码看起来更加优雅,还能在不改变原有函数定义的情况下,增加额外的功能。本文将通过生动的例子和易于理解的语言,带你领略装饰器的奥秘,从基础概念到实际应用,一起开启Python装饰器的奇妙旅程。
29 11
|
7天前
|
测试技术 开发者 Python
探索Python中的装饰器:从入门到实践
装饰器,在Python中是一块强大的语法糖,它允许我们在不修改原函数代码的情况下增加额外的功能。本文将通过简单易懂的语言和实例,带你一步步了解装饰器的基本概念、使用方法以及如何自定义装饰器。我们还将探讨装饰器在实战中的应用,让你能够在实际编程中灵活运用这一技术。
24 7
|
7天前
|
Python
探索Python中的装饰器:简化代码,增强功能
在Python的世界里,装饰器就像是给函数穿上了一件神奇的外套,让它们拥有了超能力。本文将通过浅显易懂的语言和生动的比喻,带你了解装饰器的基本概念、使用方法以及它们如何让你的代码变得更加简洁高效。让我们一起揭开装饰器的神秘面纱,看看它是如何在不改变函数核心逻辑的情况下,为函数增添新功能的吧!
|
7天前
|
程序员 测试技术 数据安全/隐私保护
深入理解Python装饰器:提升代码重用与可读性
本文旨在为中高级Python开发者提供一份关于装饰器的深度解析。通过探讨装饰器的基本原理、类型以及在实际项目中的应用案例,帮助读者更好地理解并运用这一强大的语言特性。不同于常规摘要,本文将以一个实际的软件开发场景引入,逐步揭示装饰器如何优化代码结构,提高开发效率和代码质量。
31 6
|
7天前
|
存储 缓存 Python
Python中的装饰器深度解析与实践
在Python的世界里,装饰器如同一位神秘的魔法师,它拥有改变函数行为的能力。本文将揭开装饰器的神秘面纱,通过直观的代码示例,引导你理解其工作原理,并掌握如何在实际项目中灵活运用这一强大的工具。从基础到进阶,我们将一起探索装饰器的魅力所在。
|
7天前
|
测试技术 开发者 Python
深入理解Python装饰器:从基础到高级应用
本文旨在为读者提供一个全面的Python装饰器指南,从其基本概念讲起,逐步深入探讨其高级应用。我们将通过实例解析装饰器的工作原理,并展示如何利用它们来增强函数功能、控制程序流程以及实现代码的模块化。无论你是Python初学者还是经验丰富的开发者,本文都将为你提供宝贵的见解和实用的技巧,帮助你更好地掌握这一强大的语言特性。
21 4
|
9天前
|
开发者 Python
Python中的装饰器:从入门到实践
本文将深入探讨Python的装饰器,这一强大工具允许开发者在不修改现有函数代码的情况下增加额外的功能。我们将通过实例学习如何创建和应用装饰器,并探索它们背后的原理和高级用法。
24 5