GithubBackup/MoviePilot
1
0
Fork 0
mirror of https://github.com/jxxghp/MoviePilot.git synced 2026-08-28 19:47:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #4339 from jtcymc/v2

Browse Source
This commit is contained in:
jxxghp
2025-05-25 08:01:00 +08:00
committed by GitHub
parent 72fce20905 677e38c62d
commit 96f603bfd1
1 changed files with 28 additions and 28 deletions
Download Patch File Download Diff File Expand all files Collapse all files
app/chain/search.py
+28 -28
View File
@@ -329,36 +329,36 @@ class SearchChain(ChainBase):
self.progress.update(value=0, self.progress.update(value=0,
text=f"开始搜索，共 {total_num} 个站点 ...", text=f"开始搜索，共 {total_num} 个站点 ...",
key=ProgressKey.Search) key=ProgressKey.Search)
# 多线程
executor = ThreadPoolExecutor(max_workers=len(indexer_sites))
all_task = []
for site in indexer_sites:
if area == "imdbid":
# 搜索IMDBID
task = executor.submit(self.search_torrents, site=site,
keywords=[mediainfo.imdb_id] if mediainfo else None,
mtype=mediainfo.type if mediainfo else None,
page=page)
else:
# 搜索标题
task = executor.submit(self.search_torrents, site=site,
keywords=keywords,
mtype=mediainfo.type if mediainfo else None,
page=page)
all_task.append(task)
# 结果集 # 结果集
results = [] results = []
for future in as_completed(all_task): # 多线程
if global_vars.is_system_stopped: with ThreadPoolExecutor(max_workers=len(indexer_sites)) as executor:
break all_task = []
finish_count += 1 for site in indexer_sites:
result = future.result() if area == "imdbid":
if result: # 搜索IMDBID
results.extend(result) task = executor.submit(self.search_torrents, site=site,
logger.info(f"站点搜索进度：{finish_count} / {total_num}") keywords=[mediainfo.imdb_id] if mediainfo else None,
self.progress.update(value=finish_count / total_num * 100, mtype=mediainfo.type if mediainfo else None,
text=f"正在搜索{keywords or ''}，已完成 {finish_count} / {total_num} 个站点 ...", page=page)
key=ProgressKey.Search) else:
# 搜索标题
task = executor.submit(self.search_torrents, site=site,
keywords=keywords,
mtype=mediainfo.type if mediainfo else None,
page=page)
all_task.append(task)
for future in as_completed(all_task):
if global_vars.is_system_stopped:
break
finish_count += 1
result = future.result()
if result:
results.extend(result)
logger.info(f"站点搜索进度：{finish_count} / {total_num}")
self.progress.update(value=finish_count / total_num * 100,
text=f"正在搜索{keywords or ''}，已完成 {finish_count} / {total_num} 个站点 ...",
key=ProgressKey.Search)
# 计算耗时 # 计算耗时
end_time = datetime.now() end_time = datetime.now()
# 更新进度 # 更新进度