关于装饰器

装饰器的功能：

　　　　装饰器作用就是给其他函数添加额外功能。

装饰器=高阶函数+函数嵌套+闭包

下面通过一段代码来了解一下装饰器的结构和使用方法：

import time
#装饰器模板
def timer(func):
    def wrapper():
        start_time=time.time()
        func()
        end_time=time.time()
        print("函数执行时间：%s" %(end_time-start_time))
    return wrapper

#测试函数
@timer #等价于test=timer(test)
def test():
    time.sleep(2)
    print("test函数执行完毕！")

#执行过程:存在缺点，因为每次都要先调用装饰器函数，在赋值给变量，麻烦！！！
# test=timer(test)
# # test()

#语法糖@
res=test()
print(res)

那么现在有一个问题：如果被修饰函数有返回值怎么办？别着急，下面一段代码告诉你！

#Author : Kelvin
#Date : 2019/1/5 16:56

import time
#装饰器模板
def timer(func):
    def wrapper():
        start_time=time.time()
        res=func()
        end_time=time.time()
        print("函数执行时间：%s" %(end_time-start_time))
        return res
    return wrapper

#测试函数
@timer #等价于test=timer(test)
def test():
    time.sleep(2)
    print("test函数执行完毕！")
    return "这是test的返回值！"


res=test()
print(res)

如果被修饰函数带参数又怎么办呢？继续......

#Author : Kelvin
#Date : 2019/1/5 17:39

import time
def timer(func):
    def wrapper(*args,**kwargs):
        start_time=time.time()
        func(*args,**kwargs)
        end_time=time.time()
        print("函数执行时间：%s" %(end_time-start_time))
    return wrapper

@timer
def test(name):
    time.sleep(2)
    print("%s的test函数执行完毕！"%name)

res=test("kelvin")
print(res)

下面我们使用装饰器来完成一个简单的示例（用户登陆成功够才能执行函数内容）

#Author : Kelvin
#Date : 2019/1/5 19:10

def auth(func):
    def wrapper(*args,**kwargs):
        name=input("请输入用户名：").strip()
        passwd=input("请输入密码：").strip()
        if name=="kelvin" and passwd=="123":
            res=func(*args,**kwargs)
            return res
        else:
            print("用户名或密码错误！")
    return wrapper
@auth
def index():
    print("首页！")

@auth
def home(name):
    print("%s的个人首页！"%name)

@auth
def shopping_car(name):
    print("%s购物车！"%name)

@auth
def order(name):
    print("%s订单！"%name)

index()
home("kelvin")
order("kelvin")
# 输出结果
# 请输入用户名：kelvin
# 请输入密码：123
# 首页！
# 请输入用户名：kelvin
# 请输入密码：123
# kelvin的个人首页！
# 请输入用户名：kelvin
# 请输入密码：123
# kelvin订单！

根据上面代码，示例存在缺陷，因为每次执行函数都会要求用户输入用户名密码，那么怎么样才能让用户输入一次账号密码后就能随意执行其他函数呢？下面我们来使用装饰器模拟session的功能。

#Author : Kelvin
#Date : 2019/1/5 21:00

#已注册用户信息
users=[
    {"name":"kelvin","passwd":"123"},
    {"name":"elvin","passwd":"123"},
    {"name":"bob","passwd":"123"},
    {"name":"sun","passwd":"123"},
]

current_user={"name":None,"login":False}  #模拟session存储用户登录信息
def auth(func):
    def wrapper(*args,**kwargs):
        if current_user["name"] and current_user["login"]: #如果session中有数据则说明用户已经登录，可以执行函数
            res = func(*args, **kwargs)
            return res
        name=input("请输入用户名：").strip() #如果session中没有记录用户登录信息则要求用户输入账号密码
        passwd=input("请输入密码：").strip()
        for user in users:
            if name==user["name"] and passwd==user["passwd"]:
                current_user["name"]=name
                current_user["login"]=True
                res=func(*args,**kwargs)
                return res
        else:
            print("用户名或密码错误！")
    return wrapper
@auth
def index():
    print("欢迎进入首页！")

@auth
def home(name):
    print("欢迎进入%s的个人首页！"%name)

@auth
def shopping_car(name):
    print("查看%s的购物车！"%name)

@auth
def order(name):
    print("查看%s的订单！"%name)



print(current_user) #登录前查看session中的信息
index() #执行index函数
print(current_user) #登陆后查看session中的信息
home("kelvin") #再次执行home函数可直接执行函数，不用输入账号密码
order("kelvin") #再次执行order函数可直接执行函数，不用输入账号密码

#输出结果：
# {'name': None, 'login': False}
# 请输入用户名：kelvin
# 请输入密码：123
# 欢迎进入首页！
# {'name': 'kelvin', 'login': True}
# 欢迎进入kelvin的个人首页！
# 查看kelvin的订单！

由于上面的代码中，用户信息直接放在列表中，现实中用户信息是放在各种不同的数据库中，所以如果装饰器能选择用户验证方式最好，怎么实现呢？看下面代码......

# Author : Kelvin
# Date : 2019/1/5 21:00

# 已注册用户信息
current_user = {"name": None, "login": False}  # 模拟session存储用户登录信息


def auth_type(type="fileDB"):
    def auth(func):
        def wrapper(*args, **kwargs):
            print("当前用户认证类型：",type)
            if type == "fileDB":
                if current_user["name"] and current_user["login"]:  # 如果session中有数据则说明用户已经登录，可以执行函数
                    res = func(*args, **kwargs)
                    return res
                name = input("请输入用户名：").strip()  # 如果session中没有记录用户登录信息则要求用户输入账号密码
                passwd = input("请输入密码：").strip()
                for user in open("users", "r+"):
                    user = eval(user)
                    if name == user["name"] and passwd == user["passwd"]:
                        current_user["name"] = name
                        current_user["login"] = True
                        res = func(*args, **kwargs)
                        return res
                else:
                    print("用户名或密码错误！")
            else:
                print("暂时不支持这种验证类型！")

        return wrapper

    return auth


@auth_type(type="fileDB")
def index():
    print("欢迎进入首页！")


@auth_type(type="fileDB")
def home(name):
    print("欢迎进入%s的个人首页！" % name)


@auth_type(type="fileDB")
def shopping_car(name):
    print("查看%s的购物车！" % name)


@auth_type(type="fileDB")
def order(name):
    print("查看%s的订单！" % name)


print(current_user)  # 登录前查看session中的信息
index()  # 执行index函数
print(current_user)  # 登陆后查看session中的信息
home("kelvin")  # 再次执行home函数可直接执行函数，不用输入账号密码
order("kelvin")  # 再次执行order函数可直接执行函数，不用输入账号密码

#输出结果：
# {'name': None, 'login': False}
# 当前用户认证类型： fileDB
# 请输入用户名：kelvin
# 请输入密码：123
# 欢迎进入首页！
# {'name': 'kelvin', 'login': True}
# 当前用户认证类型： fileDB
# 欢迎进入kelvin的个人首页！
# 当前用户认证类型： fileDB
# 查看kelvin的订单！

这种方式很少使用，最常用的还是前面的装饰器使用方式，应把主要精力放在前面装饰器模式的使用上。

多练习，多思考，越努力，越幸运！