python —— 装饰器函数

装饰器                                                                 

装饰器原则：遵循开放封闭原则。对外扩展开放；对修改是封闭。

装饰器的作用：在不改变原函数的情况下，在函数前后添加新的功能

装饰器的本质：闭包函数

先看思维导图能更好的学习装饰器函数                                     

1、基础版本的装饰器模板：                                                         

def wrapper(func):
    def inner(*args,**kwargs):
        print('被装饰函数前添加内容')
        ret = func(*args,**kwargs)   #被装饰函数，在这里指 my_func 函数
        print('被装饰函数后添加内容')
        return ret
    return inner

@wrapper   # 等价于 my_func = wrapper(my_func)相当于：my_func = inner
def my_func():   #正常定义的函数，也就是要被装饰的函数
    c = 1+3
    return c

#函数调用
g = my_func()   # my_func ==inner
print(g) # 4

直接过程如图所示：

2、练习：在不改变定义的my_func程序中，测试代码执行的时间？ 

import  time
def wrapper(func):
    def inner(*args,**kwargs):
        start_time = time.time()
        ret = func(*args,**kwargs) #被装饰函数，在这里指 my_func 函数
        time.sleep(0.01)
        end_time = time.time()
        print('程序执行时间为：',(end_time - start_time))
        return ret
    return inner

@wrapper   # 等价于 my_func = wrapper(my_func)相当于：my_func = inner
def my_func():   #正常定义的函数，也就是要被装饰的函数
    c = 1+3
    return c

#函数调用
g = my_func()   # my_func ==inner
print(g) # 4

如果有一天同事想取到您这个函数明德时候，上述装饰器就不能正确的给出，存在缺陷，如图：

解决办法：

import  time
from functools import wraps    
def wrapper(func):
    @wraps(func)
    def inner(*args,**kwargs):
        start_time = time.time()
        ret = func(*args,**kwargs) #被装饰函数，在这里指 my_func 函数
        time.sleep(0.01)
        end_time = time.time()
        print('程序执行时间为：',(end_time - start_time))
        return ret
    return inner

@wrapper   # 等价于 my_func = wrapper(my_func)相当于：my_func = inner
def my_func():   #正常定义的函数，也就是要被装饰的函数
    c = 1+3
    return c

#函数调用
g = my_func()   # my_func ==inner
print(g) # 4
print(my_func.__name__)    # my_func   获取函数名

3、函数是可以传参的，那么装饰器能否传参呢？                           

假如：领导要检测您程序运行时间，我们定义装饰器函数，检测过后，要如何删除装饰或者关闭装饰器呢？

import  time
inp = input('是否开启装饰器(y/n):')
flag = True if inp == 'y' else False   #装饰器开关标志位
def outer(flag):
    def wrapper(func):
        def inner(*args,**kwargs):
            if flag:
                start_time = time.time()
                ret = func(*args,**kwargs) #被装饰函数，在这里指 my_func 函数
                time.sleep(0.01)
                end_time = time.time()
                print('程序执行时间为：',(end_time - start_time))
            else:
                ret = func(*args,**kwargs)
            return ret
        return inner
    return wrapper

@outer(flag)   #先执行outer(flag),返回wrapper函数：--》 @wrapper
def my_func():   #正常定义的函数，也就是要被装饰的函数
    c = 1+3
    return c

#函数调用
g = my_func()  # my_func ==inner
print(g)  #4

 4、 有些时候，我们也会用到多个装饰器装饰同一个函数的情况。 

wrapper1(func):
    def inner1(*args,**kwargs):
        ret = func(*args,**kwargs)  #被装饰的函数
        return ret
    return inner1

def wrapper2(func):
    def inner2(*args,**kwargs):
        ret = func(*args,**kwargs)  #被装饰的函数
        return ret
    return inner2

@wrapper1
@wrapper2
def my_func():
    print('*********')
    return('11111')