BabelDOC 0.3.61__tar.gz → 0.3.62__tar.gz

{babeldoc-0.3.61 → babeldoc-0.3.62}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: BabelDOC
-Version: 0.3.61
+Version: 0.3.62
 Summary: Yet Another Document Translator
 Project-URL: Homepage, https://github.com/funstory-ai/BabelDOC
 Project-URL: Issues, https://github.com/funstory-ai/BabelDOC/issues

babeldoc-0.3.62/babeldoc/__init__.py

@@ -0,0 +1 @@
+__version__ = "0.3.62"

{babeldoc-0.3.61 → babeldoc-0.3.62}/babeldoc/const.py

@@ -3,7 +3,7 @@ import shutil
 import subprocess
 from pathlib import Path
 
-__version__ = "0.3.61"
+__version__ = "0.3.62"
 
 CACHE_FOLDER = Path.home() / ".cache" / "babeldoc"
 

{babeldoc-0.3.61 → babeldoc-0.3.62}/babeldoc/high_level.py

@@ -595,6 +595,12 @@ def do_translate(
         # not being able to copy translated text can be fixed,
         # the macOS preview is broken
         # fix_cmap(result, translation_config)
+        try:
+            migrate_toc(translation_config, result)
+        except Exception as e:
+            logger.error(
+                f"Failed to migrate TOC from {translation_config.input_file}: {e}"
+            )
         pm.translate_done(result)
         return result
 
@@ -612,6 +618,51 @@ def do_translate(
         translation_config.cleanup_temp_files()
 
 
+def migrate_toc(
+    translation_config: TranslationConfig, translate_result: TranslateResult
+):
+    old_doc = Document(translation_config.input_file)
+    if not old_doc:
+        return
+
+    try:
+        fix_filter(old_doc)
+        fix_null_xref(old_doc)
+    except Exception:
+        logger.exception("auto fix failed, please check the pdf file")
+
+    toc_data = old_doc.get_toc()
+
+    if not toc_data:
+        logger.info("No TOC found in the original PDF, skipping migration.")
+        return
+
+    files = {
+        translate_result.dual_pdf_path,
+        # translate_result.mono_pdf_path,
+        translate_result.no_watermark_dual_pdf_path,
+        # translate_result.no_watermark_mono_pdf_path
+    }
+
+    for f in files:
+        mig_toc_temp_input = translation_config.get_working_file_path(
+            "mig_toc_temp.pdf"
+        )
+        shutil.copy(f, mig_toc_temp_input)
+        new_doc = Document(mig_toc_temp_input.as_posix())
+        if not new_doc:
+            continue
+
+        new_doc.set_toc(toc_data)
+        PDFCreater.save_pdf_with_timeout(
+            new_doc,
+            f.as_posix(),
+            translation_config=translation_config,
+            clean=not translation_config.skip_clean,
+            tag="mig_toc",
+        )
+
+
 def fix_media_box(doc: Document) -> None:
     mediabox_data = {}
     for x in range(1, doc.xref_length()):

{babeldoc-0.3.61 → babeldoc-0.3.62}/babeldoc/main.py

@@ -21,7 +21,7 @@ from babeldoc.translation_config import TranslationConfig
 from babeldoc.translation_config import WatermarkOutputMode
 
 logger = logging.getLogger(__name__)
-__version__ = "0.3.61"
+__version__ = "0.3.62"
 
 
 def create_parser():

{babeldoc-0.3.61 → babeldoc-0.3.62}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "BabelDOC"
-version = "0.3.61"
+version = "0.3.62"
 description = "Yet Another Document Translator"
 license = "AGPL-3.0"
 readme = "README.md"
@@ -154,7 +154,7 @@ pythonpath = [".", "src"]
 testpaths = ["tests"]
 
 [bumpver]
-current_version = "0.3.61"
+current_version = "0.3.62"
 version_pattern = "MAJOR.MINOR.PATCH[.PYTAGNUM]"
 
 [bumpver.file_patterns]

{babeldoc-0.3.61 → babeldoc-0.3.62}/uv.lock

@@ -48,7 +48,7 @@ wheels = [
 
 [[package]]
 name = "babeldoc"
-version = "0.3.61"
+version = "0.3.62"
 source = { editable = "." }
 dependencies = [
     { name = "bitstring" },

babeldoc-0.3.61/babeldoc/__init__.py

@@ -1 +0,0 @@
-__version__ = "0.3.61"