mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-03-21 04:27:48 +08:00
825fc35134
- 在 `types` 里，针对各个模块的类型进行子级分类。 - 为每个模块统一添加 `get_subtype` 方法，这样一来，能更精准快速地区分与调用子类的每个模块，又能获取 ModuleType 所规定的分类以及对应存在的子模块类型支持列表，从而有效解决当下调用时需繁琐遍历每个 module 以获取 get_name 或 _channel 的问题。 - 解决因消息渠道前端返回所保存的 type 与后端规定值不一致，而需要频繁调用 _channel 私有方法才能获取分类所可能产生的问题。
73 lines
2.1 KiB
Python
73 lines
2.1 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 tvdbapi
|
||
from app.schemas.types import ModuleType, MediaRecognizeType
|
||
from app.utils.http import RequestUtils
|
||
|
||
|
||
class TheTvDbModule(_ModuleBase):
|
||
tvdb: tvdbapi.Tvdb = None
|
||
|
||
def init_module(self) -> None:
|
||
self.tvdb = tvdbapi.Tvdb(apikey=settings.TVDB_API_KEY,
|
||
cache=False,
|
||
select_first=True,
|
||
proxies=settings.PROXY)
|
||
|
||
@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):
|
||
self.tvdb.close()
|
||
|
||
def test(self) -> Tuple[bool, str]:
|
||
"""
|
||
测试模块连接性
|
||
"""
|
||
ret = RequestUtils(proxies=settings.PROXY).get_res("https://api.thetvdb.com/series/81189")
|
||
if ret and ret.status_code == 200:
|
||
return True, ""
|
||
elif ret:
|
||
return False, f"无法连接 api.thetvdb.com，错误码：{ret.status_code}"
|
||
return False, "api.thetvdb.com 网络连接失败"
|
||
|
||
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.tvdb[tvdbid].data
|
||
except Exception as err:
|
||
logger.error(f"获取TVDB信息失败: {str(err)}")
|