diff --git a/plugins/autosubv2/__init__.py b/plugins/autosubv2/__init__.py index 9c58d27..e812d13 100644 --- a/plugins/autosubv2/__init__.py +++ b/plugins/autosubv2/__init__.py @@ -28,6 +28,10 @@ from app.schemas.types import NotificationType # todo # 监听入库事件,自动调用翻译 +class UserInterruptException(Exception): + """用户中断当前任务的异常""" + pass + class AutoSubv2(_PluginBase): # 插件名称 plugin_name = "AI字幕自动生成(v2)" @@ -199,6 +203,7 @@ class AutoSubv2(_PluginBase): except Exception as e: logger.error(f"处理异常: {e}") + logger.error(traceback.format_exc()) finally: logger.info(f"处理完成: " f"成功{self.success_count} / 跳过{self.skip_count} / 失败{self.fail_count} / 共{self.process_count}") @@ -208,7 +213,7 @@ class AutoSubv2(_PluginBase): if self.asr_engine == 'whisper.cpp': if not self.whisper_main or not self.whisper_model: logger.warn(f"配置信息不完整,不进行处理") - return + return False if not os.path.exists(self.whisper_main): logger.warn(f"whisper.cpp主程序不存在,不进行处理") return False @@ -222,7 +227,7 @@ class AutoSubv2(_PluginBase): elif self.asr_engine == 'faster-whisper': if not self.faster_whisper_model_path or not self.faster_whisper_model: logger.warn(f"配置信息不完整,不进行处理") - return + return False if not os.path.exists(self.faster_whisper_model_path): logger.info(f"创建faster-whisper模型目录:{self.faster_whisper_model_path}") os.mkdir(self.faster_whisper_model_path) @@ -287,6 +292,9 @@ class AutoSubv2(_PluginBase): if self.send_notify: self.post_message(mtype=NotificationType.Plugin, title="【自动字幕生成】", text=message) self.success_count += 1 + except UserInterruptException: + logger.info(f"用户中断当前任务:{video_file}") + self.fail_count += 1 except Exception as e: logger.error(f"自动字幕生成 处理异常:{e}") end_time = time.time() @@ -364,7 +372,7 @@ class AutoSubv2(_PluginBase): for segment in segments: if self._event.is_set(): logger.info(f"whisper音轨转录服务停止") - raise Exception(f"用户中断当前任务") + raise UserInterruptException(f"用户中断当前任务") for word in segment.words: idx += 1 subs.append(srt.Subtitle(index=idx, @@ -376,7 +384,7 @@ class AutoSubv2(_PluginBase): for i, segment in enumerate(segments): if self._event.is_set(): logger.info(f"whisper音轨转录服务停止") - raise Exception(f"用户中断当前任务") + raise UserInterruptException(f"用户中断当前任务") subs.append(srt.Subtitle(index=i, start=timedelta(seconds=segment.start), end=timedelta(seconds=segment.end), @@ -799,7 +807,7 @@ class AutoSubv2(_PluginBase): for item in valid_subs: if self._event.is_set(): logger.info(f"字幕{source_subtitle}翻译停止") - raise Exception(f"用户中断当前任务") + raise UserInterruptException(f"用户中断当前任务") current_batch.append(item) if len(current_batch) >= self.batch_size: