day17 作业

#实现注册与登录的数据的保存
"""author = chen11"""
import json

class LoginSystem:

    def start_system(self):
        print("=======欢迎使用=======\
              1.注册\
              2.登录\
              请选择:")
        choice = input()
        if choice == 1:
            self.sign()
        elif choice == 2:
            self.login()
        else:
            print("错误")

    def sign(self):
        print("注册")
        username = input('用户名:')
        password = input('密码:')
        username = username
        password = password
        choice = input("是否注册:")
        if choice == '是' or choice == 'yes':
            with open('./files/user_info.json', 'r', encoding='utf-8') as f:
                all_user = json.loads(f, encoding='utf-8')
                user = {"用户名":username, "密码":password}
                all_user.append(user)

            with open('./files/user_info.json', 'w', encoding='utf-8') as f:
                json.dump(all_user, f)

            print("注册成功")

    def login(self):
        print()
        print('请登录')
        username = input('用户名:')
        password = input('密码:')
        with open('./files/user_info.json', 'r', encoding='utf-8') as f:
            user_info = json.load(f)    #user_info是列表
            for user in user_info:  #user是字典
                if user['用户名']==username and user['密码']==password:
                    print("登录成功")
                else:
                    print("用户名或者密码错误")

s1 = LoginSystem()
s1.start_system()