码迷,mamicode.com
首页 > 编程语言 > 详细

python wraps装饰器

时间:2015-08-26 15:22:45      阅读:165      评论:0      收藏:0      [点我收藏+]

标签:

这是一个很有用的装饰器。看过前一篇反射的朋友应该知道,函数是有几个特殊属性比如函数名,在被装饰后,上例中的函数名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)。

 # -*- coding=utf-8 -*- 

from functools import wraps   

def my_decorator(func):

    @wraps(func)

    def wrapper(*args, **kwargs):

        print(Calling decorated function...)

        return func(*args, **kwargs)

    return wrapper  

@my_decorator
def example(): """Docstring""" print(Called example function) print(example.__name__, example.__doc__)

打印出:example Docstring

python wraps装饰器

标签:

原文地址:http://www.cnblogs.com/yasmi/p/4760278.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!