python 装饰器保持登入状态功能

import time

current_user = {
    'username':None,
}
def auth(func):
    def wrapper(*args,**kwargs):
        if current_user['username'] != None:
            res = func(*args, **kwargs)
        else:
            username = input('please input username:').strip()
            pwd = input('please input password:').strip()
            if username == 'raimond' and pwd == '123':
                current_user['username'] = username
                res = func(*args, **kwargs)
        return res
    return wrapper

@auth#index_page = time_count(index_page)
def index_page():
    time.sleep(2)
    print('Welcome to index page...')
    return 'index_page'

@auth#home_page = time_count(home_page)
def home_page(name):
    time.sleep(1)
    print('Welcome to home page...%s'% name)
    return 'home_page'

print(index_page())
print(home_page('raimond'))

C:\Users\user\AppData\Local\Programs\Python\Python36\python.exe “C:/Users/user/PycharmProjects/hellow python/test.py”
please input username:raimond
please input password:123
Welcome to index page…
index_page
Welcome to home page…raimond
home_page

Process finished with exit code 0