mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-09-08 08:57:09 +08:00
优化 transfer.py 中任务处理逻辑,增强错误信息反馈
This commit is contained in:
+27
-17
@@ -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):
|
||||
"""
|
||||
|
||||
Reference in New Issue
Block a user