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: if not mediainfo:
logger.error(f'媒体信息识别失败!') logger.error(f'媒体信息识别失败!')
return [] return []
# 缺失的媒体信息 # 缺失的季集
if no_exists and no_exists.get(mediainfo.tmdb_id): if no_exists and no_exists.get(mediainfo.tmdb_id):
# 过滤剧集 # 过滤剧集
season_episodes = {sea: info.episodes season_episodes = {sea: info.episodes
for sea, info in no_exists[mediainfo.tmdb_id].items()} for sea, info in no_exists[mediainfo.tmdb_id].items()}
else: else:
season_episodes = None 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( torrents: List[TorrentInfo] = []
mediainfo=mediainfo, for keyword in keywords:
keyword=keyword, torrents = self.__search_all_sites(
sites=sites mediainfo=mediainfo,
) keyword=keyword,
sites=sites
)
if torrents:
break
if not torrents: if not torrents:
logger.warn(f'{keyword or mediainfo.title} 未搜索到资源') logger.warn(f'{keyword or mediainfo.title} 未搜索到资源')
return [] return []

View File

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