python装饰器，为了不修改源函数前提下，扩展函数功能

Posted 2022-04-04 飞凡可期

def add(a,b):
    #print("开始计算...")
    return a+b

def pt(s):
    print("hello bug! ",s)

def begin_(old):#输入是函数参数，函数指针

    #创建一个新函数
    def new_func(*args,**kwargs): #装饰套路，传入出函数，内层用通用形参
        #扩展了函数功能，定制需求（装饰代码）
        print("开始计算...")
        r = old(*args,**kwargs) #两个*不一样？，一个是拆包？参数值，成对的key-val参数
        print("结果是",r)
        return r

    return new_func #输出也是函数参数，函数指针

f1 = begin_(add)
f2 = begin_(add)
print(f1,'\\n',f2)
print("add rst", f1(5,10))

f3 = begin_(pt)
f3("who !")



# 新麻烦，参数不是定制的，反复修改，变量参数，kwargs参数

#这就是装饰器，可以不修改原来函数来扩展新函数
# 我们可以用 @函数名，来用装饰器来装饰函数；
# 同时为一个函数可添加多个装饰器，函数按照从内到外装饰。近水楼台先执行
#手动定义不多，直接调用就行。。 装饰器好处，不修改源函数，而增强了它功能
def explain_(old):#输入是函数参数，函数指针

    #创建一个新函数
    def new_func(*args,**kwargs): #装饰套路，传入出函数，内层用通用形参
        #扩展了函数功能，定制需求（装饰代码）
        print("我们一起做运算...")
        r = old(*args,**kwargs) #两个*不一样？，一个是拆包？参数值，成对的key-val参数
        print("计算结束")
        return r

    return new_func #输出也是函数参数，函数指针

@explain_
@begin_
def minus(a,b):
    return a-b

minus(10,5)

minus(100,200)

summary

总结为，通过函数名称def之前，加上 @装饰器来扩展函数功能。装饰器是一种自动生成函数的函数（包装别的函数的自动函数），将函数作为输入，输出仍然是一个函数，只是之前、之后，输出结果经过了装饰器的额外扩展。 @decirator \\def func(x): func, 调用仍然是一样的func(a)，但是注意经过装饰器修饰的是新函数，函数地址和原本func()不是一个。

多个装饰器，内层先生效，先装饰的后生效。。按照栈规则，先入后出FILO。