问题:
想要去编写一个可接受参数的装饰器函数
例子:
假设我们想要去定义一个打印log的装饰器,且用户可以设置log等级以及其他的一些细节作为参数:
from functools import wraps
import logging
def logged(level,name=None,message=None):
def decorate(func):
logname =name if name else func.__module__
log = logging.getLogger(logname)
logmsg = message if message else func.__name__
@wraps(func)
def wrapper(*args,**kwargs):
log.log(level,logmsg)
return func(*args,**kwargs)
return wrapper
return decorate
@logged(logging.CRITICAL,name='example',message='hello')
def spam():
print('spam is spam')
spam()
'''
#结果为:
hello
spam is spam
'''
有些人对于装饰器不理解怎么调用内层函数的,实际上上述装饰器就是如下使用方法,可以参考,帮助理解:
from functools import wraps
import logging
def logged(level,name=None,message=None):
def decorate(func):
logname =name if name else func.__module__
log = logging.getLogger(logname)
logmsg = message if message else func.__name__
@wraps(func)
def wrapper(*args,**kwargs):
log.log(level,logmsg)
return func(*args,**kwargs)
return wrapper
return decorate
def spam():
print('spam is spam')
spam1 = logged(logging.CRITICAL,name='example',message='hello')(spam)
spam1()#运行后和上述例子结果相同
#解释下:
#logged(logging.CRITICAL,name='example',message='hello')这里会返回decorate的引用,
#之后加上()(即:logged(logging.CRITICAL,name='example',message='hello')())相当于调用decorate(),
#而同时需要传递参数,此时便传入spam引用,这个函数(decorate)返回wrapper的引用,
#所以这里spam1实际上是wrapper引用的别名,最后加括号(spam1())即调用了wrapper。
#而wrapper()结尾时会调用这个传入的spam()函数,所以会得到spam的执行结果,并且将其返回值(None)返回,尽管spam没有返回值。
#以上便是装饰器理解的核心,实际上可以参考其他语言关于回调函数的概念,这里便能很快理解了。