python 装饰器第七步：带有参数的装饰器

转载于 2017-12-10 23:09:00 发布 · 51 阅读

0 ·

CC 4.0 BY-SA版权

原文链接：http://www.cnblogs.com/zhangboblogs/p/8018695.html

文章标签：

#python

本文通过实例详细介绍了如何使用带有参数的装饰器来扩展函数的功能。装饰器可以根据传入的不同参数来选择不同的增强逻辑，实现了代码的灵活复用。

#第七步：带有参数的装饰器
#两个基本函数用同一个装饰器装饰
def outer(arg):
    print(arg)
    #这是装饰器的代码
    def kuozhan(func):
        print(func) #func接收的依然是eat和chifan
        #未来的eat函数
        def neweat():
            # 扩展功能1
            print('饭前洗洗手')
            # 调用基本函数
            func()
            # 扩展功能2
            print('饭后溜溜腿')
        def newdrink():
            # 扩展功能1
            print('酒前真君子')
            # 调用基本函数
            func()
            # 扩展功能2
            print('酒后耍流氓')
        if arg == 'la':
            return neweat
        elif arg == 'chi':
            return newdrink
    #返回装饰器
    return kuozhan
#基本函数1
result = outer('la')
@result #@装饰器函数
def eat():
    print('吃饭')
#基本函数2
@outer('chi')
def drink():
    print('喝酒')

#调用基本函数
eat()
# print(drink)
# drink()