파이썬 class 성적관리 프로그램

import pickle
import os
import copy
 
 
class studentInfo:
    __dataList = dict()
 
    def __init__(self):
 
        if os.path.isfile("db.p"):  # 파일 존재 확인
            dpSize = os.path.getsize('db.p')  # 파일이 비어있는지 검사
            if dpSize != 0:
                with open('db.p', 'rb') as file:  # 파일 열어서 읽기
                    self.__dataList = pickle.load(file)  # __dataList 변수에 저장하기
 
    def getDataList(self):
        return self.__dataList
 
    def setDataList(self, datalistTemp):
        with open('db.p', 'wb') as file:
            pickle.dump(datalistTemp, file)
 
 
class studentManagement:
    oj = studentInfo()
    __stddataList = oj.getDataList()
 
    def insertinfo(self):  # 학생정보 입력 메서드
        category = ['학과', '이름', '국어', '영어', '수학', '총점', '평균']
 
        data = input('학번, 학과, 이름, 국어 점수, 영어 점수, 수학 점수를 입력하시오 : ').split()
        if len(data) == 6:
            std_id = data[0]  # 학번
            if std_id not in self.__stddataList:  # 학생정보딕셔너리에 현제 입력한 학번이 존재하지않을 경우
                del data[0]  # 입력받은 데이터에서 학번 삭제
                data.append(int(data[2]) + int(data[3]) + int(data[4]))
                data.append(int(data[5]) // 3)
                dataDict = dict(zip(category, data))  # 관리 항목과 입력받은 데이터를 key : value 형식으로 딕셔너리 생성
                self.__stddataList.setdefault(std_id, dataDict)  # 리스트에 정보 추가
            else:
                print('해당 학번 학생이 이미 존재합니다')
        else:
            print('형식에 맞게 입력하시오')
 
    def dispALLInfo(self):  # 학생 전체 출력
        if not self.__stddataList:
            print('출력할 값이 존재하지 않습니다')
        else:
            for key, value in self.__stddataList.items():  #
                print(key, end=' ===>   ')
                for key, value in value.items():
                    print(key, end=' = ')
                    print(value, end='   ')
                print()
 
    def dispInfo(self, std_id):  # 학생 한명 출력
        if std_id not in self.__stddataList:
            print('출력할 값이 존재하지 않습니다')
        else:
            print(std_id, end=' ===>   ')
            for key, value in self.__stddataList.get(std_id).items():
                print(key, end=' = ')
                print(value, end='   ')
            print()
 
    def dispInfoTp(self, *temp):  # sorted 반환값 전용 출력 함수
        for i in temp:
            x, y = i
            self.dispInfo(x)
 
    def deleteInfo(self, a):  # 사번으로 삭제
        try:
            del self.__stddataList[a]
            print('삭제 후 결과\n')
            self.dispALLInfo()  # 삭제 후 학생정보 전체 출력력
        except Exception as e:
            print('해당하는 데이터가 존재하지 않습니다')
 
    def dbDump(self):  # DB 저장
        self.oj.setDataList(self.__stddataList)
 
    def searchInfo(self):  # 학생 검색
        searchChoice = input('검색 유형을 선택하시오 => 1. 학번  2. 이름  3. 학과  : ')
        if searchChoice == '1':
            std_id = input('검색할 학생의 학번을 입력하시오 : ')
            self.dispInfo(std_id)
        elif searchChoice == '2':
            std_name = input('검색할 학생의 이름을 입력하시오 : ')
            for key, value in self.__stddataList.items():
                if value.get('이름') == std_name:
                    self.dispInfo(key)
        elif searchChoice == '3':
            std_depart = input('검색할 학생의 학과를 입력하시오 : ')
            for key, value in self.__stddataList.items():
                if value.get('학과') == std_depart:
                    self.dispInfo(key)
 
    def modifyInfo(self):  # 학생 정보 수정
        std_id = input('정보수정할 학생의 학번을 입력하시오 : ')
        self.dispInfo(std_id)
        if std_id in self.__stddataList:  # 딕셔너리 키값중에 입력받은 키값이 존재 할 경우
            ch = input('\n수정할 항목을(총점, 평균 제외)입력 하시오 : ')
            value = input('변경값을 입력 하시오 : ')
            self.__stddataList[std_id].update({ch: value})
            if ch == '국어' or '수학' or '영어':  # 국어 수학 영어 항목을 수정할 경우
                self.__stddataList[std_id].update(
                    총점=(int(self.__stddataList[std_id].get('국어')) + int(self.__stddataList[std_id].get('영어')) + int(
                        self.__stddataList[std_id].get('수학'))))  # 해당 학번 학생의 총점 value값을 업데이트
                self.__stddataList[std_id].update(
                    평균=int(self.__stddataList[std_id].get('총점')) // 3)  # 해당 학번 학생의 평균 value값을 업데이트
 
        else:
            print('해당 학번 학생이 존재하지 않습니다')
 
    def printInfo(self):
        opt2 = input('1. 등록순  2. 총점순  3. 이름순 : ')
        if opt2 == '1':
            stm.dispALLInfo()
        elif opt2 == '2':
            temp = copy.deepcopy(self.__stddataList)  # deep copy
            temp = sorted(temp.items(), key=lambda x: x[1].get('총점'), reverse=True)  # 총점 기준으로 내림차순
            stm.dispInfoTp(*temp)
        elif opt2 == '3':
            temp = copy.deepcopy(self.__stddataList)
            temp = sorted(temp.items(), key=lambda x: x[1].get('이름'), reverse=False)
            stm.dispInfoTp(*temp)
        else:
            print('형식에 맞게 입력하시오')
 
    def start(self):
        while True:
            print()
            userChoic = input('1.입력  2.출력  3.검색  4.삭제  5.수정  6.종료   ==>    ')
            if userChoic == '1':
                stm.insertinfo()
            elif userChoic == '2':
                stm.printInfo()
            elif userChoic == '3':
                stm.searchInfo()
            elif userChoic == '4':
                stm.deleteInfo(input('삭제할 학번을 입력하시오  ==>  '))
            elif userChoic == '5':
                stm.modifyInfo()
            elif userChoic == '6':
                stm.dbDump()
                exit()
            else:
                print('형식에 맞게 입력하세요')
 
 
# 메인
stm = studentManagement()
stm.start()