mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-20 03:57:30 +08:00
117 lines
3.5 KiB
Python
117 lines
3.5 KiB
Python
from typing import Optional, Tuple, Union
|
|
|
|
from app.core.config import settings
|
|
from app.log import logger
|
|
from app.modules import _ModuleBase
|
|
from app.modules.thetvdb import tvdb_v4_official
|
|
from app.schemas.types import ModuleType, MediaRecognizeType
|
|
|
|
|
|
class TheTvDbModule(_ModuleBase):
|
|
tvdb: tvdb_v4_official.TVDB = None
|
|
|
|
def init_module(self) -> None:
|
|
self._initialize_tvdb_session()
|
|
|
|
def _initialize_tvdb_session(self) -> None:
|
|
"""
|
|
创建或刷新 TVDB 登录会话
|
|
"""
|
|
try:
|
|
self.tvdb = tvdb_v4_official.TVDB(apikey=settings.TVDB_V4_API_KEY,
|
|
pin=settings.TVDB_V4_API_PIN,
|
|
proxy=settings.PROXY)
|
|
except Exception as e:
|
|
logger.error(f"TVDB 登录失败: {str(e)}")
|
|
|
|
def _handle_tvdb_call(self, func, *args, **kwargs):
|
|
"""
|
|
包裹 TVDB 调用，处理 token 失效情况并尝试重新初始化
|
|
"""
|
|
try:
|
|
return func(*args, **kwargs)
|
|
except ValueError as e:
|
|
# 检查错误信息中是否包含 token 失效相关描述
|
|
if "Unauthorized" in str(e):
|
|
logger.warning("TVDB Token 可能已失效，正在尝试重新登录...")
|
|
self._initialize_tvdb_session()
|
|
return func(*args, **kwargs)
|
|
elif "NotFoundException" in str(e):
|
|
logger.warning("TVDB 剧集不存在")
|
|
return None
|
|
else:
|
|
raise
|
|
except Exception as e:
|
|
logger.error(f"TVDB 调用出错: {str(e)}")
|
|
raise
|
|
|
|
@staticmethod
|
|
def get_name() -> str:
|
|
return "TheTvDb"
|
|
|
|
@staticmethod
|
|
def get_type() -> ModuleType:
|
|
"""
|
|
获取模块类型
|
|
"""
|
|
return ModuleType.MediaRecognize
|
|
|
|
@staticmethod
|
|
def get_subtype() -> MediaRecognizeType:
|
|
"""
|
|
获取模块子类型
|
|
"""
|
|
return MediaRecognizeType.TVDB
|
|
|
|
@staticmethod
|
|
def get_priority() -> int:
|
|
"""
|
|
获取模块优先级，数字越小优先级越高，只有同一接口下优先级才生效
|
|
"""
|
|
return 4
|
|
|
|
def stop(self):
|
|
pass
|
|
|
|
def test(self) -> Tuple[bool, str]:
|
|
"""
|
|
测试模块连接性
|
|
"""
|
|
if not self.tvdb:
|
|
return False, "TheTVDB 连接失败"
|
|
try:
|
|
self._handle_tvdb_call(self.tvdb.get_series, 81189)
|
|
return True, ""
|
|
except Exception as e:
|
|
return False, str(e)
|
|
|
|
def init_setting(self) -> Tuple[str, Union[str, bool]]:
|
|
pass
|
|
|
|
def tvdb_info(self, tvdbid: int) -> Optional[dict]:
|
|
"""
|
|
获取TVDB信息
|
|
:param tvdbid: int
|
|
:return: TVDB信息
|
|
"""
|
|
try:
|
|
logger.info(f"开始获取TVDB信息: {tvdbid} ...")
|
|
return self._handle_tvdb_call(self.tvdb.get_series_extended, tvdbid)
|
|
except Exception as err:
|
|
logger.error(f"获取TVDB信息失败: {str(err)}")
|
|
return None
|
|
|
|
def search_tvdb(self, title: str) -> list:
|
|
"""
|
|
用标题搜索TVDB剧集
|
|
:param title: 标题
|
|
:return: TVDB信息
|
|
"""
|
|
try:
|
|
logger.info(f"开始用标题搜索TVDB剧集: {title} ...")
|
|
res = self._handle_tvdb_call(self.tvdb.search, title)
|
|
return [item for item in res if item.get("type") == "series"]
|
|
except Exception as err:
|
|
logger.error(f"用标题搜索TVDB剧集失败: {str(err)}")
|
|
return []
|