編寫裝飾器，為多個函數加上認證的功能（用戶的賬號密碼來源於文件），要求登錄成功一次，后續的函數都無需再輸入用戶名和密碼

小猿圈作業：

• 編寫裝飾器，為多個函數加上認證的功能（用戶的賬號密碼來源於文件），要求登錄成功一次，后續的函數都無需再輸入用戶名和密碼

• tip：account文件的格式

• dadada 123456
  da da
  da1 da
  da2 1234
  da3 1111

   

# @Time     :2019/7/2 21:38

'''
編寫裝飾器，為多個函數加上認證的功能（用戶的賬號密碼來源於文件），
要求登錄成功一次，后續的函數都無需再輸入用戶名和密碼
'''
account_status = False #用於判斷是否已經通過驗證

def login(func):
    account_dict = {} #賬號密碼驗證用的，可以換其他方式驗證  賬號當做key，密碼當做value
    def inner():
        global account_status

        if not account_status :
            with open('account',encoding='utf-8') as f:
                account_data = f.read().split('\n')
                for item in account_data:
                    user, pwd= item.split()
                    account_dict.setdefault(user,pwd)

            username = input("username>>>: ").strip()
            # username = 'da'
            password = input("password>>>: ").strip()
            # password = 'da'
            #驗證輸入的賬號是否在account_dict已存在
            #用.__contains__方法 是為了輸入的用戶名不存在，然后報 key不存在的錯誤
            if account_dict.__contains__(username) and account_dict[username] == password:
                account_status = True #修改用戶登錄狀態
                print("登陸中~~~".center(50))
                func()#執行傳入的函數
            else:
                print('賬號密碼不匹配')

        else:
            print('已通過驗證'.center(50))
            func()

    return inner

@login
def movies():
    print('--電影專區--')

@login
def articles():
    print('--小說專區--')

@login
def music():
    print('--歌曲專區--')

movies()
articles()
music()