Python---装饰器

lovegigi

# 装饰器　　
# 由于函数也是一个对象，而且函数对象可以被赋值给变量，所以，通过变量也能调用该函数
　　

　　

　　
def now():
　　
    print('2017-08-07')
　　

　　
f = now
　　
f()
　　

　　
# 函数对象有一个__name__属性，可以拿到函数的名字
　　

　　
print(now.__name__)
　　
print(f.__name__)
　　

　　
# 假设我们要增强now()函数的功能
　　
# 比如，在函数调用前后自动打印日志，但又不希望修改now()函数的定义，这种代码运行期间动态增加功能的方式，称之为"装饰器"（Decorator）
　　

　　
# 本质上，decorator就是一个返回函数的高阶函数
　　
# 定义一个能打印日志的decorator
　　

　　

　　
def log(func):
　　
    def wrapper(*args, **kw):
　　
        print('call %s():' % func.__name__)
　　
        return func(*args, **kw)
　　
    return wrapper
　　

　　
# decorator接受一个函数作为参数，并返回一个函数
　　
# 我们要借助Python的语法，把decorator置于函数的定义处
　　

　　

　　
@log
　　
def now():
　　
    print('2017-08-07')
　　

　　
now()
　　

　　
# 把@log放到now()函数的定义处，相当于执行了语句 now = log(now)
　　
# 由于log()是一个decorator，返回一个函数，所以，原来的now()函数依然存在，只是现在同名的now变量指向了新的函数
　　
# 于是调用now()将执行新函数，即在log()函数中返回的wrapper()函数
　　

　　
# wrapper()函数的参数定义是(*args, **kw)，因此，wrapper()函数可以接受任意参数的调用
　　
# 在wrapper()函数内，首先打印日志，再紧接着调用原始函数
　　

　　
# 如果decorator本身需要传入参数，那就需要编写一个返回decorator的高阶函数，写出来会更复杂
　　
# 自定义log的文本
　　

　　

　　
def log(text):
　　
    def decorator(func):
　　
        def wrapper(*args, **kw):
　　
            print('%s %s():' % (text, func.__name__))
　　
            return func(*args, **kw)
　　
        return wrapper
　　
    return decorator
　　

　　

　　
@log('execute')
　　
def now2():
　　
    print('2017-08-07')
　　

　　
now2()
　　

　　
# 和两层嵌套的decorator相比，3层嵌套的效果是  now = log('execute')(now)
　　
# 上面的语句，首先执行log('execute')，返回的是decorator函数，再调用返回的函数，参数是now函数，返回值最终是wrapper函数