feat 本地存在标识&媒体库同步

app/db/mediaserver_oper.py

@@ -0,0 +1,54 @@
+import json
+from typing import Optional
+
+from app.db import DbOper, SessionLocal
+from app.db.models.mediaserver import MediaServerItem
+
+
+class MediaServerOper(DbOper):
+    """
+    媒体服务器数据管理
+    """
+
+    def __init__(self, db=SessionLocal()):
+        super().__init__(db)
+
+    def add(self, **kwargs) -> bool:
+        """
+        新增媒体服务器数据
+        """
+        item = MediaServerItem(**kwargs)
+        if not item.get_by_itemid(self._db, kwargs.get("item_id")):
+            item.create(self._db)
+            return True
+        return False
+
+    def empty(self, server: str):
+        """
+        清空媒体服务器数据
+        """
+        MediaServerItem.empty(self._db, server)
+
+    def exists(self, **kwargs) -> Optional[MediaServerItem]:
+        """
+        判断媒体服务器数据是否存在
+        """
+        if kwargs.get("tmdbid"):
+            # 优先按TMDBID查
+            item = MediaServerItem.exist_by_tmdbid(self._db, tmdbid=kwargs.get("tmdbid"),
+                                                   mtype=kwargs.get("mtype"))
+        else:
+            # 按标题、类型、年份查
+            item = MediaServerItem.exists_by_title(self._db, title=kwargs.get("title"),
+                                                   mtype=kwargs.get("mtype"), year=kwargs.get("year"))
+        if not item:
+            return None
+
+        if kwargs.get("season"):
+            # 判断季是否存在
+            if not item.seasoninfo:
+                return None
+            seasoninfo = json.loads(item.seasoninfo) or {}
+            if kwargs.get("season") not in seasoninfo.keys():
+                return None
+        return item