python作业五

题目：注册登录

制作一个注册登录模块

登陆：将用户填入账户密码和users.bin中保存的账户密码进行比对,如果账户和密码完全相同那么登录成功，否则登录失败

一、代码部分


import hashlib
import json
import os
import sysUI = """
用户登录系统
1.注册
2.登陆
0.退出
"""def load(path):return json.load(open(path, "rt"))def save(dic, path: str):json.dump(dic, open(path, "wt"))while True:print(UI)num = input("请输入你的选择:")if num == "1":# 从users.bin读取已有的账户数据，用户注册完以后，将新账户增添到已有数据中保存。if os.path.exists("users.bin"):dc = load("users.bin")else:dc = {}account = input("请输入账户:")passwd = input("请输入密码:")confirm = input("请确认密码")# 检查用户的账户是否已存在 如果存在 则提醒用户重新输入账户密码if dc.get(account):print("账户已存在请重新输入")# 前后两次密码 是否一致？如果一致直接保存用户的注册信息elif passwd == confirm:# 使用MD5将用户密码假面md5 = hashlib.md5(passwd.encode())encrypt = md5.hexdigest()dc[account] = encryptsave(dc, "users.bin")print("注册成功")elif num == "2":print("进入到登录环节")# 输入账户密码account = input("请输入账户：")pwd = input("请输入密码：")# 读取users.bin得到一个字典dc= load("users.bin")# dc.get判断用户的账户是否存在if dc.get(account):# dc[account] == 输入的密码#将用户输入密码进行加密md5 = hashlib.md5(pwd.encode())encrypt_pwd = md5.hexdigest()if dc[account] == encrypt_pwd:print("登录成功")else:print("登录失败")# 登录成功 否则登录时失败else:print("账户或者密码错误")else:sys.exit()