装饰器的应用

简单版认证登录

# def wrapper(f):
#     def inner(*args, **kwargs):
#         ret = f(*args, **kwargs)
#         return ret
#     return inner

# 登录认证

login_status = {
    'username': None,
    'status': False,
}

def login(f):
    def inner(*args, **kwargs):
        if login_status['status']:
            ret = f(*args, **kwargs)
            return ret
        else:
            print('请先登录')
            username = input('请输入用户名').strip()
            password = input('请输入密码').strip()
            if username == '二狗' and password == '123':
                login_status['username'] = username
                login_status['status'] = True
                ret = f(*args, **kwargs)
                return ret
    return inner

@login # article = login(article)
def article():
    print('欢迎访问文章页面')

@login
def dariy():
    print('欢迎访问日记页面')
    
@login
def comment():
    print('欢迎访问评论页面')


dic = {
    1: login,
    2: article,
    3: dariy,
    4: comment,
}

while 1:
    print('''
    欢迎来到博客园首页：
    1，登录
    2，文章页面
    3，日记页面
    4，评论页面
    ''')
    num = input('请输入数字：').strip()
    dic[int(num)]()

 

转载于:https://www.cnblogs.com/wangkaiok/p/9906912.html