优化 transfer.py 中任务处理逻辑,增强错误信息反馈

This commit is contained in:
jxxghp
2026-01-21 23:55:20 +08:00
parent 90c86c056c
commit bbe831a57c
+27 -17
View File
@@ -219,7 +219,7 @@ class JobManager:
def remove_job(self, task: TransferTask) -> Optional[TransferJob]:
"""
移除作业
移除任务对应的作业
"""
with job_lock:
__mediaid__ = self.__get_media_id(media=task.mediainfo, season=task.meta.begin_season)
@@ -470,7 +470,12 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
整理完成后处理
"""
def __all_finished():
# 状态
ret_status = True
# 错误信息
ret_message = ""
def __notify():
"""
完成时发送消息、刮削事件、移除任务等
"""
@@ -492,6 +497,7 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
se_str = f"{task.meta.season} {StringUtils.format_ep(season_episodes)}"
else:
se_str = f"{task.meta.season}"
# 发送入库成功消息
self.send_transfer_message(meta=task.meta,
mediainfo=task.mediainfo,
transferinfo=transferinfo,
@@ -508,12 +514,10 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
'overwrite': False
})
# 移除已完成的任务
self.jobview.remove_job(task)
transferhis = TransferHistoryOper()
if not transferinfo.success:
# 转移失败
if not transferinfo.success:
logger.warn(f"{task.fileitem.name} 入库失败:{transferinfo.message}")
# 新增转移失败历史记录
@@ -548,19 +552,20 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
link=settings.MP_DOMAIN('#/history')
))
# 整理失败
# 设置任务失败
self.jobview.fail_task(task)
# 全部整理完成且有成功的任务时
if self.jobview.is_finished(task):
# 发送消息、刮削事件、移除任务
__all_finished()
# 发送消息、刮削事件
__notify()
return False, transferinfo.message
# task转移成功
self.jobview.finish_task(task)
# 返回失败
ret_status = False
ret_message = transferinfo.message
else:
# 转移成功
logger.info(f"{task.fileitem.name} 入库成功:{transferinfo.target_diritem.path}")
# 新增task转移成功历史记录
@@ -594,6 +599,9 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
else:
self._success_target_files[target_dir_path] = target_files
# 设置任务成功
self.jobview.finish_task(task)
# 全部整理成功时
if self.jobview.is_success(task):
# 移动模式删除空目录
@@ -612,19 +620,21 @@ class TransferChain(ChainBase, ConfigReloadMixin, metaclass=Singleton):
# 全部整理完成且有成功的任务时
if self.jobview.is_finished(task):
# 发送消息、刮削事件、移除任务
__all_finished()
# 发送消息、刮削事件
__notify()
# 全部整理完成不管成功还是失败
if self.jobview.is_done(task):
# 所有任务
# 查询作业中的所有任务
tasks = self.jobview.all_tasks(task.mediainfo, task.meta.begin_season)
for t in tasks:
if t.download_hash:
# 设置种子状态为已整理
self.transfer_completed(hashs=t.download_hash, downloader=t.downloader)
# 清理作业
self.jobview.remove_job(task)
return True, ""
return ret_status, ret_message
def put_to_queue(self, task: TransferTask):
"""