《流畅的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