# 装饰器函数的两个特点，也是必须满足的条件
# 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是谁啊，数据库里都找不到你，滚犊子。
# 这不是我们公司员工 ，不用客气
请输入你的姓名：