Python @修饰符示例理解

Posted 2023-01-24

python 中当某个函数已经存在了，想在函数前执行相应的操作，但又不希望改掉原函数的定义，利用修饰符（Decorator）可实现该功能。

一个基本的函数，可以直接执行调用，或者赋予变量再调用：

class TEST:
  def func(self):
    print("func")
  
if __name__ == "__main__":
  t = TEST()
  t.func() #print
  
  f = t.func #函数赋予给变量
  f() #print

执行结果：

func
func

现在增加一个函数 funa，每次执行原函数 func 时，都会通过 funa 执行。再增加一个函数 fund ，看看修饰符调用的是什么函数。

class TEST:
  def funa(fu):
    def wrapper(*args, **kw): #适用于任意变量
      print(call %s(): % fu.__name__)
      return fu(*args, **kw)
    return wrapper
  
  @funa  #修饰符类似执行：func = funa(func)
  def func(self):
    print("func")
  
  def fund(self,fn):
    print(fun name : %s() % fn.__name__)
    
if __name__ == "__main__":
  t = TEST()
  t.func()
  
  t.fund(t.func) #查看函数名

执行结果：

call func():
func
fun name : wrapper()

再来个更简洁的事例：

class TEST:
  def funa(fn):
    def testa(arg):
      print("funa")
      fn(arg)
    return testa
  
  def funb(fn):
    def testb(arg):
      print("funb")
      fn(arg)
    return testb
    
  @funa
  @funb
  def func(self): #func = funa(funb(func))
    print("func")
  
  @funb 
  def fund(self): #func = funb(fund)
    print("fund")
    
if __name__ == "__main__":
  t = TEST()
  t.func()
  t.fund()

执行结果：

funa
funb
func
funb
fund

更多参考：

​​廖雪峰的官方网站-python-装饰器​​

​​官方文档 PEP 318 -- Decorators for Functions and Methods​​