带有参数的装饰器

2022-09-18

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

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

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

# 装饰器
def return_decorator(flag):
    def decorator(func):
        def inner(a,b):
            if flag == "+":
                print("正在努力执行加法运算")
            elif flag == "-":
                print("正在努力执行减法运算")
            func(a,b)
        return inner
    return decorator

@return_decorator("+")
# 原来的函数
def add_num(a,b):
    result = a + b
    print(result)

@return_decorator("-")
def sub_num(a,b):
    result = a - b
    print(result)

add_num(1,2)
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行，调用原来的函数。

运行结果图：