Leon/docutranslate
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

增加save选项

Browse Source
This commit is contained in:
xunbu
2025-05-11 10:48:29 +08:00
parent 69bc0aea55
commit 847e0c6dc5
1 changed files with 16 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
docutranslate/translater.py
View File

@@ -245,7 +245,7 @@ class FileTranslater:
def translate_file(self, file_path: Path | str | None = None, to_lang="中文", output_dir="./output", def translate_file(self, file_path: Path | str | None = None, to_lang="中文", output_dir="./output",
formula=False, formula=False,
code=False, output_format: Literal["markdown", "html"] = "markdown", refine=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: if file_path is None:
assert self.file_path is not None, "未输入文件路径" assert self.file_path is not None, "未输入文件路径"
file_path = self.file_path file_path = self.file_path
@@ -255,26 +255,28 @@ class FileTranslater:
if refine: if refine:
self.refine_markdown_by_agent(refine_agent) self.refine_markdown_by_agent(refine_agent)
self.translate_markdown_by_agent(translate_agent, to_lang=to_lang) self.translate_markdown_by_agent(translate_agent, to_lang=to_lang)
if output_format == "markdown": if save:
filename = f"{file_path.stem}_{to_lang}.md" if output_format == "markdown":
self.save_as_markdown(filename=filename, output_dir=output_dir) filename = f"{file_path.stem}_{to_lang}.md"
elif output_format == "html": self.save_as_markdown(filename=filename, output_dir=output_dir)
filename = f"{file_path.stem}_{to_lang}.html" elif output_format == "html":
self.save_as_html(filename=filename, output_dir=output_dir) filename = f"{file_path.stem}_{to_lang}.html"
self.save_as_html(filename=filename, output_dir=output_dir)
return self return self
def translate_bytes(self, name:str,file: bytes, to_lang="中文", output_dir="./output", def translate_bytes(self, name:str,file: bytes, to_lang="中文", output_dir="./output",
formula=False, formula=False,
code=False, output_format: Literal["markdown", "html"] = "markdown", refine=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) self.read_bytes(name=name,file=file, formula=formula, code=code)
if refine: if refine:
self.refine_markdown_by_agent(refine_agent) self.refine_markdown_by_agent(refine_agent)
self.translate_markdown_by_agent(translate_agent, to_lang=to_lang) self.translate_markdown_by_agent(translate_agent, to_lang=to_lang)
if output_format == "markdown": if save:
filename = f"{name}_{to_lang}.md" if output_format == "markdown":
self.save_as_markdown(filename=filename, output_dir=output_dir) filename = f"{name}_{to_lang}.md"
elif output_format == "html": self.save_as_markdown(filename=filename, output_dir=output_dir)
filename = f"{name}_{to_lang}.html" elif output_format == "html":
self.save_as_html(filename=filename, output_dir=output_dir) filename = f"{name}_{to_lang}.html"
self.save_as_html(filename=filename, output_dir=output_dir)
return self return self