refactor(db): 修复异步连接池无界增长，并完成 SQLAlchemy 2.0 迁移与分层归位 (#6320)

app/db/oper/mediaserver.py

@@ -0,0 +1,152 @@
+from typing import Optional
+
+from sqlalchemy.orm import Session
+
+from app.db import DbOper
+from app.db.models.mediaserver import MediaServerItem
+
+
+class MediaServerOper(DbOper):
+    """
+    媒体服务器数据管理
+    """
+
+    def __init__(self, db: Optional[Session] = None):
+        super().__init__(db)
+
+    @staticmethod
+    def __prepare_payload(kwargs: dict) -> dict:
+        """
+        过滤数据库模型不存在或不应由远端覆盖的字段
+        """
+        return {
+            k: v for k, v in kwargs.items()
+            if hasattr(MediaServerItem, k) and k != "id"
+        }
+
+    def add(self, **kwargs) -> bool:
+        """
+        新增媒体服务器数据
+        """
+        kwargs = self.__prepare_payload(kwargs)
+        server = kwargs.get("server")
+        item_id = kwargs.get("item_id")
+        if not server or not item_id:
+            return False
+        item = MediaServerItem(**kwargs)
+        if not item.get_by_server_itemid(self._db, server, item_id):
+            item.create(self._db)
+            return True
+        return False
+
+    def upsert(self, **kwargs) -> bool:
+        """
+        按媒体服务器和条目ID新增或更新数据
+        """
+        kwargs = self.__prepare_payload(kwargs)
+        server = kwargs.get("server")
+        item_id = kwargs.get("item_id")
+        if not server or not item_id:
+            return False
+
+        item = MediaServerItem.get_by_server_itemid(self._db, server, item_id)
+        if item:
+            item.update(self._db, kwargs)
+            return False
+
+        MediaServerItem(**kwargs).create(self._db)
+        return True
+
+    def empty(self, server: Optional[str] = None):
+        """
+        清空媒体服务器数据
+        """
+        MediaServerItem.empty(self._db, server)
+
+    def delete_stale(self, server: str, sync_time: str) -> int:
+        """
+        删除本轮同步未更新的旧数据
+        """
+        return MediaServerItem.delete_stale(self._db, server, sync_time)
+
+    def delete_excluded_servers(self, servers: list[str]) -> int:
+        """
+        删除未启用或已移除媒体服务器的数据
+        """
+        return MediaServerItem.delete_excluded_servers(self._db, servers)
+
+    def exists(self, **kwargs) -> Optional[MediaServerItem]:
+        """
+        判断媒体服务器数据是否存在
+        """
+        if kwargs.get("media_source") and kwargs.get("media_id"):
+            item = MediaServerItem.exist_by_media_identity(
+                self._db,
+                media_source=kwargs.get("media_source"),
+                media_id=kwargs.get("media_id"),
+                mtype=kwargs.get("mtype"),
+            )
+        elif kwargs.get("title"):
+            # 按标题、类型、年份查
+            item = MediaServerItem.exists_by_title(self._db, title=kwargs.get("title"),
+                                                   mtype=kwargs.get("mtype"), year=kwargs.get("year"))
+        else:
+            return None
+        if not item:
+            return None
+
+        if kwargs.get("season") is not None:
+            # 判断季是否存在
+            if not item.seasoninfo:
+                return None
+            seasoninfo = item.seasoninfo or {}
+            if kwargs.get("season") not in seasoninfo.keys():
+                return None
+        return item
+
+    async def async_exists(self, **kwargs) -> Optional[MediaServerItem]:
+        """
+        异步判断媒体服务器数据是否存在
+        """
+        if kwargs.get("media_source") and kwargs.get("media_id"):
+            item = await MediaServerItem.async_exist_by_media_identity(
+                self._db,
+                media_source=kwargs.get("media_source"),
+                media_id=kwargs.get("media_id"),
+                mtype=kwargs.get("mtype"),
+            )
+        elif kwargs.get("title"):
+            # 按标题、类型、年份查
+            item = await MediaServerItem.async_exists_by_title(self._db, title=kwargs.get("title"),
+                                                               mtype=kwargs.get("mtype"), year=kwargs.get("year"))
+        else:
+            return None
+        if not item:
+            return None
+
+        if kwargs.get("season") is not None:
+            # 判断季是否存在
+            if not item.seasoninfo:
+                return None
+            seasoninfo = item.seasoninfo or {}
+            if kwargs.get("season") not in seasoninfo.keys():
+                return None
+        return item
+
+    def get_item_id(self, **kwargs) -> Optional[str]:
+        """
+        获取媒体服务器数据ID
+        """
+        item = self.exists(**kwargs)
+        if not item:
+            return None
+        return str(item.item_id)
+
+    async def async_get_item_id(self, **kwargs) -> Optional[str]:
+        """
+        异步获取媒体服务器数据ID
+        """
+        item = await self.async_exists(**kwargs)
+        if not item:
+            return None
+        return str(item.item_id)