# Author:Byron Li

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

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


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

#-----------------------查詢(xún)用戶名函數(shù)--------------------------
def inquire_userlist(name):  #查詢(xún)用戶名是否在注冊(cè)用戶名單中
  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,'*'))