装饰器添加模拟用户登陆页面（基础版）

###装饰器模拟登陆
##需求，就是比如之前的网站没有没有登陆页面，现在我需要加上一个登陆的需求
##这一步只是写在本地的用户信息，用来测试的
user,password = "caicai","13421731046"
##装饰里面的功能（3）
def auth(func):
    ##然后定义一个内嵌函数（3-1）
    def wrapper(*args,**kwargs):##传入参数，以备不时之需
        ##现在可以开始判断用户的输入了
        userInput = input("user:").strip()
        passwordInput = input("password:").strip()
        if userInput == user and passwordInput == password:
            print("\033[32:1m Welcome to \033[0m")
            ##用户登陆完成后应该要执行它之前的功能了
            res = func(*args,**kwargs)
            return res
        else:
            exit("\033[31:1m login failure \033[0m")
    return wrapper
##比如现在有两个页面要登陆（1）这个比如是之前原本就有的了
def index():
    print("welcome to index page")
##先把功能名字写上吧（2）
@auth
##注意装上装饰器后本来的功能上的返回值已经不在home身上了
##想要拿回返回结果就在谁调用了它的后面returnres = func(*args,**kwargs)   return res
def home():
    print("welcome to home page")
@auth
def bbs():
    print("welcome to bbs page")
##home页面和bbs页面需要加上登陆页面

##调用
index()
home()
bbs()