day28作业

import os
import uuid
import pickle
from conf import settings

class School:

    def __init__(self,name,addr):
        self.name = name
        self.addr = addr
        self.uuid = str(uuid.uuid4())
        self.grades = {}
        self.save()
        self.save_name_uuid()

    def create_grade(self):
        # 创建班级对象
        name = input("请输入班级名称：").strip()
        grade_obj = Grade(name)
        grade_obj.save()
        # 将班级名称与uuid存字典当成学校的属性。
        self.grades[grade_obj.name] = grade_obj.uuid
        self.save()
        print("创建成功。")

    def save(self):
        file_path = os.path.join(settings.SCHOOL_PATH,str(self.uuid))
        with open(file_path,"wb") as f:
            pickle.dump(self,f)

    def save_name_uuid(self):
        '''将所有学校和对应的uuid保存成字典存放到Info文件中。'''
        try:
            with open(settings.INFO_PATH,"rb") as f:
                school_dict = pickle.load(f)
        except EOFError:
            school_dict = {}
            # school_dict = eval(f.read())
        school_dict[self.name] = self.uuid
        with open(settings.INFO_PATH,"wb") as f:
            pickle.dump(school_dict,f)

class Grade:

    def __init__(self,name):
        self.name = name
        self.uuid = uuid.uuid4()
        self.course = None
        self.students = {}

    def create_course(self):
        name = input("请输入课程名称：").strip()
        period = input("请输入课程周期：").strip()
        price = input("请输入课程价格：").strip()
        course_obj = Course(name,period,price)
        self.course = course_obj.uuid
        self.save()
        print("创建课程成功")

    def save(self):
        file_path = os.path.join(settings.GRADE_PATH, str(self.uuid))
        with open(file_path, "wb") as f:
            pickle.dump(self, f)

class Course:

    def __init__(self,name,period,price):
        self.name = name
        self.period = period
        self.price = price
        self.uuid = uuid.uuid4()

    def save(self):
        file_path = os.path.join(settings.COURSE_PATH, str(self.uuid))
        with open(file_path, "wb") as f:
            pickle.dump(self, f)

class Student:

    def __init__(self,name):
        self.name = name
        self.uuid = uuid.uuid4()
        self.grade = None

    def choice_grade(self):

        with open(settings.INFO_PATH,"rb") as f:
            schools = pickle.load(f)
        while True:
            print("选课".center(40,"="))
            schools_list = list(schools)
            for k,school in enumerate(schools,0):
                print(f"编号：{k}  学校名：{school}")
            choice = input("请输入选择的学校编号：").strip()
            school_uuid = schools[schools_list[int(choice)]]
            school_path = os.path.join(settings.SCHOOL_PATH,school_uuid)
            with open(school_path,"rb") as f:
                school_obj = pickle.load(f)
            grades = list(school_obj.grades)
            for k,grade in enumerate(grades,0):
                print(f"编号：{k}   班级名：{grade}")
            choice2 = input("请输入所选班级编号：").strip()
            grade_uuid = school_obj.grades[grades[int(choice2)]]
            grade_path = os.path.join(settings.GRADE_PATH,str(grade_uuid))
            with open(grade_path,"rb") as f:
                grade_obj = pickle.load(f)
            grade_obj.students[self.name] = self.uuid
            grade_obj.save()
            self.grade = grade_obj.uuid
            print("选课成功")
            break

    def save(self):
        file_path = os.path.join(settings.STUDENT_PATH, str(self.uuid))
        with open(file_path, "wb") as f:
            pickle.dump(self, f)

# 创建两个学校
school_1 = School("魔都校区","魔都飞车站")
school_2 = School("帝都校区","帝都飞盘站")
# 创建班级
school_1.create_grade()
school_1.create_grade()

# 学生选课
stu_1 = Student("tom")
stu_1.choice_grade()