Py修行路 python基础 （十）装饰器

装饰器

一、定义

装饰器：顾名思义，就是对某个东西起到装饰修饰的功能。

python中的装饰器，其本质上就是一个python函数，它可以让其他函数在不需要任何代码变动的前提下增加额外功能。通俗理解就是 函数 闭包 的实用。

二、语法及注意事项

　　1、书写规范 @ *** ***指装饰器的函数名

　　2、装饰器作为一个函数，他会把其下一行的主函数名作为变量，传递到自己的函数去调用。再重新赋值主函数。

　　3、装饰器必须放到被装饰的函数上边，并且独占一行；多个装饰器下 ，按照先下后上执行。

三、为什么要用装饰器？

开放封闭原则

源代码上线之后，主函数的定义和调用方式尽量避免更改。装饰器本身在源代码中就是可调用的对象。

四、装饰器使用举例

原函数如下：

 import time
 def index():
     time.sleep(3)
     print("welcome to oldboy")
 index()

执行结果：

welcome to oldboy

在原函数不被更改的情况下，增加新的功能：计算主函数执行的时间！

1、最原始的无参装饰器（无位置参数）。

 import time

 def timer(func):
     def wrapper():
         start_time=time.time()
         func()
         stop_time = time.time()
         print("running time:%s"%(stop_time-start_time))
     return wrapper
 @timer
 def index():
     time.sleep(3)
     print("welcome to oldboy")
 index()

执行结果：

 welcome to oldboy
 running time:3.000171422958374

2、无参装饰器（有位置参数）

 #为定义可以传值的函数，添加计时的装饰器，此时要注意函数和装饰器中的位置参数传值的问题！！！
 import time

 def timer(func):
     def wrapper(*args,**kwargs):  #有权接管，无权为空！
         start_time = time.time()
         func(*args,**kwargs)  #home(name)
         stop_time = time.time()
         print("run time is %s"%(stop_time-start_time))
     return wrapper

 @timer   #index = timer(index)
 def index():
     time.sleep(1.5)
     print("welcome to beijing!")

 @timer
 def home(name):  #wrapper('alex')
     time.sleep(1)
     print("welcome to %s home page"%name)

 @timer
 def auth(name,passwd):
     time.sleep(0.5)
     print(name,passwd)

 @timer
 def tell():
     print("---***---")

 index()
 home("alex")
 auth("alex","alex3714")
 tell()

执行结果如下：

 welcome to beijing!
 run time is 1.5000858306884766
 welcome to alex home page
 run time is 1.0000569820404053
 alex alex3714
 run time is 0.5010290145874023
 ---***---
 run time is 0.0

3、无参装饰器（主函数带有返回值的情况！）

 #给带有返回值的函数添加统计时间的装饰器!

 import time

 def timer(func):
     def wrapper(*args,**kwargs):
         start_time = time.time()
         res = func(*args,**kwargs) #my_max(1,2)
         stop_time = time.time()
         print("run time is %s"%(stop_time-start_time))
         return res  #先将所有的功能接收出来，然后在返回
     return wrapper

 @timer
 def my_max(x,y):
     time.sleep(2)
     print("max:")
     res = x if x>y else y
     return res   #函数的返回值！
 res = my_max(1,2)   # res = wrapper(1,2)
 print(res)

执行结果如下：

 max:
 run time is 2.0001144409179688
 2

4、装饰器的实质既然是函数的调用，那么上面介绍完无参的情况，现在举例介绍有参的情况。

有参装饰器（登陆认证）

 #为主函数添加认证功能。主函数不变，在主函数上@*** 一个装饰器，装饰器是靠函数来实现。
 #认证功能是在真正的函数调用之前添加上认证的功能，所以认证是在装饰器中添加。
 #有参装饰器是在装饰器上添加区分，认证的时候，需要区分不同类型信息的来源（如：本地文件，数据库），
 #所以在装饰器上就需要添加认证参数，以区分开认证的来源，执行认证的时候，会在认证主函数中进行逻辑操作。
 def auth2(auth_type):  #参数认证函数
     def auth(func):
         def wrapper(*args,**kwargs):  #认证函数
             if auth_type == "file":
                 name = input("username:")
                 passwd = input("passwd：")
                 if name == "song" and passwd =="":
                     print("auth successful")
                     res = func(*args,**kwargs)
                     return res   #认证成功之后源代码再运行。
                 else:
                     print("auth error")
             elif auth_type == "sql":
                 print("不会！")
         return wrapper
     return auth

 #@auth2(auth_type="sql")    #调用  不同认证类型的参数
 #def index():  #index = auth(index)
 #    print("welcome to index page")

 @auth2(auth_type="file")  #调用  不同认证类型的参数
 def my_index():
     print("welcome to inexd page")
 #index()
 my_index()

执行结果如下：

 username:song
 passwd123
 auth successful
 welcome to inexd page

5、多个装饰器调用！实现主函数的登录认证功能

 #多个装饰器调用！
 #1、加上了统计函数的执行时间
 #2、加上了认证功能，为避免重复登录认证，执行功能完成之后再判断登录状态。已登录则跳过，第一次登录会有登录信息的认证。

 import time
 current_login={'name':None,'login':False}

 def timmer(func):
     def wrapper(*args,**kwargs):
         start_time=time.time()
         res=func(*args,**kwargs)  #my_max(1,2)
         stop_time=time.time()
         print('run time is %s' %(stop_time-start_time))
         return res
     return wrapper

 def auth2(auth_type='file'):
     def auth(func):
         # print(auth_type)
         def wrapper(*args,**kwargs):
             if current_login['name'] and current_login['login']:
                 res=func(*args,**kwargs)
                 return res
             if auth_type == 'file':
                 name=input('username: ')
                 password=input('password: ')
                 if name == 'zhejiangF4' and password == 'sb945':
                     print('auth successfull')
                     res=func(*args,**kwargs)
                     current_login['name']=name
                     current_login['login']=True
                     return res
                 else:
                     print('auth error')
             elif auth_type == 'sql':
                 print('还他妈不会玩')
         return wrapper
     return auth

 @timmer
 @auth2(auth_type='file') #@auth  #index=auth(index)
 def index():
     print('welcome to inex page')

 @auth2()
 def home():
     print('welcome to home page')

 #调用阶段
 index()
 home()

执行结果如下：

 username: zhejiangF4
 password: sb945
 auth successfull
 welcome to inex page
 run time is 10.760615348815918
 welcome to home page