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

app/db/models/mediaserver.py

@@ -0,0 +1,61 @@
+from datetime import datetime
+
+from sqlalchemy import Column, Integer, String, Sequence
+from sqlalchemy.orm import Session
+
+from app.db.models import Base
+
+
+class MediaServerItem(Base):
+    """
+    站点表
+    """
+    id = Column(Integer, Sequence('id'), primary_key=True, index=True)
+    # 服务器类型
+    server = Column(String)
+    # 媒体库ID
+    library = Column(String)
+    # ID
+    item_id = Column(String, index=True)
+    # 类型
+    item_type = Column(String)
+    # 标题
+    title = Column(String, index=True)
+    # 原标题
+    original_title = Column(String)
+    # 年份
+    year = Column(String)
+    # TMDBID
+    tmdbid = Column(Integer, index=True)
+    # IMDBID
+    imdbid = Column(String, index=True)
+    # TVDBID
+    tvdbid = Column(String, index=True)
+    # 路径
+    path = Column(String)
+    # 季集
+    seasoninfo = Column(String)
+    # 备注
+    note = Column(String)
+    # 同步时间
+    lst_mod_date = Column(String, default=datetime.now().strftime("%Y-%m-%d %H:%M:%S"))
+
+    @staticmethod
+    def get_by_itemid(db: Session, item_id: str):
+        return db.query(MediaServerItem).filter(MediaServerItem.item_id == item_id).first()
+
+    @staticmethod
+    def empty(db: Session, server: str):
+        db.query(MediaServerItem).filter(MediaServerItem.server == server).delete()
+        db.commit()
+
+    @staticmethod
+    def exist_by_tmdbid(db: Session, tmdbid: int, mtype: str):
+        return db.query(MediaServerItem).filter(MediaServerItem.tmdbid == tmdbid,
+                                                MediaServerItem.item_type == mtype).first()
+
+    @staticmethod
+    def exists_by_title(db: Session, title: str, mtype: str, year: str):
+        return db.query(MediaServerItem).filter(MediaServerItem.title == title,
+                                                MediaServerItem.item_type == mtype,
+                                                MediaServerItem.year == str(year)).first()