下面是以注册和登录来做为例子
import hashlib
def md5(pwd):
obj = hashlib.md5("dfdfderewcsf".encode('utf-8')) # 加盐
obj.update(pwd.encode('utf-8'))
password = obj.hexdigest()
return password
def register():
user = input("username: ")
pwd = input("password: ")
password = md5(pwd)
# 注册信息存到 users.txt 文件里
with open("users.txt", mode='a', encoding='utf-8') as f:
line = "username: {} | password: {}\n".format(user, password)
f.write(line)
def login():
user = input("username: ")
pwd = input("password: ")
password = md5(pwd)
isSuccess = False
with open("users.txt", mode='r', encoding='utf-8') as f:
for line in f:
line = line.strip()
dbUser, dbPwd = line.split('|')
if dbUser.strip().split(':')[1].strip() == user and dbPwd.strip().split(':')[1].strip() == password:
isSuccess = True
break
if isSuccess:
print("登录成功")
else:
print("登录失败")
def run():
func_dict = {
"1": register,
"2": login
}
print("1. 注册 2. 登录")
choice = input("序号:")
func = func_dict.get(choice)
if not func:
print("输入错误")
else:
func()
if __name__ == '__main__':
run()
结果如下:
# 注册
1. 注册 2. 登录
序号:1
username: 4
password: 1
# 登录
1. 注册 2. 登录
序号:2
username: 4
password: 1
登录成功
给个赞呗~