def decorator(func): def inner(info): print('inner') func(info) return inner @decorator def show_info(info): print(info) show_info('hello')
装饰器在装饰函数的时候由于返回的是inner的函数地址,所以函数的名称也会改变 show_info.__name__会变成inner,防止这种现象可以使用functools
import functools def decorator(func): @functools.wraps(func) def inner(info): print('inner') func(info) return inner @decorator def show_info(info): print(info) show_info('hello')
这样写就不会改变被装饰函数的名称
此方法在Flask框架的app.Route()的源码中体现
class Commands(object): def __init__(self): self.cmd = {} def regist_cmd(self, name: str) -> None: def decorator(func): self.cmd[name] = func print('func:',func) return func return decorator commands = Commands() # 使得s1的值指向show_h的函数地址 @commands.regist_cmd('s1') def show_h(): print('show_h') # 使得s2的值指向show_e的函数地址 @commands.regist_cmd('s2') def show_e(): print('show_e') func = commands.cmd['s1'] func()
在阅读装饰器代码时可以使用加(func_name)的方式
以为例
@commands.regist_cmd('s2') def show_e(): print('show_e')
即 show_e = commands.regist_cmd('s2')(show_e)