装饰器

装饰器

由于函数也是一个对象,而且函数对象可以被赋值给变量,所以通过变量也能调用该函数.

>>>def now()
...		print('2018')
>>> f = now
>>> f()
2018

# 函数对象有一个__name__属性, 可以拿到函数名字:
>>> now.__name__
'now'
>>> f.__name__
'now'

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

本质上,decorator就是返回函数的高阶函数. 所以,

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

@log
def now():
    print('2018') 

# 将@log放到now()函数的定义处,相当于执行了语句:now = log(now)
    
>>> now()
call now():
    2018

由于log()是一个装饰器, 返回一个函数, 所以原来的now()函数仍然存在, 只是现在同名的now变量指向了新的函数,于是调用now()将执行新函数, 即在log()函数中返回的wrapper()函数.

wrapper()函数的参数定义是(*args, **kwargs), 所以wrapper()函数可以接受任意参数的调用.