From 847e0c6dc5473b6359d873f141ff92a9d6d5f317 Mon Sep 17 00:00:00 2001 From: xunbu Date: Sun, 11 May 2025 10:48:29 +0800 Subject: [PATCH] =?UTF-8?q?=E5=A2=9E=E5=8A=A0save=E9=80=89=E9=A1=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docutranslate/translater.py | 30 ++++++++++++++++-------------- 1 file changed, 16 insertions(+), 14 deletions(-) diff --git a/docutranslate/translater.py b/docutranslate/translater.py index 0b64983..68b293a 100644 --- a/docutranslate/translater.py +++ b/docutranslate/translater.py @@ -245,7 +245,7 @@ class FileTranslater: def translate_file(self, file_path: Path | str | None = None, to_lang="中文", output_dir="./output", formula=False, code=False, output_format: Literal["markdown", "html"] = "markdown", refine=False, - refine_agent: Agent | None = None, translate_agent: Agent | None = None): + refine_agent: Agent | None = None, translate_agent: Agent | None = None,save=True): if file_path is None: assert self.file_path is not None, "未输入文件路径" file_path = self.file_path @@ -255,26 +255,28 @@ class FileTranslater: if refine: self.refine_markdown_by_agent(refine_agent) self.translate_markdown_by_agent(translate_agent, to_lang=to_lang) - if output_format == "markdown": - filename = f"{file_path.stem}_{to_lang}.md" - self.save_as_markdown(filename=filename, output_dir=output_dir) - elif output_format == "html": - filename = f"{file_path.stem}_{to_lang}.html" - self.save_as_html(filename=filename, output_dir=output_dir) + if save: + if output_format == "markdown": + filename = f"{file_path.stem}_{to_lang}.md" + self.save_as_markdown(filename=filename, output_dir=output_dir) + elif output_format == "html": + filename = f"{file_path.stem}_{to_lang}.html" + self.save_as_html(filename=filename, output_dir=output_dir) return self def translate_bytes(self, name:str,file: bytes, to_lang="中文", output_dir="./output", formula=False, code=False, output_format: Literal["markdown", "html"] = "markdown", refine=False, - refine_agent: Agent | None = None, translate_agent: Agent | None = None): + refine_agent: Agent | None = None, translate_agent: Agent | None = None,save=True): self.read_bytes(name=name,file=file, formula=formula, code=code) if refine: self.refine_markdown_by_agent(refine_agent) self.translate_markdown_by_agent(translate_agent, to_lang=to_lang) - if output_format == "markdown": - filename = f"{name}_{to_lang}.md" - self.save_as_markdown(filename=filename, output_dir=output_dir) - elif output_format == "html": - filename = f"{name}_{to_lang}.html" - self.save_as_html(filename=filename, output_dir=output_dir) + if save: + if output_format == "markdown": + filename = f"{name}_{to_lang}.md" + self.save_as_markdown(filename=filename, output_dir=output_dir) + elif output_format == "html": + filename = f"{name}_{to_lang}.html" + self.save_as_html(filename=filename, output_dir=output_dir) return self