看到廖雪峰python教程上,python装饰器一章
https://www.liaoxuefeng.com/wiki/0014316089557264a6b348958f449949df42a6d3a2e542c000/0014318435599930270c0381a3b44db991cd6d858064ac0000
在最后的作业题上
再思考一下能否写出一个@log
的decorator,使它既支持:
@log def f(): pass
又支持:
@log('execute') def f(): pass
即在装饰器头上加str,从而牵扯出了一个装饰器参数传入的问题。
讨论区有人留言
import functools import time def log2(text=None): def decorator(func): @functools.wraps(func) def wrapper(*args,**kw): if isinstance(text,(int,str)): print('%s begin call %s():' %(text,func.__name__)) func(*args,**kw) print('%s end call %s():' %(text,func.__name__)) else: print('begin call %s():' % func.__name__) func(*args,**kw) print('end call %s():' % func.__name__) return return wrapper return decorator if isinstance(text,(int,str)) else decorator(text) @log2 def now2(): print('now is:'+time.asctime()) now2() @log2('timeshow') def now3(): print('now is:'+'2017-07-10') now3()
最后的if语句其实我是参考其他人的,但是我突然觉得很不理解, 为什么if有参数的时候是 return decorator, 而else(无参数)的时候是 return decorator(text)
谁能给解释一下。。我怎么总觉得写反了呢?不过从结果来看是对的~
以及下面的回复
在调试里面跑了一下,不写参数的时候,text拿到的实际是函数对象,就是本来func应该拿到的那个,结果让text提前吃掉了,截胡了你程序里的now。
这里不写字符串的话,text不会拿到None,而是直接被赋值成函数。调试里面跑到这一步,text的属性是指向一个function。
如果把return里面那个decorator(text) 直接改成decorator,执行会提示缺参数。因为内层的func这个量已经没参数可接了。
decorator(text) 的意思就是,如果发现text不是str类型,那么就是本该传给内层func的函数让外层提前拿走了,并且放在了text里面。这时应该手动把text传给里面那层,于是就return了 decorator(text)
通过这个问题,大概搞明白装饰器参数传入的问题了。
python应该会从外层向内层逐层匹配。