python3装饰器用法示例

装饰器在编写后台的逻辑时有可能会用到,比方说一个场景:公司的员工想要登录自己公司的考勤记录系统去修改自己的考勤,以前是随便谁都有权限去修改,这样老板不同意了,现在,要在你登录前加一个权限验证的逻辑,如果你权限没有达到等级要求,你就不能登录系统去修改自己的考勤记录。如何在不改动原函数的情况下去增加这部分权限验证的功能呢?这就用到的装饰器函数。

装饰器函数的一大有点就是,在不改动原函数的基础上,去给函数增加新的功能。

装饰器有两种,一种是没有参数的,一种是有参数的,本示例两种装饰器都有,不过只运行了有参数的。

有关用法和解释在代码中已经注释。下面是一个例子。

# 装饰器函数的两个特点,也是必须满足的条件
# 1. 闭包
# 2. 外部函数的返回值必须是内部函数

# 这是一个没有参数的装饰器的例子
def decorator(func):
    def wrapper():
        print('在登录前,先检查有没有此账号。')
        # !@#$%^&*,在数据库中查询
        # 如果数据库的已注册表中查到此用户,就执行登录函数 login()。
        # 如果没有查到,则显示“请先注册,然后登录”
        func()
    return wrapper

# 上面已经写好了一个装饰器,下面是用法
@decorator
def login1():
    # 比方说这里是一段很复杂的逻辑,现在用登录操作代替
    print('我现在要登录了')


# 这是一个有参数的装饰器例子
def decorator2(func):
    def wrapper2(name):
        # 数字越小,权限越高
        priLevelDict = {'赵总': 1, '赵总': 2, '小王': 3}

        # 判断字典中是否存在某个key,可以用 in 或者 not in来判断
        # print('小王' in priLevelDict)

        if name in priLevelDict:
            print('正在对{}进行权限验证'.format(name))
            # 如果权限验证通过,则可以进行下一步的登录,管理
            # 如果权限验证不通过,则不能登录。逻辑自己设计就好
            # print(priLevelDict[name])
            if priLevelDict[name] == 1 or priLevelDict[name] == 2:
                print('验证通过')
                func(name)
            elif priLevelDict[name] == 3:
                print('对不起,{},您没有权限登录,请有点自知之明。'.format(name))
                print()
        else:
            print('你tm是谁啊,数据库里都找不到你,滚犊子。')

    return wrapper2

@decorator2
def login2(name):
    print('尊敬的{}, 现在您可以登录管理系统进行管理'.format(name))
    print()

if __name__ == '__main__':
    while 1:
        name = input('请输入你的姓名:')
        login2(name)

运行结果:

请输入你的姓名:赵总
正在对赵总进行权限验证
验证通过
尊敬的赵总, 现在您可以登录管理系统进行管理
# 这可是领导,老板要查岗,必须得通过啊

请输入你的姓名:钱经理
正在对钱经理进行权限验证
验证通过
尊敬的钱经理, 现在您可以登录管理系统进行管理
# 这是顶头上司,也要好好巴结一下

请输入你的姓名:小王
正在对小王进行权限验证
对不起,小王,您没有权限登录,请有点自知之明。
# 这是同事,说话比较随意

请输入你的姓名:老李
你tm是谁啊,数据库里都找不到你,滚犊子。
# 这不是我们公司员工 ,不用客气
请输入你的姓名:

 

    原文作者:正态分个布
    原文地址: https://www.cnblogs.com/zrmw/p/11244215.html
    本文转自网络文章,转载此文章仅为分享知识,如有侵权,请联系博主进行删除。
点赞