116 lines
3.8 KiB
Python
116 lines
3.8 KiB
Python
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
|