def log(func):
def wrapper(*args, **kwargs):
print "call %s()" % func.__name__
return func(*args, **kwargs)
return wrapper
@log
def now():
print '2015-4-10'
now() # print 'call now()'
其实装饰器修饰函数相当于, now = log(now) 也就是装饰器函数把被修饰的函数当参数后赋给同名的变量
functools.wraps 函数
当我们使用了装饰器后now的__name__值发生了改变
# 没有使用前
now.__name__ # print 'now'
# 使用后
now.__name__ # print 'wrapper'
当我们使用装饰器前,now.__name__使用的是当前now函数,但使用后 now这个函数其实是 log(now) 也就是log函数的返回值也就是被包裹的wrapper. 解决方法是functools.wraps函数.
装饰闭包,使用前得调用 import functools
def log(func):
@functools.wraps(func)
def wrapper(*args, **kwargs):
...
带参数的装饰器
如果decorator需要传入参数, 那就需要在写一个返回decorator的高阶函数. 写出来更复杂.
def login(level):
def _deco(func):
def wrapper(*args, **kwargs):
if level >= 5:
print '用户 VIP 等级 %d' % int(level-5)
else:
print '用户 屌丝 等级 %d' % abs(level-5)
return func(*args, **kwargs)
return wrapper
return _deco
@login(5)
def user(username):
print 'welcome, %s' % username
# 用户vip 等级0
# welcome, mink
user('mink')
带参数的decorator等于func = 装饰器函数(装饰器参数)(func)
装饰器类
通过类的__call__可以想使用函数一样使用类
class A(object):
def __init__(self, func):
self.func = func
def __call__(self):
return self.func() ** 2
@A
def foo():
return 10
print foo() # print 100
本条技术文章来源于互联网,如果无意侵犯您的权益请点击此处反馈版权投诉