day 11 homework

"""__author__  夏伟鑫"""
import json


def get_num():
    return int(input('请输入一个序号'))


def login_m():
    while 1:
        print('========Welcome to the student management system=========')
        print('                      1.登录                             ')
        print('                      2.注册                             ')
        print('                      3.退出                             ')
        print('=========================================================')

        num1 = get_num()
        if num1 == 1:
            login_ing()
        elif num1 == 2:
            register_r()
        elif num1 == 3:
            pass


def register_r():
    stu_name = input('请输入您的姓名:')
    stu_id = input('请输入您的账号:')
    stu_tel = input('请输入您的电话:')
    stu_pwd = input('请输入您的密码:')

    print('注册成功')
    with open('student.json', 'r', encoding='utf-8') as f:
        content = f.read()
        list_stu = json.loads(content, encoding='utf-8')

    student = {'name': stu_name, 'pwd': stu_pwd, 'tel': stu_tel, 'id': stu_id}
    list_stu.append(student)

    with open('student.json', 'w', encoding='utf-8') as f1:
        f1.write(json.dumps(list_stu))


def login_ing():
    stu_id = input('请输入您的账号:')
    stu_pwd = input('请输入您的密码:')
    with open('student.json', 'r', encoding='utf-8') as f:
        content = f.read()
        list_stu = json.loads(content, encoding='utf-8')

        temp = 0
        for x in list_stu:
            if x['id'] == stu_id and x['pwd'] == stu_pwd:
                temp = 1

        if temp == 1:
            print('登录成功')
        else:
            print('登录失败')
            print('请输入正确的用户名或者密码')


login_m()