Implmented course_info query method

titles/idac/schema/item.py

@@ -764,8 +764,8 @@ class IDACItemData(BaseData):
         return result.fetchall()
 
     def get_vs_info_by_mode(self, aime_id: int, battle_mode: int) -> Optional[List[Row]]:
-        sql = select(vs_info_new).where(
-            and_(vs_info_new.c.user == aime_id, vs_info_new.c.battle_mode == battle_mode)
+        sql = select(vs_info).where(
+            and_(vs_info.c.user == aime_id, vs_info.c.battle_mode == battle_mode)
         )
 
         result = self.execute(sql)
@@ -773,6 +773,21 @@ class IDACItemData(BaseData):
             return None
         return result.fetchone()
 
+    #This method returns a list of course_info
+    def get_vs_course_infos_by_mode(self, aime_id: int, battle_mode: int) -> Optional[List[Row]]:
+        sql = select(vs_course_info).where(vs_info.c.user == aime_id, vs_info.c.battle_mode == battle_mode)
+
+        result = self.execute(sql)
+        if result is None:
+            return None
+            
+        d, a = {}, []
+        for row in result:
+            for column, value in row.items():
+                d = {**d, **{column: value}}
+            a.append(d)
+        return a
+
     def get_stamps(self, aime_id: int) -> Optional[List[Row]]:
         sql = select(stamp).where(
             and_(