class userLogReg:
  """
  Created on 2018.11
  @author: tox33
  """
  def __init__(self):
    """
      :param userFile: 操作的文件
    """
    self.userFile = "user.txt"
 
  def userLogin(self,username,password):
    """
      用戶登錄
      :param username:用戶名
      :param paaword:密碼
      :return:True，用戶登錄成功；False，用戶登錄失敗
    """
    try:
      f = open(self.userFile,"r",encoding="utf-8")
      for line in f:
        line = line.strip()
        line_list = line.split("#")
        if line_list[0] == username and line_list[1] == password:
          return True
        elif line_list[0] == username and line_list[1] != password:
          print("密碼錯(cuò)誤??！")
      return False
    except IOError:
      return False
 
  def userRegister(self,username,password):
    """
      用戶注冊(cè)
      1、打開(kāi)文件
      2、用戶名#密碼
      :param username:用戶名
      :param password:密碼
      :return:True，注冊(cè)成功
    """
    with open(self.userFile,"a",encoding="utf-8")as f:
      temp = "\n" + username + "#" + password
      f.write(temp)
      return True
 
  def user_exist(self,username):
    """
      檢測(cè)用戶名是否存在
      :param username:要檢測(cè)的用戶名
      :return: True，用戶名存在；False，用戶名不存在
    """
    try:
      with open(self.userFile,"r",encoding="utf-8") as f:
        for line in f:
          line = line.strip()
          line_new = line.split("#")
          if line_new[0] == username:
            return True
        return False
    except IOError:
      return False
 
  def main(self):
    """
      主控制函數(shù)
      :操作選擇參數(shù)arg: 0-注冊(cè) 1-登錄
    """
    print("歡迎來(lái)到Al用戶管理系統(tǒng)")
    while(True):
      arg = input("0：注冊(cè) ，1：登錄\n")
      if arg == "0":
        user = input("請(qǐng)?jiān)O(shè)置用戶名：")
        if self.user_exist(user):
          print("用戶名已存在，請(qǐng)重新設(shè)置！")
          continue
        else:
          pwd = input("請(qǐng)?jiān)O(shè)置密碼：")
          if self.userRegister(user,pwd):
            print("注冊(cè)成功！")
            continue
          else:
            print("注冊(cè)失??！")
            continue
      elif arg == "1":
        user = input("請(qǐng)輸入用戶名：")
        if not self.user_exist(user):
          print("用戶名不存在，請(qǐng)檢查！")
          continue
        else:
          pwd = input("請(qǐng)輸入登錄密碼：")
          if self.userLogin(user,pwd):
            print("登錄成功！")
            break
          else:
            print("登錄失敗，請(qǐng)檢查！")
            continue
      else:
        print("輸入錯(cuò)誤，請(qǐng)檢查！")
        continue
 
if __name__ == '__main__':
  test = userLogReg()
  test.main()