Python装饰器介绍

1、问题提出

希望在不修改原函数的情况下，对函数进行扩展。

2、问题解决

• 引入装饰器

# 创建几个函数
def add(a , b):
    '''
        求任意两个数的和
    '''
    r = a + b
    return r

def mul(a , b):
    '''
        求任意两个数的积
    '''
    r = a * b
    return r    

# 希望函数可以在计算前，打印开始计算，计算结束后打印计算完毕
# 我们可以直接通过修改函数中的代码来完成这个需求，但是会产生以下一些问题
#   ① 如果要修改的函数过多，修改起来会比较麻烦
#   ② 并且不方便后期的维护
#   ③ 并且这样做会违反开闭原则（OCP）
# OCP：程序的设计，要求开放对程序的扩展，要关闭对程序的修改

r = add(123,456)
print(r)

在不修改原函数的情况下，来对函数进行扩展。只需要根据现有的函数，来创建一个新的函数

def fn():
    print('我是fn函数....')

def fn2():
    print('函数开始执行~~~')
    fn()
    print('函数执行结束~~~')

fn2()

此时就达到了对fn函数的一个扩展

对第一个加法示例进行扩展：

def new_add(a,b):
    print('计算开始~~~')
    r = add(a,b)
    print('计算结束~~~')
    return r

r = new_add(111,222)    
print(r)

上述方式，完成在不修改源代码的情况下对函数进行扩展。

但是，这种方式要求每扩展一个函数就要手动创建一个新的函数，实在是太麻烦了。

为了解决这个问题，我们创建一个函数，让这个函数可以自动的帮助我们生产函数。

• 使用装饰器

每扩展一个函数就可以手动创建一个新的函数，但是这个方式实在是太麻烦了。为解决这个问题，我们可以创建一个函数，让这个函数可以自动的帮助我们生产函数。

接下来就介绍一下这个自动生成函数的函数怎么写：

因为我们在扩展的时候，不同的函数可能携带的参数不同，类型不同或者数量不同导致每次都要修改新函数的类型，所以这里介绍一下*args 这种通用的用法解决该问题。

def begin_end(old):
    '''
用来对其他函数进行扩展，使其他函数可以在执行前打印开始执行，执行后打印执行结束

        参数：
            old 要扩展的函数对象
    '''
    # 创建一个新函数
    def new_function(*args , **kwargs):
        print('开始执行~~~~')
        # 调用被扩展的函数
        result = old(*args , **kwargs)
        print('执行结束~~~~')
        # 返回函数的执行结果
        return result

    # 返回新函数        
    return new_function

f = begin_end(fn)
f2 = begin_end(add)
f3 = begin_end(mul)

# r = f()
# r = f2(123,456)
r = f3(123,456)
print(r)

像begin_end()这种函数我们就称它为装饰器（旧函数作为一个参数传进去，返回一个新函数）。

通过装饰器，可以在不修改原来函数的情况下来对函数进行扩展。（在开发中，我们都是通过装饰器来扩展函数功能的。）

• 装饰器的典型用法

def fn3(old):
    '''
        用来对其他函数进行扩展，使其他函数可以在执行前打印开始执行，执行后打印执行结束

        参数：
            old 要扩展的函数对象
    '''
    # 创建一个新函数
    def new_function(*args , **kwargs):
        print('fn3装饰~开始执行~~~~')
        # 调用被扩展的函数
        result = old(*args , **kwargs)
        print('fn3装饰~执行结束~~~~')
        # 返回函数的执行结果
        return result

    # 返回新函数        
    return new_function

@fn3
@begin_end
def say_hello():
    print('大家好~~~')

say_hello()

此处先装饰了离函数最近的begin_end，又在外部装饰了fn3装饰器。如果调换了位置，装饰顺序就不同了。

总结：在定义函数时，可以通过@装饰器，来使用指定的装饰器，来装饰当前的函数。

可以同时为一个函数指定多个装饰器，这样函数将会按照从内向外的顺序被装饰 。

3、参考

装饰器的使用 ：https://developer.aliyun.com/article/750028?spm=a2c6h.12873639.0.0.5e3f65cah9iK90&groupCode=python