理解python装饰器的格式依据,需要会python元编程的内容。
不带参数的函数装饰器
def wrapper(fn):
def inner(*args,**kwargs):
return fn(*args,**kwargs)
return inner
@wrapper()
def run()
pass
理解
run = wrapper(run)
wrapper(run)== inner
调用run()等价于== inner()
返回值是调用 fun(),就是fun函数的返回值。
run(*args,**kwargs) == wrapper(run)(*args,**kwargs) == inner(*args,**kwargs)
带参数的函数装饰器
def log(text):
def decorator(func):
def wrapper(*args, **kwargs):
print('%s %s():' % (text, func.__name__))
return func(*args, **kwargs)
return wrapper
return decorator
@log('execute')
def now():
print('2015-3-25')
等价于下面
now = log(text)(func)
now() = log(text)(func)(*args,**kwargs) = wrapper(*args, **kwargs)
带参数函数装饰器:
捕获异常,并且执行重试的带参数的函数装饰器:
def retry(cout=3,wait=0,exceptions=(Exception,)):
def wrapper(f):
def inner(*args,**kwargs):
res = f(*args, **kwargs)
return res
return inner
return wrapper
def f(*args, **kwargs)
pass
类装饰器
# 类装饰器
类装饰器
类装饰器:使用类实现装饰器,装饰函数,增加函数功能
魔术方法:__call__()
功能:使用类装饰器装饰的函数调用时,默认调用类的这个函数
原理: 把类的对象,想函数一样调用时,(在对象名后面加一对括号),默认调用__call__()方法
装饰器原理:就是把函数,看成一个类,将被装饰函数作为一个参数传给类,在类实例化对象时。
@sum
def func():
pass
func = sum(func)
func() = sum(func)()=sum1()
class sum()
pass
sum1 = sum(func)
demo 1: 需求:在函数之前打印一串字符串
class AddPrint(object):
def __init__(self,f):
self.f = f
def __call__(self,*args,**kwargs):
print("-------------------")
# 如果类中没有调用原函数的功能,那么执行函数时就不会实现原函数的功能,只会执行__call__()函数下面对语句
res = self.f(*args,**kwargs)
@AddPrint
def func(name,age):
print("------%s------%s---"%(name,age))
# 等价于:AddPrint(func)
func()
# 调用执行等价于 AddPrint(func)(),类的对象调用__call__()函数
demo 2:需求:执行函数时,函数会被执行3次
class Count(object):
def __init__(self,count):
self.count = count
def __call__(self,f):
def inner(*args,**kwargs):
for i in range(self.count):
f(*args,**kwargs)
return inner
@count(5)
def func(name,age):
print("-------%s-----------%age-----------"%(name,age))
func('shine',19)