def regist_student(self):
student_name = input('\033[34;1m输入姓名:\033[0m').strip()
for key in self.school_db:
print('\033[32;1m学校名【%s】\033[0m' % key)
school_name = input('\033[34;1m选择学校名:\033[0m').strip()
if school_name in self.school_db:
self.school_name = school_name
self.school = self.school_db[school_name]
for key in self.school.school_grade:
self.grade_obj = self.school.school_grade[key]
self.grade_obj.cat_grade()
grade_name = input('\033[34;1m输入选择的班级名:\033[0m').strip()
if grade_name in self.school.school_grade:
self.grade = self.school.school_grade[grade_name]
print('\033[32;1m课程【%s】的价格为【%s元】\033[0m' %
(self.grade.course_obj.course_name,
self.grade.course_obj.course_price))
if_pay = input('\033[33;1m是否支付当前费用[y]:').strip()
if if_pay == 'y':
self.grade.add_student(student_name)
self.school_db[self.school_name] = self.school
file_oper(main_db_file, 'wb', self.school_db)
print('\033[35;1m选课成功.\033[0m')
def add_grade(self):
while True:
if self.school.school_grade:
for key in self.school.school_grade:
grade_obj = self.school.school_grade[key]
grade_obj.cat_grade()
if_cont = input('是否创建班级[y/b]:').strip()
if if_cont == 'y':
grade_name = input('\033[34;1m输入班级名:\033[0m').strip()
course_name = input('\033[34;1m输入班级要上的课程:\033[0m').strip()
if course_name in self.school.school_course:
course_obj = self.school.school_course[course_name]
if grade_name not in self.school.school_grade:
grade = Grade(grade_name, course_obj)
if self.school.school_teacher:
for key in self.school.school_teacher:
self.teacher = self.school.school_teacher[key]
if self.teacher.course_obj.course_name == grade.course_obj.course_name:
self.teacher.add_grade(grade_name, grade)
data = {
self.teacher.teacher_name: self.teacher
}
file_oper(teacher_db_file, 'wb', data)
self.school.create_grade(grade_name, grade)
self.school_db[self.school_name] = self.school
file_oper(main_db_file, 'wb', self.school_db)
else:
print('\033[31;1m讲师信息不存在，请先招聘讲师.\033[0m')
else:
print('\033[31;1m班级信息已存在。\033[0m')
else:
print('\033[31;1m错误：课程不存在，请先创建课程.\033[0m')
elif if_cont == 'b':
break
def init_database(self):
if not os.path.exists(main_db_file):
sh = School('上海', '上海市')
bj = School('北京', '北京市')
data = {'上海': sh, '北京': bj}
file_oper(main_db_file, 'wb', data)
if not os.path.exists(teacher_db_file):
data = {}
file_oper(teacher_db_file, 'wb', data)
def run(self):
while True:
self.school_db = file_oper(main_db_file, 'rb')
choice = option(Student_center.student_list)
if choice == '1':
self.regist_student()
else:
break
def add_course(self):
while True:
if self.school.school_course:
for key in self.school.school_course:
course_obj = self.school.school_course[key]
course_obj.cat_course()
if_cont = input('是否创建课程[y/b]:').strip()
if if_cont == 'y':
course_name = input('\033[34;1m输入课程名:\033[0m').strip()
if course_name not in self.school.school_course:
course_price = input('\033[34;1m输入课程价格:\033[0m').strip()
course_time = input('\033[34;1m输入课程周期:\033[0m').strip()
self.school.create_course(course_name, course_price,
course_time)
self.school_db[self.school_name] = self.school
file_oper(main_db_file, 'wb', self.school_db)
else:
print('\033[31;1m课程【%s】已存在.\033[0m' % course_name)
elif if_cont == 'b':
break
def run(self):
Flag = True
while Flag:
self.school_db = file_oper(main_db_file, 'rb')
if self.school_db:
for key in self.school_db:
print('学校名【%s】' % key)
school_name = input('\033[34;1m选择学校名:\033[0m').strip()
if school_name in self.school_db:
self.school_name = school_name
self.school = self.school_db[school_name]
while Flag:
choice = option(School_center.school_list)
self.school_db = file_oper(main_db_file, 'rb')
if choice == '1':
self.add_course()
elif choice == '2':
self.add_teacher()
elif choice == '3':
self.add_grade()
elif choice == '4':
Flag = False
def add_teacher(self):
while True:
if self.school.school_teacher:
for key in self.school.school_teacher:
teacher_obj = self.school.school_teacher[key]
teacher_obj.cat_teacher()
if_cont = input('是否招聘讲师[y/b]:').strip()
if if_cont == 'y':
teacher_name = input('\033[34;1m输入讲师名:\033[0m').strip()
course_name = input('\033[34;1m输入讲师要上的课程名:\033[0m').strip()
# grade_name = input('\033[34;1m输入讲师要关联的班级:\033[0m').strip()
if course_name in self.school.school_course:
course_obj = self.school.school_course[course_name]
if teacher_name not in self.school.school_teacher:
self.school.hire_teacher(teacher_name, course_obj)
self.school_db[self.school_name] = self.school
file_oper(main_db_file, 'wb', self.school_db)
else:
print('\033[31;1m讲师信息已存在.\033[0m')
else:
print('\033[31;1m错误：课程不存在，请先创建课程.\033[0m')
elif if_cont == 'b':
break
def __init__(self):
self.school_db = file_oper(main_db_file, 'rb')
self.teacher_db = file_oper(teacher_db_file, 'rb')
self.run()