12.前方高能-装饰器进阶

Posted 2020-11-19 zhaoyang110

本节主要内容：

1.通过装饰器回顾

2.函数的有用信息

3.带参数的装饰器

4.多个装饰器同时装饰一个函数

一.通过装饰器回顾

开闭原则：对增加功能开放,对修改代码封闭

装饰器的作用：在不改变原有代码的基础上给一个函数增加功能

同用装饰器的写法：

def wrapper(fn):
    def inner(*args,**kwargs):
        """在目标函数之前做的事情"""
        ret =fn(*args,**kwargs)
        """在目标函数之后做的事情"""
        return  ret
    return inner
@wrapper
def target_func():
    print("目标函数")  #"""目标函数"""
target_func()   #函数调用

执行过程：

1.程序从上向上，当执行到@wrapper的时候，把函数作为参数传递给wrapper，得到inner函数，重新复制给target_func

2.当执行到target_func的时候，我们实际上执行的是inner函数，inner函数会先执行目标函数之前的代码，然后再执行你的目标函数，执行完目标函数最后执行的是目标函数之后的代码

二.函数的有用信息

1.如何给函数添加注释

def chi(food,drink):
    """
    这里是函数的注释，先写一下当前这个函数是干什么的，比如我这个函数就是一个吃
    :param food:   参数food是什么意思
    :param drink:  参数drink是什么意思
    :return:   返回的是什么东东
    """
    print(food,drink)
    return "very good"

2.如何获取带函数的相关信息

def chi(food,drink):
    """
    这里是函数的注释，先些一下当前这个函数是干什么的，比如我这个函数就是一个吃
    :param food:   参数food是什么意思
    :param drink:   参数drink是什么意思
    :return:   返回的是什么东东
    """
    print(food,drink)
    """
    这个可以获取到么
    """
    print(chi.__doc__)  # 获取函数的文档注释
    print(chi.__name__)   #获取到函数名称
    return  "very good"
chi("吃嘛嘛香","我不想吃")
print(chi.__doc__)
print(chi.__name__)

函数名.__name__可以查看函数的名字

函数名.__doc__可以查看函数的文档注释

接下来,我们来看一看被装饰器装饰之后的函数名：

def wrapper(func):
    def inner(*args,**kwargs):
        """在执行目标函数之前要执行的内容"""
        ret = func(*args,**kwargs)
        return ret
    return inner
# @wrapper 相当于 target_func = wrapper(target_func)语法糖
@wrapper
def target_func():
    print("我是目标函数")
target_func()     #调用函数
print(target_func.__name__)    #inner

  我们虽然访问的是target_func函数，但是实际上执行的是inner函数，这样就会给下游的程序员带俩困惑，之前不是一直执行的是target_func么，为什么突然换成了inner，inner是个什么鬼？？为了不让下游程序员有这样的困惑，我们需要把函数名修改一下，具体修改方案：

from functools import wraps    # 引入函数模块

#装饰器：对传递进来的函数进行包装，可以在目标函数之前和之后添加任意功能.
def wrapper(func):
    @wraps(func)   #使用函数原来的名字
    def inner(*args,**kwargs):
        """在执行目标函数之前要执行的内容"""
        ret = func(*args,**kwargs)
        """在执行目标函数之后要执行的内容"""
        return ret
    return inner
# @wrapper 相当于 target_func = wrapper(target_func) 语法糖
@wrapper
def target_func():
    print("我是目标函数")
#调用目标函数
target_func()
print(target_func.__name__)   #不在是inner，而是target_func了
@wrapper
def new_target_func():
    print("我是另一个目标函数")
new_target_func()
print(new_target_func.__name__)  # 不在是inner，而是new_target_func

PS：*args和**kwargs什么时候打散，什么时候聚合

1.接收参数的时候聚合，参数声明

2.传递参数的时候打散，给函数传递实参

def wrapper(func):
    # @wraps(func)
    def inner(*args,**kwargs):  #这里是聚合
        """在执行目标函数之前要执行的内容"""
        ret = func(*args,**kwargs)  #这里是打散，这里的作用，其实就是为了保证我可以装饰所有函数而准备的
        """在执行目标函数之后要执行的内容"""
        return  ret
    return inner

三.装饰器传参

现在来这样一个场景，还是昨天的约。

from functools import wraps
def wrapper(fn):
    @wraps(fn)
    def inner(*args,**kwargs):  #这里是聚合
        print("问问金老板啊，行情怎么样。")  #在执行目标函数之前要指向性的内容
        ret =fn(*args,**kwargs)
        print("++，金老板骗我")  #在执行目标函数之后要执行的内容
        return ret
    return inner
@wrapper
def yue():
    print("约一次又不会死")
yue()    #函数调用

那么现在如果查的严，怎么办呢？打电话问金老板严不严，那如果整体风声都不是那么紧呢，是不是就不需要问金老板了，所以，我们需要一个开关来控制是否要询问金老板，这时我们就需要给装饰器传递一个参数，来通知装饰器要用怎么样的方式来装饰你的目标函数

from functools import wraps  #引入一个模块
def wrapper_out(flag):
    def wrapper(fn):
        @wraps(fn)
        def inner(*args,**kwargs):
            if flag ==True:  #查的严不严呀，先问问吧
                print("问问金老板呀，行情怎么样。")
                ret = fn(*args,**kwargs)
                print("++，金老板骗我")
                return ret
            else:   # 查的不严，你慌什么
                ret =fn(*args,**kwargs)
                return ret
        return  inner
    return wrapper
@wrapper_out(False)    #传递True和False来控制装饰器内部的运行结果
def yue():
    print("约一次又不会死")
yue()

  注意：咋们之前的写法是@wrapper 其中wrapper是一个函数，那么也就是说，如果我能让wrapper这里换成个函数就行了。wrapper（True）返回的结果是wrapper也是一个函数，刚刚好和前面的@组合成一个@wrapper。依然还是原来那个装饰器，只不过这里套了3层，但你要能看懂，其实还是原来那个装饰器@wrapper

  执行步骤：先执行wrapper(True) 然后再@返回值，返回值恰好是wrapper。结果就是@wrapper

四.多个装饰器装饰同一个函数

先读一下这样一个代码

def wrapper1(fn):
    def inner(*args,**kwargs):
        print("111")
        ret =fn(*args,**kwargs)
        print("222")
        return  ret
    return inner

def wrapper2(fn):
    def inner(*args,**kwargs):
        print("333")
        ret =fn(*args,**kwargs)
        print("444")
        return ret
    return inner
@wrapper2
@wrapper1
def eat():
    print("我想吃水果")
eat()

执行顺序：首先@wrapper1装饰起来，然后获取到一个新函数是wrapper1中的inner，然后执行@wrapper2，这个时候，wrapper2装饰的就是wrapper1中的inner了，所有，执行顺序就想：

第二层装饰器前（第一层装饰器前(目标)第一层装饰器后）第二层装饰器后。程序从左到右执行起来，就是我们看到的结果