# Author:Byron Li

import os
BASE_DIR = os.path.dirname(__file__)  #獲取文件目錄路徑
userlist='userlist.txt'   #存取已注冊用戶名單文件，包括用戶名和密碼
blacklist='blacklist.txt'  #存取鎖定用戶的黑名單文件
userlist_path = os.path.join(BASE_DIR,userlist)  #注冊用戶名單文件路徑
blacklist_path = os.path.join(BASE_DIR,blacklist) #黑名單文件路徑

#---------------------------登陸函數(shù)（主函數(shù)）--------------------------
def login(): #登陸函數(shù)
  username = input("請輸入用戶名：")
  while(True):
    if inquire_blacklist(username):   #查詢輸入用戶名是否在黑名單中
      print("不好意思,此用戶名\"%s\"已被鎖定,禁止登陸!"%username)
      return False
    else:
      user_password = inquire_userlist(username) #查詢輸入用戶名是否存在注冊用戶名單中，若存在則返回該用戶密碼
      if user_password:
        for i in range(3):
          password = input("請輸入密碼：") if i==0 else input("密碼輸入錯誤，請重新輸入密碼：")
          if password==user_password:
            print('歡迎%s登陸!'%username)
            return True
        else:
          print("您已經(jīng)連續(xù)輸錯密碼３次,用戶名將被鎖定，禁止再登陸!")
          add_blacklist(username)       #添加用戶名到黑名單文件
          return False
      else:
        username = input('無效的用戶名,請重新輸入：')
# ------------------------------------------------------------


#-----------------查詢黑名單函數(shù)-------------------------------
def inquire_blacklist(name):  #查詢用戶名是否在黑名單中
  with open(blacklist_path, 'r') as f:
    for line in f:
      if name==line.strip():
        return True
  return False
# ------------------------------------------------------------- 

#-----------------------查詢用戶名函數(shù)--------------------------
def inquire_userlist(name):  #查詢用戶名是否在注冊用戶名單中
  with open(userlist_path, 'r') as f:
    for line in f:
      [username,password] = line.split()
      if name==username:
        return password
  return None
# ------------------------------------------------------------


#-----------------------------添加黑名單函數(shù)--------------------
def add_blacklist(name): #添加用戶名到黑名單文件
  with open(blacklist_path, 'a') as f:
    f.write(str(name)+'\n')
  return True
# ------------------------------------------------------------------------------
if login(): #執(zhí)行登陸函數(shù)
  print('登陸成功！'.center(47,'*'))
else:
  print('登陸失??！'.center(47,'*'))