Python进阶-IV-Wrapper高阶

一、装饰器回顾：

1、标准的装饰器示例

 def trapper(func):
     def inner(*args, **kwargs):
         print('插入到被装饰函数前的功能！')
         res = func(*args, **kwargs)
         print('插入到被装饰函数后的功能！')
         return res

     return inner

 @trapper
 def func():
     print('this is func!')

 func()

2、代码运行时间装饰器的优化

考虑如果同时有500个函数要解除检查代码运行时间，会不会费太多时间取注释掉语法糖？
是不是可以用标志控制，只改它就可以关闭或者打开装饰器

 import time
 ON_OFF = True  # 装饰器控制开关，默认是关的

 def count_time(func):
     def inner(*args, **kwargs):
         global ON_OFF   # 如果不能使用该关键字，毕竟不安全
         if ON_OFF:
             start = time.time()
             res = func(*args, **kwargs)
             end = time.time()
             print('the function run use: %s'%(end - start))
             return res
         re =  func(*args, **kwargs)
         return re

     return inner

 @count_time
 def func_wrapped():
     time.sleep(0.233)
     print('wrapping me!')

 @count_time
 def woqu():
     time.sleep(0.31415926)
     print('wo ququququuuuuuuuu')

 func_wrapped()
 woqu()

3、高阶优化代码运行时间装饰器

如果不使用global关键字，在局部域中使用全局变量，毕竟不安全，如何改造？

 ef timmer_out(flag):   # 在装饰器外再加一层函数，传入标志位，函数的最多嵌套三层！
     def timmer(func):
         def inner(*args, **kwargs):
             if ON_OFF:
                 start = time.time()
                 res = func(*args, **kwargs)
                 end = time.time()
                 print('the function run use: %s' % (end - start))
                 return res
             re = func(*args, **kwargs)
             return re

         return inner
     return timmer

 @timmer_out(ON_OFF)  # --> 等同于先执行timmer_out(ON_OFF)函数，返回：timmer；接着执行@timmer语法糖
 def func_use_time():
     time.sleep(0.233)
     print('use ti me!')

 @timmer_out(ON_OFF)
 def func_waste_time():
     time.sleep(0.31415926)
     print('wo waste')

 func_use_time()
 func_waste_time()

二、@wraps的使用

解决在被装饰函数得不到原本函数名的问题

 print(func.__name__)  # 打印的结果时：inner ，因为被装饰后，func = trapper(func) 即 func = inner
 # 如何得到它原本的名字了？
 # 引入库
 from functools import wraps

 def a_new_trapper(a_func):
     @wraps(a_func)
     def wrap_the_function():
         print("I am doing some boring work before executing a_func()")
         a_func()
         print("I am doing some boring work after executing a_func()")

     return wrap_the_function

 @a_new_trapper
 def a_function_requiring_trapper():
     """Hey yo! Decorate me!"""
     print("I am the function which needs some trap to "
           "remove my foul smell")

 print('a_function_requiring_trapper.__name__')
 print(a_function_requiring_trapper.__name__)

三、多层装饰器

即多个装饰器同时装饰一个函数；

引入场景：网上购物时登录后，其他操作如：添加商品到购物车，删除购物车中的商品就不用再登录了！

同时登录用户的每个动作都要记录到日志中。

 FLAG = False

 def login(func):
     @wraps(func)  # 此处也要加，否者写入日志的还是inner
     def inner(*args, **kwargs):
         global FLAG
         while 1:
             if FLAG is False:
                 name = input('请输入您的登录名：')
                 passwd = input('请输入您的登录密码：')
                 if name == 'lisi' and passwd == '':
                     FLAG = True
                     break
                 print('您输入的密码或者登录名不正确！请重新输入！')
                 continue
             break
         res = func(*args, **kwargs)
         return res

     return inner

 def log(func):
     @wraps(func)
     def inner(*args, **kwargs):
         res = func(*args, **kwargs)
         with open('log', mode='a', encoding='utf-8') as f:
             f.write(func.__name__ + '\n')
         return res

     return inner

 @log
 @login
 def add_merc_to_cart():
     print('I just add some merchandise to the cart!')

 @log
 @login
 def del_merc_in_cart():
     print('I just del some merchandise in the cart!')

 add_merc_to_cart()
 del_merc_in_cart()

多层装饰器的标准演示流程

 ef wrapper1(func):
     def inner(*args, **kwargs):
         print('第一个装饰器wrapper1，在函数前的添加！')
         res = func(*args, **kwargs)
         print('第一个装饰器wrapper1，在函数后的添加！')
         return res

     return inner

 def wrapper2(func):
     def inner2(*args, **kwargs):
         print('第二个装饰器wrapper2，在函数前的添加！')
         res = func(*args, **kwargs)
         print('第二个装饰器wrapper3，在函数后的添加！')
         return res

     return inner2

 def wrapper3(func):
     def inner3(*args, **kwargs):
         print('第三个装饰器wrapper3，在函数前的添加！')
         res = func(*args, **kwargs)
         print('第三个装饰器wrapper3，在函数后的添加！')
         return res

     return inner3

 @wrapper3
 @wrapper2
 @wrapper1
 def wuha():
     print('wu haaaaaaaaaaaaaaaaa!')

 wuha()