#!/usr/bin/env python
# -*- coding:utf-8 -*-
import hashlib
#3、设置加密密码
def md5(arg):
ooo = hashlib.md5(bytes("asd;sad",encoding="utf-8"))
ooo.update(bytes(arg,encoding="utf-8"))
#返回加密后的
return ooo.hexdigest()
def login(user,pwd):
with open("db","r",encoding="utf-8") as f:
for line in f:
u,p = line.strip().split("|")
if u == user and p == md5(pwd):
return True
#2、注册
def register(user,pwd):
with open("db","a",encoding="utf-8") as f:
tmp = user + "|" + md5(pwd)
f.write(tmp)
#1、
i = input("1,登录；2，注册")
if i == "2":
user = input("用户名：")
pwd = input("密码")
register(user,pwd)
elif i =="1":
user = input("用户名")
pwd = input("密码")
r = login(user,pwd)
if r:
print("登录成功")
else:
print("登录失败")
