装饰器进阶

---恢复内容开始---

装饰器的形成的过程：最简单的装饰器，有返回值，有一个参数，万能参数

装饰器的作用----不想修改函数的调用方式，但是还想再原来的函数前后添加功能

装饰器的原则-----开放封闭式原则：开放--对外是开放的    封闭---对修改的是封闭的

1）最简单的装饰器

import time   --------别人已经写好的一个方法，就是返回1970到现在的秒数
def wrapper(f):            --------装饰器函数    f是被装饰器的函数
    def inner():             -------- 内部函数，形成一个闭包
       ra= time.time()    -------- 返回当前的秒数
       f()                        -------- 被装饰器函数
       ba= time.time()   -------返回当前秒数
       print(ba-ra)          -------计算程序需要多少时间运行
    return inner            --------函数也是可以用于返回值的
@wrapper                   --------语法糖    @函数装饰器
def fun():                    --------被装饰器函数
    time.sleep(2)          ---------停顿2秒，在执行下面程序
    print('123')
fun()                          --------相当于==inner 因为@wrapper 相当于
                                            fun=wraaper(fun)

2)有返回值的装饰器

import time
def wrapper(f):
    def inner():
       ra= time.time()
       ret = f()   -----因为被装饰器有返回值，所以需要返回值接受
       ba= time.time()
       print(ba-ra)
       return ret   --------返回值
    return inner
@wrapper
def fun():
    time.sleep(2)
    print('123')
    return '你好吗'
ret = fun()
print(ret)

3）万能参数

import time
def wrapper(f):
    def inner(*args,**kwargs):----因为fun相当于inner那么里面也要有参数
       ra= time.time()
       ret = f(*args,**kwargs)-------万能参数，无敌
       ba= time.time()
       print(ba-ra)
       return ret
    return inner
@wrapper
def fun(a,b=2):
    time.sleep(2)
    print('123')
    return '你好吗'
ret = fun(2,4)
print(ret)

 4)wraps用法

from functools import  wraps
def wrapper(f):
    @wraps(f)
    def inner(*args, **kwargs):
        print('快放假了')
        ret = f(*args, **kwargs)
        print('真的放假了')
        return ret

    return inner


@wrapper
def fun(day):
    """
     放假通知
    """
    print('还有%s天放假' % day)
    return '好快心啊'

print(fun.__name__)
print(fun.__doc__)
ra = fun(3)
print(ra)

 5)带参数的装饰器，用于如果你有多个被装饰的函数，哪些不要用的，判断一下即可

import time
FLAGE = False
def wrapper_na(FLAGE): 
    def wrapper(f):
        def inner(*args, **kwargs):
            if FLAGE:
                an = time.time()
                ret = f(*args, **kwargs)
                ba = time.time()
                print(ba - an)
                return ret
            else:
                 ret = f(*args, **kwargs)
                 return ret
        return inner
    return wrapper


@wrapper_na(FLAGE) ---这个相当于 @==fun == wrapper(fun)  然后 wrapper=wrapper_na(flage) 
def fun(a, b):
    time.sleep(2)
    print('大家好')
    return '你好'
@wrapper_na(FLAGE)
def fun1(a,b):
    time.sleep(3)
    print('好好好')
    return '大集合'

ret = fun(1, 2)
print(ret)

ret1 = fun1(1,32)
print(ret1)

6)多个装饰器装饰一个函数

def wrapper1(f):
    def inner():
        print('你好1')
        f()
        print('很好1')
    return inner

def wrapper2(f):
    def inner2():
        print('你好2')
        f()
        print('很好2')
    return inner2
@wrapper1
@wrapper2
def fun():
    print('都很好')
fun()

输出--------------21 你好2
都很好
很好2
很好1

执行顺序：

 

 

 

---恢复内容结束---