Day10作业

print("功能菜单".center(30,'-'))
print("注册请输入0，登录请输入1")
flag = input(">")
if flag == "0":
    print("注册".center(30,'-'))
    user = input("请输入用户名：").strip()
    password = input("请输入密码：").strip()
    with open('db.txt','a',encoding='utf-8') as f:
        info = "%s:%s\n"%(user, password)
        f.write(info)
        print("注册成功！")
elif flag =="1":
    print("登录".center(30, '-'))
    user = input("请输入用户名：").strip()
    password = input("请输入密码：").strip()
    with open('db.txt', 'r', encoding='utf-8') as f:
        for i in f:
            name, pwd = i.strip("\n").split(":")
            if user == name and password == pwd:
                print("登录成功！")
                break
        else:
            print("登录失败！")

else:
    print("请输入正确的指令")