diff --git a/db/database_manager.py b/db/database_manager.py index 939cc2e..55b16c2 100644 --- a/db/database_manager.py +++ b/db/database_manager.py @@ -124,3 +124,22 @@ WHERE """ data = (student.student_name, student.student_number, student.user_id, student.major_id, student.class_name) return self.execute(sql, data) + + def get_course_name(self, student_number, day_of_week): + # 从student表获取class_name + sql_student = "SELECT class_name FROM student WHERE student_number = %s;" + class_name = self.fetch(sql_student, (student_number,))[0]['class_name'] + + # 使用class_name和day_of_week从schedule表获取course_id + sql_schedule = "SELECT course_id, FROM schedule WHERE day_of_week = %s AND class_name = %s;" + course_id_list = self.fetch(sql_schedule, (day_of_week, class_name)) + + # 对于每一个course_id,从course表中查询course_name + course_names = [] + for course_id in course_id_list: + sql_course = "SELECT course_name FROM course WHERE course_id = %s;" + course_name = self.fetch(sql_course, (course_id['course_id'],)) + if course_name: + course_names.extend([cn['course_name'] for cn in course_name]) + + return course_names