Python第五次作业

完成用户注册登录系统

import sys
import hashlib
users = []   #存储用户信息
slat = "asd!@#$%^"
def password_md5(password):
    md5 = hashlib.md5(password.encode("utf-8"))
    md5.update(slat.encode("utf-8"))
    return md5.hexdigest()
def main():   #菜单设置
    print("~*" * 20)
    print("\t\t 1.用户注册")
    print("\t\t 2.用户登录")
    print("\t\t 3.退出系统")
    print("~*" * 20)
    choice = input("请输入您要操作的选项:")
    return choice
def register():   #登录设置
    username = input("请输入你的用户名称:")
    password = input("请输入你的用户密码:")
    if username == None or username.strip() == " ":
        print("用户名不能为空")
        return
    if password == None or password.strip() == " " or len(password) < 3:
        print("密码长度不能小于3位")
        return

    if exits_user(username):
        print("对不起，该用户已经存在，请请重新输入")
        return
    user = {}
    user["username"] = username
    user["password"] = password_md5(password)
    users.append(user)
    print(users)
def exits_user(username):    #用户存在判断
    for i in users:
        if i.get("username") == username:
            return True
    return False
def is_login(username, password):  #密码检测
    for i in users:
        if i.get("username") == username and i.get("password") == password:
            print("登录成功")
            return True
    return False
def login():   #登陆设置
    username = input("输入用户名：")
    password = input("输入用户密码：")
    password = password_md5(password)
    if is_login(username, password):
        print("恭喜你登录成功")
    else:
        print("对不起，登录失败，请重新登录")
while True:
    choice = main()
    if choice == "1":
        print("用户注册")
        register()
    elif choice == "2":
        print("用户登录")
        login()
    else:
        print("程序正常退出")
        sys.exit()