fix dboper

app/db/subscribe_oper.py

@@ -0,0 +1,58 @@
+from typing import Tuple, List
+
+from app.core.context import MediaInfo
+from app.db import DbOper
+from app.db.models.subscribe import Subscribe
+
+
+class SubscribeOper(DbOper):
+    """
+    订阅管理
+    """
+
+    def add(self, mediainfo: MediaInfo, **kwargs) -> Tuple[int, str]:
+        """
+        新增订阅
+        """
+        subscribe = Subscribe(name=mediainfo.title,
+                              year=mediainfo.year,
+                              type=mediainfo.type.value,
+                              tmdbid=mediainfo.tmdb_id,
+                              imdbid=mediainfo.imdb_id,
+                              tvdbid=mediainfo.tvdb_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 subscribe.id, "新增订阅成功"
+        else:
+            return subscribe.id, "订阅已存在"
+
+    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