首页 > 代码库 > Python札记 -- 装饰器补充
Python札记 -- 装饰器补充
本随笔是对Python札记 -- 装饰器的一些补充。
使用装饰器的时候,被装饰函数的一些属性会丢失,比如如下代码:
1 #!/usr/bin/env python 2 3 def deco(func): 4 def wrapper(): 5 print "Wrap start" 6 func() 7 print "Wrap end\n" 8 return wrapper 9 10 @deco11 def foo():12 """Docstring for foo"""13 print "In foo():"14 15 foo()16 print foo.__name__17 print foo.__doc__
输出如下:
$ python decorator_test.pyWrap startIn foo():Wrap endwrapperNone
可以发现,__name__属性成了wrapper,而__doc__属性则成了None。这对于平常使用多少带来些不便,总不能给每个使用装饰器的函数都重写__name__和__doc__吧。
Python的functools提供的update_wrapper和wraps可以有效解决这个问题。不过update_wrapper是用方法的形式进行调用,而wraps则是用装饰器来封装了update_wrapper。示例代码分别如下:
1 #!/usr/bin/env python 2 from functools import update_wrapper 3 4 def deco(func): 5 def wrapper(): 6 print "Wrap start" 7 func() 8 print "Wrap end\n" 9 return update_wrapper(wrapper,func) #调用update_wrapper方法10 11 @deco12 def foo():13 """Docstring for foo"""14 print "In foo():"15 16 foo()17 print foo.__name__18 print foo.__doc__
1 #!/usr/bin/env python 2 from functools import wraps 3 4 def deco(func): 5 @wraps(func) #使用装饰器来实现 6 def wrapper(): 7 print "Wrap start" 8 func() 9 print "Wrap end\n"10 return wrapper11 12 @deco13 def foo():14 """Docstring for foo"""15 print "In foo():"16 17 foo()18 print foo.__name__19 print foo.__doc__
现在被deco装饰过的foo方法,可以保留之前的__name__和__doc__属性了。
$ python decorator_test_with_update_wrapper.pyWrap startIn foo():Wrap endfooDocstring for foo$ python decorator_test_with_wraps.pyWrap startIn foo():Wrap endfooDocstring for foo
Python札记 -- 装饰器补充
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。