编程语言
首页 > 编程语言> > 《流畅的python》学习日志----装饰器工厂函数

《流畅的python》学习日志----装饰器工厂函数

作者:互联网

使用装饰器工厂函数实现不同的装饰器行为


想要一个装饰器对不同的函数表现出不同的行为,可以使用装饰器工厂函数,例如:

def decorate(type):
    def calc(func):
        if type == 'adam':
            print('use adam')
        elif type == 'rmsprop':
            print('use rmsprop')
        else:
            print('None')
        return func
    return calc

在这里函数decorate作为一个装饰器工厂函数,它依靠不同的传入参数,实现不同的装饰功能,calc是真正的装饰器,由decorate调用。可以这么使用:

@decorate('adam')
def use_adam_function():
    pass

@decorate('rmsprop')
def use_rms_prop_function():
    pass

@decorate('None')
def use_None():
    pass

工厂函数decorate接受字符串输入,在calc中对字符串进行检查,从而选择不同的装饰效果,而在函数本体中都是使用pass跳过,运行结果如下:

use adam
use rmsprop
None

标签:use,流畅,函数,python,decorate,日志,装饰,def,adam
来源: https://blog.csdn.net/scrence/article/details/100824383