python-使用装饰器实现简单的登陆验证

#!/usr/bin/env python
# -*- coding:utf-8 -*-

user_dic={'username':None,'login':False}

def auth_func(func):
    def wrapper(* args):
        if user_dic['username'] and user_dic['login']:
            return func(* args)

        username=input("输入用户名:")
        passwd=input("输入密码:")

        while True:
            if username == "***" and passwd == "******":
                res=func(* args)
                user_dic['username']=username
                user_dic['login']=True
                break
            else:
                username = input("请输入用户名:")
                passwd = input("输入密码:")
        return res
    return wrapper

@auth_func
def index():
    print("欢迎来到>>>>>>>>>>>>>.")

@auth_func
def home(name):
    print("欢迎回来 %s" %name)


index()
home("***")

 

你可能感兴趣的:(python)