fix transfer count

This commit is contained in:
jxxghp
2024-12-26 19:25:23 +08:00
parent 419f2e90ce
commit 15cc02b083

View File

@@ -216,6 +216,17 @@ class JobManager:
with job_lock:
return all([job["state"] == "completed" for job in self._job_view[__mediaid__]["tasks"]])
def count(self, media: MediaInfo, season: int = None) -> int:
"""
获取某项任务总数
"""
__mediaid__ = self.__get_id(media=media, season=season)
with job_lock:
# 计算状态为完成的任务数
if __mediaid__ not in self._job_view:
return 0
return len([task for task in self._job_view[__mediaid__]["tasks"] if task["state"] == "completed"])
def total(self) -> int:
"""
获取所有task任务总数
@@ -356,6 +367,8 @@ class TransferChain(ChainBase, metaclass=Singleton):
se_str = f"{task.meta.season} {StringUtils.format_ep(season_episodes)}"
else:
se_str = f"{task.meta.season}"
# 更新文件数量
transferinfo.file_count = self.jobview.count(task.mediainfo, task.meta.begin_season) or 1
self.send_transfer_message(meta=task.meta,
mediainfo=task.mediainfo,
transferinfo=transferinfo,