学员管理系统-面向对象

import json
import os


class DataToDB():

    def readData(self):
        if os.path.exists('data.txt'):
            with open('data.txt', 'r+', encoding='utf-8') as fp:
                content = fp.read()
        else:
            content = False

        if content:
            return json.loads(content)
        else:
            return []

    def writeData(self, data):
        content = json.dumps(data)
        with open('data.txt', 'w+', encoding='utf-8') as fq:
            fq.write(content)


class StudentInfo():

    def __init__(self):
        self.db = DataToDB()
        self.stuList = self.db.readData()

    # 增
    def add(self, code, name, age):
        stuinfo = {}
        stuinfo['code'] = code
        stuinfo['name'] = name
        stuinfo['age'] = age
        self.stuList.append(stuinfo)

        self.db.writeData(self.stuList)

    # 删
    def delete(self, code):
        for item in self.stuList:
            if code == item['code']:
                self.stuList.remove(item)

        self.db.writeData(self.stuList)

    #  改
    def edit(self, code, name, age):
        for item in self.stuList:
            if code == item['code']:
                item['name'] = name
                item['age'] = age

        self.db.writeData(self.stuList)

    #  查
    def select(self):
        for item in self.stuList:
            print(item)


stu = StudentInfo()
while True:
    print("请选择您要进行的操作")
    checked = input("添加（A）,修改（E），删除（D），查询(C)，退出（Q）")
    if checked == 'a':
        code = input('请输入学号')
        name = input('请输入姓名')
        age = input('请输入年龄')
        stu.add(code, name, age)
    elif checked == 'e':
        code = input('请输入需要修改的学号')
        print('-----------------请输入新的学员信息------------------')
        name = input('请输入姓名')
        age = input('请输入年龄')
        stu.edit(code, name, age)
    elif checked == 'd':
        code = input('请输入需要删除的学号')
        stu.delete(code)
    elif checked == 'c':
        stu.select()
    elif checked == 'q':
        exit('拜拜了，您嘞~')