編寫裝飾器,為多個函數加上認證功能(用戶賬號和密碼來源於文件)

　　這里做個最簡單的賬號密碼驗證,主要為了實現裝飾器同時對多個函數起作用

FLAG = False    #加一個變量防止多次登錄驗證
def wrapper(f):
    def inner(*args,**kwargs):
        '''登錄程序'''
        global FLAG #將FLAG變量設置成全局變量
        if FLAG:
            ret = f(*args, **kwargs)
            return ret
        else:
            username = input('username:')
            password = input('password:')
            if username == '123' and password == '345':
                FLAG = True
                ret = f(*args,**kwargs)
                return ret
            else:
                print('登錄失敗')
    return inner

@wrapper
def func_1():
    print('func1 is running!')

@wrapper
def func_2():
    print('func2 is running!')

func_1()
func_2()

　　運行結果:

username:123
password:345
func1 is running!
func2 is running!