#-*- coding=utf8 -*- 
# @author:sololi
# date: 2020/11/3
# 文件說(shuō)明 :
import sys
def register(username,password):#登錄功能，且與存儲(chǔ)用戶表的文本文件進(jìn)行比較
  #驗(yàn)證用戶名
  shuju=readfile()
  jg1 = 0
  i = 0
  while (i < len(shuju)):
   if (username == shuju[i]["用戶名"]):
    print("用戶名正確")
    jg1 = 1
    break
   i += 1
  # 用戶名錯(cuò)誤將不再驗(yàn)證密碼
  if (jg1 != 1):
   print("用戶名錯(cuò)誤")
  # 驗(yàn)證密碼
  if (jg1 == 1):
   jg2 = 0
   i = 0
   while (i < len(shuju)):
    if (password == shuju[i]["密碼"]):
     print("密碼正確")
     jg2 = 1
     break
    i += 1
   if (jg2 != 1):
    print("密碼錯(cuò)誤")

def logon(username):#注冊(cè)功能，且以正確格式存入文本文件
   shuju=readfile()
   jg3 = 0
   i = 0
   while (i < len(shuju)):
    if (username == shuju[i]["用戶名"]):
     print("用戶名已經(jīng)存在")
     jg3 = 1
     break
    i += 1
   if(jg3 == 0):
    while True:
     password = input("請(qǐng)輸入注冊(cè)的密碼(密碼不能小于6位，且不能為純數(shù)字)")
     if (str.isdigit(password)==1) or (len(password)<6):
       print("密碼格式錯(cuò)誤")
     else:
      break
    passwordagain=input("請(qǐng)?jiān)俅未_認(rèn)密碼")
    while True:
     if(password==passwordagain):
      break
     else:
      print("兩次密碼不一致")
      passwordagain = input("請(qǐng)?jiān)俅未_認(rèn)密碼")
    # 將注冊(cè)的用戶信息存儲(chǔ)到文本文件中
    f = open("data", mode='a+', encoding="utf8")
    if shuju == []:
     f.write("用戶名:{},密碼:{}".format(username, password))
    if shuju != []:
     f.write("\n用戶名:{},密碼:{}".format(username, password))
    print("注冊(cè)成功")
    f.close()

def readfile():#將數(shù)據(jù)轉(zhuǎn)換成列表字典形式，放在data.txt中便于后面登錄與注冊(cè)存放數(shù)據(jù)
 f = open('data', "r+", encoding="utf8")
 shuju = []
 b = []
 aa = {}
 for line in f.readlines():
  line = line.strip('\n')
  a = line.split(' ')
  i = 0
  while i < len(a):
   b = a[i].split(',')
   i += 1

  j = 0
  while j < len(b):
   if b == " ":
    break
   c = b[j].split(':', 1)
   aa[c[0]] = c[1]
   i += 1
   j += 1

  shuju.append(aa.copy()) # copy是為了防止添加是數(shù)據(jù)類型不同出錯(cuò)
 f.close()
 return shuju

while True:
 choice=input("登錄輸入1,注冊(cè)輸入2,其他任意鍵退出")
 if choice=="1":
  id=input("輸入您的賬號(hào)")
  pw=input("輸入您的密碼")
  register(id,pw)
  break
 if choice=="2":
  id=input("輸入你注冊(cè)的賬號(hào)")
  logon(id)
  continue
 else:
  print("退出成功")
  sys.exit(0)