Python 基于内存实现用户注册、登录系统（加密方式）

import sys
import hashlib


# 用来存储所有的用户信息
users = []
slat = "liujianhong!@#$%^&*()_+-"


def password_by_md5(password):
	md5 = hashlib.md5(password.encode("utf-8"))
	# 盐值
	md5.update(slat.encode("utf-8"))
	return md5.hexdigest()


def main():
	print("~*" * 20)
	print("")
	print("\t\t 1、用户注册\n")
	print("\t\t 2、用户登录\n")
	print("\t\t 3、退出系统\n")
	print("~*" * 20)

	choice = input("请输入您要操作的选项：")
	return choice


def exists_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 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 exists_user(username):
		print("对不起，该用户已经存在，请重新输入！！")
		return
 
	# 组建成一个字典对象
	user = {}
	user["username"] = username
	user["password"] = password_by_md5(password)
	users.append(user)
	print(users)


def login():
	username = input("请输入您的用户名称：")
	password = input("请输入您的用户密码")

	# 加密密码
	password = password_by_md5(password)
	if is_login(username, password):
		print("恭喜你，登录成功")
	else:
		print("对不起，登录失败，请重新登录")


while True:
	choice = main()
	if choice == "1":
		register()

	elif choice == "2":
		login()
	else:
		print("程序正常退出")
		sys.exit(0)