python作业1.1，编写登录模块
实现功能：
1、用户输入账户密码
2、验证账户是否存在于黑名单，如果存在于黑名单，则执行1，否则往下执行
3、验证用户名和密码。
3.1、如果验证成功，则打印欢迎信息并退出程序；
3.2、如果用户名存在，但是密码错误，则记录这个账户的错误次数（当前会话有效）
3.2.1、如果错误次数等于等于3，则将这个用户写入锁定文件保存
4、当错误次数（用户名存在但是密码错误 + 用户名不存在）大于等于3时，sleep(5),然后重置错误次数
1 #!/usr/bin/env python3 2 __author__ = 'DSOWASP' 3 import getpass 4 import time 5 6 error_count = 0 #认证错误次数，包括用户名错误和用户名正确但是密码错误 7 error_users = [] #用于存放用户名存在，但是密码错误的列表，存放数据格式如： 8 # [{'name':'ds','count':1},{'name':owasp','count':2}],当某个用户的count为3时，就把这个 9 # 用户加到locklist.txt文件中， 由于error_users保存在内存中，因此只有在当前会话有效 10 11 flag1 = 0 #0表示用户名不存在，1表示用户名密码正确，2表示用户名正确，但是密码错误 12 flag2 = 0 #0表示当前用户不存在与error_users中,1当前用户存在于error_users中或error为非空 13 while True: 14 username = input("用户名:").strip() 15 password = input("密码:").strip() 16 if username == "EXIT": #为了能正确退出程序而设置的 17 break 18 flag1 = 0 19 with open('locklist.txt','r') as f_lock: #验证此用户是否被锁定 20 u1 = f_lock.read().split("\n") 21 if u1.count(username) != 0: 22 print("此账户已被锁定！请选择其他账户登录") 23 time.sleep(1) 24 continue 25 with open('userlist.txt','r') as f_user: #验证用户名和密码 26 for line in f_user: 27 user = line.split()[0] 28 passwd = line.split()[1] 29 if username == user: 30 if password == passwd: #用户和密码都正确，设置flag=1 31 print("Welcome to My Python Page!!!") 32 flag1 = 1 33 break 34 elif flag2 == 1: #密码不对，且当前error_users为非空 35 flag1 = 2 #用户名正确，但是密码不对 36 for i in error_users: #判断这个用户是否已存在于error_users中 37 # print("循环字典1",error_users) 38 if username == i.get("name"): #在error_users中找到了这个账户 39 # print("找到字典元素:",username,error_users) 40 flag2 = 1 #当前用户存在于error_users中 41 i["count"] = i.get("count") + 1 #置当前用户的密码错误次数加1 42 if i.get("count") >= 3: #如果错误次数等于3 43 with open('locklist.txt','a') as f_lock: #就锁定这个账户 44 f_lock.write("\n%s" % username) 45 print("输入错误次数过多，%s账户已被锁定！请联系管理员！" %(username)) 46 else: #如果错误次数小于3，则打印提示信息 47 print("用户%s已输入错误密码%d次，此账户还剩%d次尝试机会" % (i.get("name") , i.get("count"),3 - i.get("count"))) 48 break #如果再error_users中找到了当前的用户，则退出error_users的循环 49 else: #如果在error_users中没找到这个用户，则打印提示信息，并设置flag2=0 50 flag2 = 0 51 if flag2 == 0: #添加当前用户到error_users中。 52 flag1 = 2 #用户名和密码都不正确 53 print("添加前字典情况",error_users) 54 error_users.append({'name':username,'count':1}) 55 print("用户%s已输入错误密码1次，此账户还剩2次尝试机会" % (username)) 56 print("添加%s用户到字典"%username) 57 print(error_users) 58 flag2 = 1 #设置flag2=1表示error_users为非空 59 60 if flag1 == 1: #flag1=1表示用户名密码正确，则退出 61 break 62 elif flag1 == 0: #flag=0表示用户名不存在斌，error_count加1 63 error_count += 1 64 print("用户名不存在，请重新输入，还剩%d次尝试机会" %(3 - error_count)) 65 elif flag1 == 2: #flag1=2表示用户名正确但是密码错误.error_count 加1 66 error_count += 1 67 if error_count >= 3: #如果输错的此时大于等于3则打印提示信息，并sleep5秒，重置error_count 68 print("输入错误次数超过3次，请5秒后重新输入") 69 error_count = 0 70 time.sleep(5)
