Merge pull request 'develop' (#15) from develop into prism_plus_support

Reviewed-on: https://gitea.tendokyu.moe/SoulGateKey/artemis/pulls/15

titles/mai2/schema/item.py

@@ -1,7 +1,8 @@
+from collections.abc import Iterable
 from datetime import datetime
 from typing import Dict, List, Optional
 
-from sqlalchemy import Column, Table, UniqueConstraint, and_, or_
+from sqlalchemy import Column, Table, UniqueConstraint, and_, or_, not_
 from sqlalchemy.dialects.mysql import insert
 from sqlalchemy.engine import Row
 from sqlalchemy.schema import ForeignKey
@@ -550,25 +551,36 @@ class Mai2ItemData(BaseData):
         if result:
             return result.fetchall()
 
-    async def add_fav_music(self, user_id: int, music_id: int, order_id: Optional[int] = None) -> Optional[int]:
-        sql = insert(fav_music).values(
-            user = user_id,
-            musicId = music_id,
-            orderId = order_id
-        )
+    async def put_fav_music(self, user_id: int, fav_list: Iterable[tuple[int, Optional[int]]]) -> Optional[int]:
+        row_count = 0
+        processed_music_ids = []
+
+        for music_id, order_id in fav_list:
+            sql = insert(fav_music).values(
+                user = user_id,
+                musicId = music_id,
+                orderId = order_id
+            )
+            
+            conflict = sql.on_duplicate_key_update(orderId = order_id)
+            result = await self.execute(conflict)
+
+            processed_music_ids.append(music_id)
+
+            if not result:
+                self.logger.error(f"Failed to add music {music_id} as favorite for user {user_id}!")
+                continue
+
+            row_count += result.rowcount
+
+        clear_stale_entries_stmt = fav_music.delete(and_(fav_music.c.user == user_id, not_(fav_music.c.musicId.in_(processed_music_ids))))
+        result = await self.execute(clear_stale_entries_stmt)
+
+        if result is None:
+            self.logger.error(f"Failed to clear stale favorite music entries for user {user_id}!")
+            return None
         
-        conflict = sql.on_duplicate_key_update(orderId = order_id)
-        
-        result = await self.execute(conflict)
-        if result:
-            return result.lastrowid
-        
-        self.logger.error(f"Failed to add music {music_id} as favorite for user {user_id}!")
-        
-    async def remove_fav_music(self, user_id: int, music_id: int) -> None:
-        result = await self.execute(fav_music.delete(and_(fav_music.c.user == user_id, fav_music.c.musicId == music_id)))
-        if not result:
-            self.logger.error(f"Failed to remove music {music_id} as favorite for user {user_id}!")
+        return row_count + result.rowcount
 
     async def put_card(
         self,

titles/mai2/schema/static.py

@@ -254,7 +254,7 @@ class Mai2StaticData(BaseData):
 
     async def put_card(self, version: int, card_id: int, card_name: str, opt_id: int = None, **card_data) -> int:
         sql = insert(cards).values(
-            version=version, cardId=card_id, cardName=card_name, opt=coalesce(cards.c.opt, opt_id) **card_data
+            version=version, cardId=card_id, cardName=card_name, opt=coalesce(cards.c.opt, opt_id), **card_data
         )
 
         conflict = sql.on_duplicate_key_update(opt=coalesce(cards.c.opt, opt_id), **card_data)