带有参数的装饰器

2022-09-18

带有参数的装饰器的实质:

  因为装饰器是不能在带有参数,所以要定义带有参数的装饰器应该换一种思路。在装饰器的外面套一个函数,使用该函数返回这个装饰器。

带有参数的装饰器的实例:

 1   # 装饰器  2   def return_decorator(flag):  3   def decorator(func):  4   def inner(a,b):  5               if  flag ==  "  +  "  :  6                  print( "  正在努力执行加法运算  "  )  7              elif flag ==  "  -  "  :  8                  print( "  正在努力执行减法运算  "  )  9   func(a,b)  10           return  inner  11       return  decorator  12  
 13  @return_decorator( "  +  "  )  14   # 原来的函数  15   def add_num(a,b):  16      result = a + b  17   print(result)  18  
 19  @return_decorator( "  -  "  )  20   def sub_num(a,b):  21      result = a - b  22   print(result)  23  
 24  add_num( 1 , 2  )  25  sub_num( 1 , 4 )

  说明:

    第2行,定义了一个外部函数,内部放置装饰器。这个外部函数是有参数的。第5-8行,在inner函数内部,进行判断,如果传入的参数“flag”为“+”,输出xx;为“-”,输出xx。

    第9行,是func(a,b)参数,如果此处写为了add_num(a,b)或者sub_num(a,b),会出现“栈内存溢出”的错误。

    第11行,返回装饰器。

    第13行,给带参数的装饰器传一个flag参数“+”。第19行,传“-”参数。

    第24-25行,调用原来的函数。

运行结果图:

 

 

    

    

标签: python

添加新评论