#!/usr/local/env python
#coding:utf-8
#Auto:Panwenbin
#function:
lock_file=open('file_lock.txt','r+')
user_pass=open('username_file.txt','r+')
count=0

cmd=input(''' 
1：登錄系統(tǒng) 
2：退出系統(tǒng) 
請輸入您的操作：''') 
if cmd.isdigit() and int(cmd)==2: 
exit() 
elif cmd.isdigit() and int(cmd)==1: 
  while count <3:#只要重試不超過3次就不斷循環(huán) 
    username=input('請輸入您的用戶名:') 
    for i in lock_file.readlines(): #判斷用戶名是否在鎖文件中 
      i=i.split() 
      if username in i[0]: 
        print('對不起 %s 已鎖定'%username) 
        exit()

match = False 
    for j in user_pass.readlines(): 
      user,password = j.strip('\n').split() #去掉每行多余的\n并把這一行按空格分成兩列，分別賦值為user,passwd兩個變量 
      if username == user:   #判斷輸入的用戶是否存在 
          passwd=input ('請輸入密碼:') 
          if password == passwd: 
             print ('用戶名和密碼正確') 
             match = True 
             break

          elif password != passwd:#在用戶名正確的前提下，判斷輸入的密碼是否正確 
             for i in range(2): 
              passwd=input ('密碼錯誤，請重新輸入密碼:') 
              if password == passwd: 
                 print ('用戶名和密碼正確') 
                 match = True 
                 break

if username != user :#判斷用戶不存在 
      print('您輸入用戶名不存,程序已退出') 
      exit() 
    elif match == False :#如果match還為False,代表上面的循環(huán)中跟本就沒有match上用戶名和密碼 
      print('密碼和用戶名不匹配，嘗試超過三次，用戶被鎖定') 
      lock_file.write('%s \n'%username) 
      lock_file.close() 
      user_pass.close() 
      exit() 
    elif match==True: 
      print('登錄成功') 
      break 
else: 
  print('無效選項，程序已退出')