init

app/modules/douban/__init__.py

@@ -0,0 +1,115 @@
+from typing import List, Optional, Tuple, Union
+
+from app.core import MediaInfo, settings
+from app.core.meta import MetaBase
+from app.modules import _ModuleBase
+from app.modules.douban.apiv2 import DoubanApi
+from app.utils.types import MediaType
+
+
+class Douban(_ModuleBase):
+
+    def __init__(self):
+        super().__init__()
+        self.doubanapi = DoubanApi()
+
+    def init_module(self) -> None:
+        pass
+
+    def init_setting(self) -> Tuple[str, Union[str, bool]]:
+        pass
+
+    def douban_info(self, doubanid: str) -> Optional[dict]:
+        """
+        获取豆瓣信息
+        :param doubanid: 豆瓣ID
+        :return: 识别的媒体信息，包括剧集信息
+        """
+        if not doubanid:
+            return None
+        douban_info = self.doubanapi.movie_detail(doubanid)
+        if douban_info:
+            celebrities = self.doubanapi.movie_celebrities(doubanid)
+            if celebrities:
+                douban_info["directors"] = celebrities.get("directors")
+                douban_info["actors"] = celebrities.get("actors")
+        else:
+            douban_info = self.doubanapi.tv_detail(doubanid)
+            celebrities = self.doubanapi.tv_celebrities(doubanid)
+            if douban_info and celebrities:
+                douban_info["directors"] = celebrities.get("directors")
+                douban_info["actors"] = celebrities.get("actors")
+        return self.__extend_doubaninfo(douban_info)
+
+    @staticmethod
+    def __extend_doubaninfo(doubaninfo: dict):
+        """
+        补充添加豆瓣信息
+        """
+        # 类型
+        if doubaninfo.get("type") == "movie":
+            doubaninfo['media_type'] = MediaType.MOVIE
+        elif doubaninfo.get("type") == "tv":
+            doubaninfo['media_type'] = MediaType.TV
+        else:
+            return doubaninfo
+        # 评分
+        rating = doubaninfo.get('rating')
+        if rating:
+            doubaninfo['vote_average'] = float(rating.get("value"))
+        else:
+            doubaninfo['vote_average'] = 0
+
+        # 海报
+        if doubaninfo.get("type") == "movie":
+            poster_path = doubaninfo.get('cover', {}).get("url")
+            if not poster_path:
+                poster_path = doubaninfo.get('cover_url')
+            if not poster_path:
+                poster_path = doubaninfo.get('pic', {}).get("large")
+        else:
+            poster_path = doubaninfo.get('pic', {}).get("normal")
+        if poster_path:
+            poster_path = poster_path.replace("s_ratio_poster", "m_ratio_poster")
+        doubaninfo['poster_path'] = poster_path
+
+        # 简介
+        doubaninfo['overview'] = doubaninfo.get("card_subtitle") or ""
+
+        return doubaninfo
+
+    def search_medias(self, meta: MetaBase) -> Optional[List[MediaInfo]]:
+        """
+        搜索媒体信息
+        :param meta:  识别的元数据
+        :reutrn: 媒体信息
+        """
+        # 未启用豆瓣搜索时返回None
+        if settings.SEARCH_SOURCE != "douban":
+            return None
+
+        if not meta.get_name():
+            return []
+        result = self.doubanapi.search(meta.get_name())
+        if not result:
+            return []
+        # 返回数据
+        ret_medias = []
+        for item_obj in result.get("items"):
+            if meta.type and meta.type.value != item_obj.get("type_name"):
+                continue
+            if item_obj.get("type_name") not in (MediaType.TV.value, MediaType.MOVIE.value):
+                continue
+            ret_medias.append(MediaInfo(douban_info=item_obj.get("target")))
+
+        return ret_medias
+
+    def scrape_metadata(self, path: str, mediainfo: MediaInfo) -> None:
+        """
+        TODO 刮削元数据
+        :param path: 媒体文件路径
+        :param mediainfo:  识别的媒体信息
+        :return: 成功或失败
+        """
+        if settings.SCRAP_SOURCE != "douban":
+            return None