python第五次作业

补全用户注册登录界面代码

制作一个注册登录模块
注册：将用户填入的账户和密码保存到一个文件(users.bin)
登陆：将用户填入账户密码和users.bin中保存的账户密码进行比对,如果账户和密码完全相同 那么登录成功，否则登录失败

代码

import hashlib
import json
import os
import sys

# user interface 用户是界面
UI = """
用户登录系统
1.注册
2.登陆
0.退出
"""


# 读取users.bin
def load(path):
    return json.load(open(path, "rt"))


# 保存user.bin
def save(dic, path: str):
    json.dump(dic, open(path, "wt"))


while True:
    input("按回车继续...")
    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] = encrypt
            save(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()

首先导入需要的库：hashlib用于加密，json用于处理JSON格式的数据，os用于检查文件是否存在，sys用于退出程序。

定义用户界面UI

定义load函数，用于加载users.bin文件中的用户数据

定义save函数，将用户数据保存到users.bin文件中

使用while = True无限循环显示用户界面以获取用户输入

当用户输入1时，进入注册操作，首先尝试加载users.bin文件中的用户数据，文件存在即加载，不存在就创建一个新的文件，然后让用户输入账户和密码，如果账户已经存在，则提示已经存在该用户，如果不存在，经过两次确认密码则将该新用户添加到文件中。

当用户输入2时，进入登录操作，首先加载users.bin中的用户数据，然后提示用户输入账户和密码，如果账户名存在于字典中，则继续对比加密后的密码与字典中存储的密码是否相同，如果相同则提示登录成功，否则提示账户名与密码不匹配

当用户输入0时，退出程序

如果用户输入其他值，则提示用户输入无效

注意：在实际应用中，密码不建议使用明文保存，一般可以使用md5,但md5也有被还原的风险，可以添加盐值来提高安全性。

运行结果

按回车继续...

用户登录系统
1.注册
2.登陆
0.退出

请输入你的选择:1
请输入账户:xx
请输入密码:123
请确认密码123
账户已存在请重新输入
按回车继续...

用户登录系统
1.注册
2.登陆
0.退出

请输入你的选择:2
进入到登录环节
请输入账户xx
请输入密码123
登录成功
按回车继续...

用户登录系统
1.注册
2.登陆
0.退出

请输入你的选择:0

Process finished with exit code 0