From 8eb77e192a5fe5ea7f94f6f4a1bdbe650b2cf990 Mon Sep 17 00:00:00 2001
From: xunbu <xunbu3@qq.com>
Date: Tue, 10 Jun 2025 19:38:15 +0800
Subject: [PATCH] =?UTF-8?q?=E5=8E=BB=E9=99=A4Filetranslater.file=5Fpath?=
 =?UTF-8?q?=EF=BC=8C=E6=94=B9=E7=94=A8Filetranslater.document?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 docutranslate/translater.py | 39 +++++++++++++------------------------
 1 file changed, 14 insertions(+), 25 deletions(-)

diff --git a/docutranslate/translater.py b/docutranslate/translater.py
index 2d2146a..5d6b14b 100644
--- a/docutranslate/translater.py
+++ b/docutranslate/translater.py
@@ -28,9 +28,6 @@ class FileTranslater:
                  mineru_token: str = None, cache=True):
         self.convert_engin = convert_engin
         self.mineru_token = mineru_token.strip() if mineru_token is not None else None
-        if isinstance(file_path, str):
-            file_path = Path(file_path)
-        self.file_path: Path = file_path
         self._mask_dict = MaskDict()
         self.markdown: str = ""
         self.chunksize = chunksize
@@ -50,6 +47,8 @@ class FileTranslater:
         self.document: Document | None = None
         self.cache = cache
         self.cacher = document_cacher_global
+        if file_path:
+            self.read_file(file_path=file_path)
 
     def _markdown_format(self):
         # 该方法还需要改进
@@ -182,12 +181,12 @@ class FileTranslater:
     def read_file(self, file_path: Path | str | None = None, formula=True, code=True, save=False,
                   save_format: Literal["markdown", "html"] = "markdown", refine=False,
                   refine_agent: Agent | None = None):
-        if file_path is None:
-            if self.file_path is None:
-                translater_logger.debug("未设置文件路径")
-                raise Exception("未设置文件路径")
-            file_path = self.file_path
-        document = Document(path=file_path)
+        if file_path:
+            document = Document(path=file_path)
+        else:
+            document=self.document
+        if document is None:
+            raise Exception("未读取文件")
         translater_logger.info(f"读取文件：{document.filename}")
         self.read_document(document, formula=formula, code=code, save=save, save_format=save_format, refine=refine,
                            refine_agent=refine_agent)
@@ -196,12 +195,12 @@ class FileTranslater:
     async def read_file_async(self, file_path: Path | str | None = None, formula=True, code=True, save=False,
                               save_format: Literal["markdown", "html"] = "markdown", refine=False,
                               refine_agent: Agent | None = None):
-        if file_path is None:
-            if self.file_path is None:
-                translater_logger.debug("未设置文件路径")
-                raise Exception("未设置文件路径")
-            file_path = self.file_path
-        document = Document(file_path)
+        if file_path:
+            document = Document(path=file_path)
+        else:
+            document=self.document
+        if document is None:
+            raise Exception("未读取文件")
         translater_logger.info(f"读取文件：{document.filename}")
         # 如果是markdown，直接读取
         await self.read_document_async(document, formula=formula, code=code, save=save, save_format=save_format,
@@ -370,11 +369,6 @@ class FileTranslater:
                        custom_prompt_translate=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
-        if isinstance(file_path, str):
-            file_path = Path(file_path)
         self.read_file(file_path, formula=formula, code=code)
         if refine:
             self.refine_markdown_by_agent(refine_agent)
@@ -393,11 +387,6 @@ class FileTranslater:
                                    code=True, output_format: Literal["markdown", "html"] = "markdown",
                                    custom_prompt_translate=None, refine=False,
                                    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
-        if isinstance(file_path, str):
-            file_path = Path(file_path)
         await asyncio.to_thread(
             self.read_file,
             file_path,