feat 中英文名搜索

This commit is contained in:
jxxghp 2023-08-05 07:21:20 +08:00
parent 0a2d9913e3
commit 95360acea8
2 changed files with 18 additions and 6 deletions

View File

@ -105,19 +105,30 @@ class SearchChain(ChainBase):
if not mediainfo:
logger.error(f'媒体信息识别失败!')
return []
# 缺失的媒体信息
# 缺失的季集
if no_exists and no_exists.get(mediainfo.tmdb_id):
# 过滤剧集
season_episodes = {sea: info.episodes
for sea, info in no_exists[mediainfo.tmdb_id].items()}
else:
season_episodes = None
# 搜索关键词
if keyword:
keywords = [keyword]
elif mediainfo.title != mediainfo.original_title:
keywords = [mediainfo.title, mediainfo.original_title]
else:
keywords = [mediainfo.title]
# 执行搜索
torrents: List[TorrentInfo] = self.__search_all_sites(
mediainfo=mediainfo,
keyword=keyword,
sites=sites
)
torrents: List[TorrentInfo] = []
for keyword in keywords:
torrents = self.__search_all_sites(
mediainfo=mediainfo,
keyword=keyword,
sites=sites
)
if torrents:
break
if not torrents:
logger.warn(f'{keyword or mediainfo.title} 未搜索到资源')
return []

View File

@ -48,6 +48,7 @@ class IndexerModule(_ModuleBase):
and site.get('language') == "en" \
and StringUtils.is_chinese(search_word):
# 不支持中文
logger.warn(f"{site.get('name')} 不支持中文搜索")
return []
# 开始索引