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是谁啊，数据库里都找不到你，滚犊子。

# 这不是我们公司员工 ，不用客气

请输入你的姓名：