fix api

app/chain/douban.py

@@ -1,15 +1,18 @@
 from pathlib import Path
-from typing import Optional, Union
+from typing import Optional, List
+from typing import Union
 
 from app.chain import ChainBase
 from app.chain.download import DownloadChain
 from app.chain.search import SearchChain
 from app.chain.subscribe import SubscribeChain
 from app.core.config import settings
-from app.core.metainfo import MetaInfo
+from app.core.context import Context
 from app.core.context import MediaInfo
+from app.core.metainfo import MetaInfo
 from app.helper.rss import RssHelper
 from app.log import logger
+from app.schemas import MediaType
 
 
 class DoubanChain(ChainBase):
@@ -28,6 +31,48 @@ class DoubanChain(ChainBase):
         self.searchchain = SearchChain()
         self.subscribechain = SubscribeChain()
 
+    def recognize_by_doubanid(self, doubanid: str) -> Optional[Context]:
+        """
+        根据豆瓣ID识别媒体信息
+        """
+        logger.info(f'开始识别媒体信息，豆瓣ID：{doubanid} ...')
+        # 查询豆瓣信息
+        doubaninfo = self.douban_info(doubanid=doubanid)
+        if not doubaninfo:
+            logger.warn(f'未查询到豆瓣信息，豆瓣ID：{doubanid}')
+            return None
+        meta = MetaInfo(title=doubaninfo.get("original_title") or doubaninfo.get("title"))
+        # 识别媒体信息
+        mediainfo: MediaInfo = self.recognize_media(meta=meta)
+        if not mediainfo:
+            logger.warn(f'{meta.name} 未识别到TMDB媒体信息')
+            return Context(meta=meta, mediainfo=MediaInfo(douban_info=doubaninfo))
+        logger.info(f'{doubanid} 识别到媒体信息：{mediainfo.type.value} {mediainfo.title_year}{meta.season}')
+        mediainfo.set_douban_info(doubaninfo)
+        return Context(meta=meta, mediainfo=mediainfo)
+
+    def douban_movies(self, sort: str, tags: str, start: int = 0, count: int = 30) -> List[MediaInfo]:
+        """
+        浏览豆瓣电影列表
+        """
+        logger.info(f'开始获取豆瓣电影列表，排序：{sort}，标签：{tags}')
+        movies = self.douban_discover(mtype=MediaType.MOVIE, sort=sort, tags=tags, start=start, count=count)
+        if not movies:
+            logger.warn(f'豆瓣电影列表为空，排序：{sort}，标签：{tags}')
+            return []
+        return [MediaInfo(douban_info=movie) for movie in movies]
+
+    def douban_tvs(self, sort: str, tags: str, start: int = 0, count: int = 30) -> List[MediaInfo]:
+        """
+        浏览豆瓣剧集列表
+        """
+        logger.info(f'开始获取豆瓣剧集列表，排序：{sort}，标签：{tags}')
+        tvs = self.douban_discover(mtype=MediaType.TV, sort=sort, tags=tags, start=start, count=count)
+        if not tvs:
+            logger.warn(f'豆瓣剧集列表为空，排序：{sort}，标签：{tags}')
+            return []
+        return [MediaInfo(douban_info=tv) for tv in tvs]
+
     def remote_sync(self, userid: Union[int, str]):
         """
         同步豆瓣想看数据，发送消息