Day-2

import sys,getpass,os

count = 0

print( "------欢迎登录------")

login_name = input( "请输入用户名：")

lock_file = open( 'user_lock.txt', 'r+')

lock_list = lock_file.readlines()

for lock_info in lock_list:

lock = lock_info.split()

if login_name in lock:

sys.exit( '您已被锁定...请联系管理员')

user_file = open( 'user_info.txt', 'r+')

user_list = user_file.readlines()

for user_info in user_list:

user = user_info.split()

if login_name in user:

while count < 3:

login_pwd = getpass.getpass( '请输入密码：')

if login_pwd == user[ 1]:

sys.exit( '登录成功，欢迎 %s 登录'% user[ 0])

else:

count += 1

print( '密码错误，请重新输入...您还有 %s 次机会' % ( 3-count))

else:

lock_file.write(login_name + ' \n ')

sys.exit( '密码3次错误，该用户已被锁定')

else:

sys.exit( '用户不存在')

user_file.close()

lock_file.close()