[Python] 학생성적관리 프로그램 (클래스, json형식 사용)

 

import json
class Student():
    studentCnt = 0
    def __init__(self):
        Student.studentCnt += 1
        self.__name = input("이름 : ")
        self.__kor = int(input("국어 성적 : "))
        self.__eng = int(input("영어 성적 : "))
        self.__math = int(input("수학 성적 : "))

    def inputStudent(self, me, path):
        if not os.path.isfile(path):
            new = dict({1:{"name" : "name"}})
            with open(path, 'w', encoding='utf-8') as file:
                json.dump(new,file, indent="\t", ensure_ascii=False)
                file.close()
            me = 0

        with open(path,'r', encoding='UTF-8') as file:
            data = json.load(file)
            if me == 0:
                stNum = len(data)
            else:
                stNum = len(data) + 1
            print(stNum, "번 학생을 입력하겠습니다")
            data[str(stNum)] = {
                "이름" : self.name, "국어" : self.kor, "영어" : self.eng, "수학" : self.math }

            with open(path, 'w', encoding='utf-8') as file:
                json.dump(data, file, indent="\t", ensure_ascii=False)
                print(len(data) + 1,"번학생 입력이 완료되었습니다")
            file.close()

    @staticmethod
    def dispClass(path):
        with open(path,'r', encoding='UTF-8') as file:
            data = json.load(file)
            for key, value in data.items():
                print ("번호 : ",end='')
                print(key, value)
    @staticmethod
    def change(path):
        num = input("번호를 입력해주세요 : ")
        with open(path, 'r', encoding='UTF-8') as file:
            data = json.load(file)
            print(data.get(num))
            subject = input("어떤 과목을 입력하시겠습니까? : ")
            if subject in data.get(num):
                grade = int(input("성적을 입력해주세요 : "))
                if subject == '수학':
                    data[num].update(국어 = grade)
                if subject == '영어':
                    data[num].update(영어 = grade)
                if subject == '수학':
                    data[num].update(수학 = grade)
                with open(path, 'w', encoding='utf-8') as file:
                    json.dump(data, file, indent="\t", ensure_ascii=False)
            else:
                print("없는 과목입니다")

    @staticmethod
    def search(path):
        num = input("번호를 입력해주세요 : ")
        with open(path,'r', encoding='UTF-8') as file:
            data = json.load(file)
            print(data.get(num))
#%%
import sys
import os.path

file_path = "/Users/dackyy/Desktop/classAdmin/"

while True:
    print("학생 성적 관리 프로그램")
    menu = int(input("1. 입력 2. 출력 3. 검색 4. 수정 5. 종료 :"))
    me = int(input("반을 입력해주세요 : "))
    now_path = (file_path + str(me) + ".json")

    if menu == 1:
        print(me, "반을 입력합니다")
        student = Student()
        student.inputStudent(me, now_path)
    elif menu == 2:
        Student.dispClass(now_path)
    elif menu == 3:
        Student.search(now_path)
    elif menu == 4:
        Student.change(now_path)
    elif menu == 5:
        sys.exit("종료")