装饰器

一个普通的装饰器一般是这样：

import functoolsdef log(func):    @functools.wraps(func)    def wrapper(*args, **kwargs):        print('call %s():' % func.__name__)        print('args = {}'.format(*args))        return func(*args, **kwargs)    return wrapper

这样就定义了一个打印出方法名及其参数的装饰器。

调用之：

@logdef test(p):    print(test.__name__ + " param: " + p)test("I'm a param")

输出：

call test():args = I'm a paramtest param: I'm a param

装饰器在使用时，用了@语法，让人有些困扰。其实，装饰器只是个方法，与下面的调用方式没有区别：

def test(p):    print(test.__name__ + " param: " + p)wrapper = log(test)wrapper("I'm a param")

@语法只是将函数传入装饰器函数，并无神奇之处。

值得注意的是@functools.wraps(func)，这是python提供的装饰器。它能把原函数的元信息拷贝到装饰器里面的 func 函数中。函数的元信息包括docstring、name、参数列表等等。可以尝试去除@functools.wraps(func)，你会发现test.__name__的输出变成了wrapper。

带参数的装饰器

装饰器允许传入参数，一个携带了参数的装饰器将有三层函数，如下所示：

import functoolsdef log_with_param(text):    def decorator(func):        @functools.wraps(func)        def wrapper(*args, **kwargs):            print('call %s():' % func.__name__)            print('args = {}'.format(*args))            print('log_param = {}'.format(text))            return func(*args, **kwargs)        return wrapper    return decorator    @log_with_param("param")def test_with_param(p):    print(test_with_param.__name__)

看到这个代码是不是又有些疑问，内层的decorator函数的参数func是怎么传进去的？和上面一般的装饰器不大一样啊。

其实道理是一样的，将其@语法去除，恢复函数调用的形式一看就明白了：

# 传入装饰器的参数，并接收返回的decorator函数decorator = log_with_param("param")# 传入test_with_param函数wrapper = decorator(test_with_param)# 调用装饰器函数wrapper("I'm a param")

输出结果与正常使用装饰器相同：

call test_with_param():args = I'm a paramlog_param = paramtest_with_param

至此，装饰器这个有点费解的特性也没什么神秘了。

装饰器这一语法体现了Python中函数是第一公民，函数是对象、是变量，可以作为参数、可以是返回值，非常的灵活与强大。

原文链接：

转载于:https://www.cnblogs.com/HDK2016/p/10973193.html

你可能感兴趣的文章