Python装饰器

一、装饰器介绍

1.1 为何要用装饰器

软件的设计应该遵循开放封闭原则，即对扩展是开放的，而对修改是封闭的：

• 对扩展开放，意味着有新的需求或变化时，可以对现有代码进行扩展，以适应新的情况。

• 对修改封闭，意味着对象一旦设计完成，就可以独立完成其工作，而不要对其进行修改。

软件包含的所有功能的源代码以及调用方式，都应该避免修改，否则一旦改错，则极有可能产生连锁反应，最终导致程序崩溃。而对于上线后的软件，新需求或者变化又层出不穷，我们必须为程序提供扩展的可能性，这就用到了装饰器。

1.2 什么是装饰器

‘装饰’代指为被装饰对象添加新的功能，‘器’代指器具/工具，装饰器与被装饰的对象均可以是任意可调用对象。

概括的讲，装饰器的作用就是在不修改被装饰对象源代码和调用方式的前提下为被装饰对象添加额外的功能。

装饰器经常用于有切面需求的场景，比如：插入日志、性能测试、事务处理、缓存、权限校验等应用场景，装饰器是解决这类问题的绝佳设计。有了装饰器，就可以抽离出大量与函数功能本身无关的雷同代码并继续重用。

提示：可调用对象有函数，方法或者类，此处我们单以本篇主题：函数为例，来介绍函数装饰器，并且被装饰的对象也是函数。

二、装饰器的实现

函数装饰器分为：无参装饰器和有参装饰器两种，二者的实现原理一样，都是‘函数嵌套+闭包+函数对象’的组合使用的产物。

2.1 无参装饰器的实现

如果想为下述函数添加统计其执行时间的功能：

import time
 
def index():
    time.sleep(3)
    print('Welcome to the index page')
    return 200
 
index()  # 函数执行

遵循不修改被装饰对象源代码的原则，我们想到的解决方法可能是这样：

start_time = time.time()
index()  # 函数执行
stop_time = time.time()
print('run time is %s' % (stop_time - start_time))

考虑到还有可能要统计其他函数的执行时间，于是我们将其做成一个单独的工具，函数体需要外部传入被装饰的函数从而进行调用，我们可以使用参数的形式传入：

def wrapper(func):  # 通过参数接收外部的值
    start_time = time.time()
    res = func()
    stop_time = time.time()
    print('run time is %s' % (stop_time - start_time))
    return res

但之后函数的调用方式都需要统一改成：

wrapper(index)
wrapper(其他函数)

这便违反了不能修改被装饰对象调用方式的原则，于是我们需要换一种为函数体传值的方式，那之前所学的闭包函数在这里就能派上用场了，修改如下：

def timer(func):
    def wrapper():  # 引用外部作用域的变量func
        start_time = time.time()
        res = func()
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper

这样我们便可以在不修改被装饰函数源代码和调用方式的前提下为其加上统计时间的功能，只不过需要事先执行一次timer将被装饰的函数传入，返回一个闭包函数wrapper重新赋值给变量名/函数名index，如下：

index = timer(index) 
index() 
"""
第1行得到index = wrapper，wrapper携带对外作用域的引用：func = 原始的index
第2行执行的是wrapper()，在wrapper的函数体内再执行最原始的index
"""

至此我们便实现了一个无参装饰器timer，可以在不修改被装饰对象index源代码和调用方式的前提下为其加上新功能。但我们忽略了若被装饰的函数是一个有参函数，便会抛出异常：

def home(name):
    time.sleep(5)
    print('Welcome to the home page', name)
 
home = timer(home)
home('jason')
# 抛出异常
TypeError: wrapper() takes 0 positional arguments but 1 was given 

之所以会抛出异常，是因为home(‘jason’)调用的其实是wrapper(‘jason’)，而函数wrapper没有参数。wrapper函数接收的参数其实是给最原始的func用的，为了能满足被装饰函数参数的所有情况，便用上*args和**kwargs组合 ，于是修正后的装饰器timer如下：

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper

此时我们就可以用timer来装饰带参数或不带参数的函数了，但是为了简洁而优雅地使用装饰器，Python提供了专门的装饰器语法糖来取代index = timer(index)的形式，需要在被装饰对象的正上方单独一行添加@timer（语法糖必须紧贴在被装饰对象的上方）；

当解释器解释到@timer时就会调用timer函数，且把它正下方的函数名当做实参传入，然后将返回的结果重新赋值给原函数名：

@timer    """index = timer(index)"""
def index():
    time.sleep(3)
    print('Welcome to the index page')
    return 200

@timer    """home = timer(home)"""
def home(name):
    time.sleep(5)
    print('Welcome to the home page', name)

装饰器语法糖内部原理：会自动将下面紧贴着的被装饰对象名字当做参数传给装饰器调用。

如果我们有多个装饰器，可以叠加多个：

@deco3
@deco2
@deco1
def index():
    pass

叠加多个装饰器也无特殊之处，上述代码语义如下：

index = deco3(deco2(deco1(index)))

2.2 有参装饰器的实现

了解无参装饰器的实现原理后，我们可以再实现一个用来为被装饰对象添加认证功能的装饰器，实现的基本形式如下：

def deco(func):
    def wrapper(*args, **kwargs):
        编写基于文件的认证,认证通过则执行res = func(*args,**kwargs),并返回res
    return wrapper

如果我们想提供多种不同的认证方式以供选择，单从wrapper函数的实现角度改写如下：

def deco(func):
	def wrapper(*args, **kwargs):
        if driver == 'file':
            编写基于文件的认证,认证通过则执行res = func(*args, **kwargs),并返回res
        elif driver == 'mysql':
            编写基于mysql认证,认证通过则执行res = func(*args, **kwargs),并返回res
	return wrapper

函数wrapper需要一个driver参数，而函数deco与wrapper的参数都有其特定的功能，不能用来接收其他类别的参数，既然需要一个参数，那我们可以再在deco的外部再包一层函数auth，用来专门接收额外的参数，这样便保证了在auth函数内无论多少层都可以引用到：

def auth(driver):
    def deco(func):
        ……
    return deco

此时我们就实现了一个有参装饰器，使用方式如下：

"""先调用auth(driver='file')，得到@deco，deco是一个闭包函数，包含了对外部作用域名字driver的引用，@deco的语法意义与无参装饰器一样"""
@auth(driver='file')
def index():
    pass
 
"""再调用auth(driver='mysql')，后续同上"""    
@auth(driver='mysql')
def home():
    pass

可以使用help(函数名)来查看函数的文档注释，本质就是查看函数的doc属性，但对于被装饰之后的函数，查看文档注释：

@timer
def home(name):
    '''
    home page function
    :param name: str
    :return: None
    '''
    time.sleep(5)
    print('Welcome to the home page', name)
    
print(help(home))
"""
打印结果：
 
Help on function wrapper in module __main__:
 
wrapper(*args, **kwargs)
 
None
"""

在被装饰之后home = wrapper，查看home.name也可以发现home的函数名其实是wrapper，如果想要保留原函数的文档和函数名属性，就需要修正装饰器：

def timer(func):
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    wrapper.__doc__ = func.__doc__
    wrapper.__name__ = func.__name__
    return wrapper

可以看到，按照上述方式来实现保留原函数属性过于麻烦，functools模块下提供一个装饰器wraps专门用来帮我们实现这件事，用法如下：

from functools import wraps
 
def timer(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        start_time = time.time()
        res = func(*args, **kwargs)
        stop_time = time.time()
        print('run time is %s' % (stop_time - start_time))
        return res
    return wrapper