要求:
1.将用户的密码以密文的形式存储
2.用户输错三次之后,再次输入此账户,显示'此账户已被冻结,无法登入'
思路:
1.先在运行脚本的路径下创建两个文件:一个存在用户信息的文件,一个存储冻结用户的文件
2.用户先注册:需要判断用户名是否存在
3.用户登入:
- 先判断用户输入的用户是否在冻结文件中
- 如果不在,再继续匹配用户名和密码
- 当用户输错三次账户或密码的时候,将此用户名写入到冻结文件中
代码如下:
import hashlib
def get_md5(user,pwd):
md5 = hashlib.md5(user.encode('utf-8')) # md5 算法加密
md5.update(pwd.encode('utf-8'))
return md5.hexdigest()
def reGISter():
'''
用户注册
:return:
'''
while True:
print('*****用户注册*****')
user = input('请输入账户(输入n返回上一级):')
if user.upper() == 'N': return
pwd = input('请输入密码:')
flag = False
with open('userinfo',mode='r',encoding='utf-8') as f: # 打开用户文件,判断用户名是否存在
for line in f:
if user in line.strip().split('|'):
flag = True
if flag:
print('用户名已存在,请重新输入')
else:
with open('userinfo',mode='a',encoding='utf-8') as f: # 将用户注册的信息写入到用户文件中
f.write('%s|%s\n' %(user,get_md5(user,pwd))) # 将用户的密码以密文的形式存储在文件中
print('\033[1;31m注册成功\033[0m')
def login():
'''
用户登入
:return:
'''
count = 1
while True:
print('*****用户登入*****')
user = input('请输入账户(输入n返回上一级):')
if user.upper() == 'N': return
pwd = input('请输入密码:')
with open('freeze',mode='r',encoding='utf-8') as f: # 打开冻结文件,判断用户输入的用户名是否在冻结文件中
content = f.read().strip().split('|')
if user in content:
print('此用户已冻结,请重新输入账户')
continue
flag = False
new_name = False
with open('userinfo',mode='r',encoding='utf-8') as f:
for line in f:
a,b = line.strip().split('|')
if user == a: # 判断用户名是否存在
new_name = True
if a == user and b == get_md5(user,pwd): # 判断账户密码是否正确
flag = True
print('\033[1;34m登入成功\033[0m')
exit()
if not new_name:
print('账户不存在请重新输入')
continue
if not flag:
print('账号或密码错误,请重新输入')
count += 1
if count == 4:
print('输错三次,已被冻结')
with open('freeze',mode='w',encoding='utf-8') as f: # 将输错三次的用户名写入到冻结文件中
f.write(user+ '\n')
return
def run():
'''
程序入口
:return:
'''
info = {'1':register,'2':login}
while True:
print('''管理界面
1.用户注册
2.用户登入''')
choice = input('请选择服务(输入N退出):')
if choice.upper() == 'N': break
data = info.get(choice)
if not data:
print('输入有误,请重新输入')
continue
data()
if __name__ == '__main__':
run()