python练习：完成用户的登陆注册

import hashlib
users = {}

'''加密'''
def jiami(password):
    md5_obj = hashlib.md5()
    md5_obj.update(password.encode('utf-8'))
    jiamimima = md5_obj.hexdigest()
    return jiamimima


def register():
    username = input("请输入用户名：")
    if username is None or username.strip() == " ":
        print("用户名不能为空")
        return
    if username in users:
        print("用户名已存在，请重新输入！")
        return
    password = input("请输入密码：")
    if password is None or password.strip() == " " or len(password) < 3:
        print("密码长度不能小于3位")
        return
    jiamimima = jiami(password)
    users[username] = jiamimima
    print("注册成功！")


def login():
    username = input("请输入用户名：")
    if username not in users:
        print("用户名不存在，请重新输入！")
        return
    password = input("请输入密码：")
    jiamimima = jiami(password)
    if users[username] == jiamimima:
        print("登录成功！")
    else:
        print("密码错误！")


while True:
    print("1. 用户注册")
    print("2. 用户登录")
    print("3. 退出")
    choice = input("请输入选项：")
    if choice == "1":
        register()
    elif choice == "2":
        login()
    elif choice == "3":
        break
    else:
        print("无效选项，请重新输入！")