python自动化开发-4

装饰器之我见

　　python里的装饰器到底是个什么东东，初看起来，可能有的小伙伴会发懵啊，哈哈。

　　先来个装饰器的例子瞅瞅：

　　　

  Author:RYB
 # __*__coding:utf-8__*__

 '''
 dec函数通过用户名和密码实现验证功能。
 '''
 def dec(func):
     def auth():
         username = "ryb"
         password = ""
         user = input("username:").strip()
         passwd = input("password:").strip()
         if user == username and passwd == password:
             msg = print("验证通过，欢迎来到京东商城！")
         elif user == username and passwd != password:
             msg = print("密码不正确，请输入正确的密码！")
         elif user != username and passwd != password:
             msg = print("用户名和密码均不正确，请输入正确的用户名和密码！")
         else:
             msg = print("用户名不正确，请输入正确的用户名！")
     return auth

 #@dec <==> jd=dec(jd)，dec函数用来给jd函数添加验证功能，所以dec函数是装饰器。
 @dec
 def jd():
     print("welcome to jd!")

 #执行jd函数
 jd()

　上面装饰器例子（函数版）：可读性更强一些哈哈

　　　　

 # Author:RYB
 # __*__coding:utf-8__*__

 #用户名和密码都正确
 def user_passd_right():
     msg = print("验证通过，欢迎来到京东商城！")

 #用户名正确，密码不正确
 def useright_passd():
     print("密码不正确，请输入正确的密码！")

 #用户不正确，密码正确
 def user_passdright():
     print("用户名不正确，请输入正确的用户名！")

 #用户名和密码都不正确
 def user_passd():
     print("用户名和密码均不正确，请输入正确的用户名和密码！")

 '''
 dec函数通过用户名和密码实现验证功能。
 '''
 def dec(func):
     def auth():
         username = "ryb"
         password = ""
         user = input("username:").strip()
         passwd = input("password:").strip()
         if user == username and passwd == password:
             user_passd_right()
             func()
         elif user == username and passwd != password:
             useright_passd()
         elif user != username and passwd != password:
            user_passd()
         else:
             user_passdright()
     return auth

 #@dec <==> jd=dec(jd)，dec函数用来给jd函数添加验证功能，所以dec函数是装饰器。
 @dec
 def jd():
     print("welcome to jd!")

 #执行jd函数
 jd()

　　　　

那么到底装饰器是个什么东西呢？我想，通过上面的例子说的通俗一些，伙伴们多少知道一些吧。装饰器其实也是一个函数，一个用来包装函数的函数，装饰即修饰，意指为其他函数添加新功能。

　　装饰器定义：本质就是函数，功能是为其他函数添加新功能

　　装饰器需要遵循的原则：

　　　　1.不修改被装饰函数的源代码（开放封闭原则）；

　　　　2.为被装饰函数添加新功能后，不修改被修饰函数的调用方式。

　　装饰器知识储备：装饰器=高阶函数+函数嵌套+闭包

。

　　装饰器的例子1：

　　　　

 # Author:RYB
 # __*__coding:utf-8__*__
 import time

 def timer(func):
     def wrapper():
         start_time = time.time()
         func()
         stop_time = time.time()
         print('run time is %s' %(stop_time-start_time))
     return wrapper
 #@timer等价index=timer(index)
 @timer
 def index():
     print("in the index")
 index()

运行结果如下：　　

　　in the index
　　run time is 0.0

　　装饰器的例子2：有参数的

　　　　

 # Author:RYB
 # __*__coding:utf-8__*__

 import time
 def timer(func):
     def wrapper(*args,**kwargs):
         start_time = time.time()
         res=func(*args,**kwargs)
         stop_time = time.time()
         print('run time is %s' %(stop_time-start_time))
         return res
     return wrapper

 #@timer等价index=timer(index)
 @timer
 def index(msg):
     print("in the index:",msg)
     return 1
 res=index("hello myworld")
 print(res)

运行结果：　　

　　in the index: hello myworld
　　run time is 0.0
　　1