python自动化开发-4

装饰器之我见

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

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

　　　

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

'''
dec函数通过用户名和密码实现验证功能。
'''
def dec(func):
    def auth():
        username = "ryb"
        password = "123456"
        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 = "123456"
        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