带有参数的装饰器

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行，调用原来的函数。

运行结果图：