nas/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-03-20 03:57:30 +08:00
Code Issues Packages Projects Releases Wiki Activity

refactor: 将 download.py 中重复的媒体识别模式选择逻辑封装进选择器函数

Browse Source
This commit is contained in:
Castell
2026-03-03 01:58:49 +08:00
parent b6b5d9f9c4
commit 28945ef153
2 changed files with 12 additions and 21 deletions
Download Patch File Download Diff File Expand all files Collapse all files

21
app/api/endpoints/download.py
View File

@@ -78,21 +78,12 @@ def add(
# 元数据
metainfo = MetaInfo(title=torrent_in.title, subtitle=torrent_in.description)
# 媒体信息
mediainfo: MediaInfo = None
plugin_available = eventmanager.check(ChainEventType.NameRecognize)
# 定义识别函数
native_recognize = lambda: MediaChain().recognize_media(meta=metainfo, tmdbid=tmdbid, doubanid=doubanid)
plugin_recognize = lambda: MediaChain().recognize_help(title=torrent_in.title, org_meta=metainfo)
if settings.RECOGNIZE_PLUGIN_FIRST and plugin_available:
# 插件优先
mediainfo = plugin_recognize()
if not mediainfo:
mediainfo = native_recognize()
else:
# 原生优先
mediainfo = native_recognize()
if not mediainfo and plugin_available:
mediainfo = plugin_recognize()
mediainfo = MediaChain().select_recognize_source(
log_name=torrent_in.title,
log_context=torrent_in.title,
native_fn=lambda: MediaChain().recognize_media(meta=metainfo, tmdbid=tmdbid, doubanid=doubanid),
plugin_fn=lambda: MediaChain().recognize_help(title=torrent_in.title, org_meta=metainfo)
)
if not mediainfo:
return schemas.Response(success=False, message="无法识别媒体信息")
# 种子信息