参考:
https://blog.csdn.net/u014665013/article/details/86064025
- 原函数无参数
def debug(func):
def wrapper():
print "[DEBUG]: enter {}()".format(func.__name__)
return func()
return wrapper
@debug
def say_hello():
print "hello!"
装饰器参数为原函数,返回值为内部定义的函数
- 原函数有参数
def debug(func):
def wrapper(*args, **kwargs): # 指定宇宙无敌参数
print "[DEBUG]: enter {}()".format(func.__name__)
print 'Prepare and say...',
return func(*args, **kwargs)
return wrapper # 返回
@debug
def say(something):
print "hello {}!".format(something)
装饰器参数为原函数,但内部定义的函数包含可变参数*args和关键字参数**kwargs,返回值为原函数(带参数)
- 装饰器有参数
def logging(level):
def wrapper(func):
def inner_wrapper(*args, **kwargs):
print "[{level}]: enter function {func}()".format(
level=level,
func=func.__name__)
return func(*args, **kwargs)
return inner_wrapper
return wrapper
@logging(level='INFO')
def say(something):
print "say {}!".format(something)
此时装饰器内部有两个函数嵌套,外层为A函数,内层为B函数,装饰器参数为自已要输入的参数,A函数参数为原函数,B函数参数为原函数所需的参数。
即 装饰器参数,原函数,原函数参数的位置是由外到内的,这三个里面有几个就需要几层函数。
装饰器除了基于函数实现以外,还可以基于类实现,因为装饰器实质是接受一个可调用对象,并返回一个可调用对象,在类中通过重载_call_方法,使类对象变为可调用对象,完成装饰器功能。