python实例 实现登录注册 密码加密

import sys
import hashlib
users = []
#盐值用于密码加密
salt = "%&***(*9"
#主页面函数
def main():
 print("*~"*20)
 print("请输入你的选择:")
 print("1.登录".center(40))
 print("2.注册".center(40))
 print("3.退出".center(40))
 print("*~"*20)
 choice =  int(input("请输入你的选择:"))
 return choice

 #注册函数
def register():
 username = input("输入用户名:")
 #strip()去掉字符串两边的空格
 if username == None or username.strip() == "":
  print("用户名不能为空:")
  return
 password = input("输入密码:")
 if password == None or password.strip() == "" or len(password) < 6:
  print("密码不能空后者长度不能小于6位!")
  return 
 for i in users:
  if i.get(username) == username:
   print("用户名重复!")
   return 
 user = {}
 user["username"] = username
 #user["password"] = password
 ##密码加密
 user["password"] = password_md5(password)
 print(user)
 users.append(user)
 print(users)

#登录函数
def login():
 unm = input("输入用户名:")
 pwd = input("输入密码:")
 is_login(unm,password_md5(pwd))

#验证登录函数
def is_login(username,password):
 for i in users:
  if i.get("username") != username:
   print("用户名称不对!")
   return 
  elif i.get("password") != password:
   print("输入密码有误!")
   return
  elif i.get("username") == username and i.get("password") == password:
   print("登录成功!")
   return

#使用md5密码加密函数
def password_md5(password):
 md5 = hashlib.md5(password.encode("utf-8"))
 md5.update(salt.encode("utf-8"))
 return md5.hexdigest()

while True:
 choice = main()
 if choice == 1:
  print("登录")
  login()
 elif choice == 2:
  print("注册")
  register()
 elif choice == 3:
  sys.exit()