init

app/db/subscribes.py

@@ -0,0 +1,76 @@
+from typing import Tuple, List
+
+from sqlalchemy.orm import Session
+
+from app.core import MediaInfo
+from app.db import SessionLocal
+from app.db.models.subscribe import Subscribe
+from app.utils.types import MediaType
+
+
+class Subscribes:
+    """
+    订阅管理
+    """
+    _db: Session = None
+
+    def __init__(self, _db=SessionLocal()):
+        self._db = _db
+
+    def add(self, mediainfo: MediaInfo, **kwargs) -> Tuple[bool, str]:
+        """
+        新增订阅
+        """
+        # 总集数
+        if mediainfo.type == MediaType.TV:
+            if not kwargs.get('season'):
+                kwargs.update({
+                    'season': 1
+                })
+            if not kwargs.get('total_episode'):
+                total_episode = len(mediainfo.seasons.get(kwargs.get('season')) or [])
+                if not total_episode:
+                    return False, "未识别到总集数"
+                kwargs.update({
+                    'total_episode': total_episode
+                })
+        subscribe = Subscribe(name=mediainfo.title,
+                              year=mediainfo.year,
+                              type=mediainfo.type.value,
+                              tmdbid=mediainfo.tmdb_id,
+                              image=mediainfo.get_poster_image(),
+                              description=mediainfo.overview,
+                              **kwargs)
+        if not subscribe.exists(self._db, tmdbid=mediainfo.tmdb_id, season=kwargs.get('season')):
+            subscribe.create(self._db)
+            return True, "新增订阅成功"
+        else:
+            return False, "订阅已存在"
+
+    def get(self, sid: int) -> Subscribe:
+        """
+        获取订阅
+        """
+        return Subscribe.get(self._db, rid=sid)
+
+    def list(self, state: str = None) -> List[Subscribe]:
+        """
+        获取订阅列表
+        """
+        if state:
+            return Subscribe.get_by_state(self._db, state)
+        return Subscribe.list(self._db)
+
+    def delete(self, sid: int):
+        """
+        删除订阅
+        """
+        Subscribe.delete(self._db, rid=sid)
+
+    def update(self, sid: int, payload: dict):
+        """
+        更新订阅
+        """
+        subscribe = self.get(sid)
+        subscribe.update(self._db, payload)
+        return subscribe