From b5b5e92b73d164ef1f83e628f5a61b88df74d35d Mon Sep 17 00:00:00 2001
From: wangsiyuan <2392948297@qq.com>
Date: Fri, 29 Dec 2023 00:21:28 +0800
Subject: [PATCH] =?UTF-8?q?=E6=9B=B4=E6=96=B0=20database=5Fmanager.py?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 db/database_manager.py | 25 +++++++++++++++----------
 1 file changed, 15 insertions(+), 10 deletions(-)

diff --git a/db/database_manager.py b/db/database_manager.py
index 55b16c2..807334c 100644
--- a/db/database_manager.py
+++ b/db/database_manager.py
@@ -125,21 +125,26 @@ 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):
+    def get_course_name(self, student_number, day_of_week, period_id):
         # 从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))
+        sql_schedule = "SELECT course_id, FROM schedule WHERE day_of_week = %s AND class_name = %s AND period_id = %s;"
+        course_id = self.fetch(sql_schedule, (day_of_week, class_name, period_id))
 
         # 对于每一个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])
+        sql_course = "SELECT course_name FROM course WHERE course_id = %s;"
+        course_name = self.fetch(sql_course, (course_id['course_id'],))
+        data = {"course_name": course_name, "course_id": course_id}
+        return data
 
-        return course_names
+    def update_sign_in_info(self, student_number, course_id, course_name, date, status):
+        sql = """
+        INSERT INTO attendance_record (student_number, course_id, course_name, date, status)
+        VALUES (%s, %s, %s, %s, %s)
+        """
+        val = (student_number, course_id, course_name, date, status)
+        result = self.execute(sql, val)
+        return result