当需求相似的函数需要使用装饰器时,这种差别不大的函数,如果定义多个相似的装饰器来各自装饰特定函数就太过赘余了。
【比如说A需要记录日志功能的装饰器,B需要记录日志+发送给指定管理员功能的装饰器,它们之间有重合的功能--记录日志】【如果相同代码量很大,那么新弄的代码重复量就更大了】
为了解决这种问题,我们可以使用装饰器的类型来解决,与之前的区别只是再加上一层嵌套而已:
def decorator(my_type): def outwrapper(func): def wrapper(*args,**kwargs): print("logged done")#假功能 if my_type=='A': res = func() return res else : print("send done")#假功能 res = func() return res return wrapper return outwrapper @decorator('A') def testA(): print("run in A") return @decorator('B') def testB(): print("run in B") return testA() print("-------------") testB()
相当于,当装饰器给定参数时,【装饰器原本默认有一个函数对象参数】,相当于会选择第二次再传入默认的函数对象参数
原本是:
test3=outwrapper(func) => test3=wrapper()
现在是:
test3=decorator(A) => outwrapper(func) => test3=wrapper()