epub-translator 0.0.6__py3-none-any.whl → 0.0.7__py3-none-any.whl

epub_translator/__init__.py

@@ -1,3 +1,3 @@
 from .llm import LLM
-from .translator import translate
+from .translator import translate, TranslatedWriteMode
 from .translation import Language, ProgressReporter

epub_translator/epub/html/dom_operator.py

@@ -13,7 +13,7 @@ def read_texts(root: Element) -> Generator[str, None, None]:
     elif position == TextPosition.TAIL:
       yield cast(str, element.tail)
 
-def append_texts(root: Element, texts: Iterable[str | Iterable[str] | None]):
+def write_texts(root: Element, texts: Iterable[str | Iterable[str] | None], append: bool):
   zip_list = list(zip(texts, search_texts(root)))
   for text, (element, position, parent) in reversed(zip_list):
     if text is None:
@@ -23,26 +23,32 @@ def append_texts(root: Element, texts: Iterable[str | Iterable[str] | None]):
       text = "".join(text)
     if position == TextPosition.WHOLE_DOM:
       if parent is not None:
-        _append_dom(parent, element, text)
+        _write_dom(parent, element, text, append)
     elif position == TextPosition.TEXT:
-      element.text = _append_text(element.text, text)
+      element.text = _write_text(element.text, text, append)
     elif position == TextPosition.TAIL:
-      element.tail = _append_text(element.tail, text)
+      element.tail = _write_text(element.tail, text, append)
 
-def _append_dom(parent: Element, origin: Element, text: str):
-  appended = Element(origin.tag, {**origin.attrib})
-  for index, child in enumerate(parent):
-    if child == origin:
-      parent.insert(index + 1, appended)
-      break
-
-  appended.attrib.pop("id", None)
-  appended.text = text
-  appended.tail = origin.tail
-  origin.tail = None
+def _write_dom(parent: Element, origin: Element, text: str, append: bool):
+  if append:
+    appended = Element(origin.tag, {**origin.attrib})
+    for index, child in enumerate(parent):
+      if child == origin:
+        parent.insert(index + 1, appended)
+        break
+    appended.attrib.pop("id", None)
+    appended.text = text
+    appended.tail = origin.tail
+    origin.tail = None
+  else:
+    for child in origin:
+      origin.remove(child)
+    origin.text = text
 
-def _append_text(left: str | None, right: str) -> str:
-  if left is None:
+def _write_text(left: str | None, right: str, append: bool) -> str:
+  if not append:
+    return right
+  elif left is None:
     return right
   else:
     return left + right

epub_translator/epub/html/file.py

@@ -2,7 +2,7 @@ import re
 
 from typing import Iterable
 from xml.etree.ElementTree import fromstring, tostring, Element
-from .dom_operator import read_texts, append_texts
+from .dom_operator import read_texts, write_texts
 from .empty_tags import to_xml, to_html
 
 
@@ -41,8 +41,8 @@ class HTMLFile:
     self._texts_length = len(texts)
     return texts
 
-  def write_texts(self, texts: Iterable[str]):
-    append_texts(self._root, texts)
+  def write_texts(self, texts: Iterable[str], append: bool):
+    write_texts(self._root, texts, append)
 
   @property
   def texts_length(self) -> int:

epub_translator/translator.py

@@ -1,5 +1,6 @@
 from os import PathLike
 from pathlib import Path
+from enum import auto, Enum
 from tempfile import mkdtemp
 from shutil import rmtree
 
@@ -9,11 +10,16 @@ from .zip_context import ZipContext
 from .translation import translate as _translate, Incision, Fragment, Language, ProgressReporter
 
 
+class TranslatedWriteMode(Enum):
+  APPEND = auto()
+  REPLACE = auto()
+
 def translate(
       llm: LLM,
       source_path: PathLike,
       translated_path: PathLike,
       target_language: Language,
+      write_mode: TranslatedWriteMode = TranslatedWriteMode.APPEND,
       user_prompt: str | None = None,
       working_path: PathLike | None = None,
       max_chunk_tokens_count: int = 3000,
@@ -29,6 +35,7 @@ def translate(
   _Translator(
     llm=llm,
     target_language=target_language,
+    write_mode=write_mode,
     user_prompt=user_prompt,
     max_chunk_tokens_count=max_chunk_tokens_count,
     max_threads_count=max_threads_count,
@@ -44,6 +51,7 @@ class _Translator:
         self,
         llm: LLM,
         target_language: Language,
+        write_mode: TranslatedWriteMode,
         user_prompt: str | None,
         max_chunk_tokens_count: int,
         max_threads_count: int,
@@ -52,6 +60,7 @@ class _Translator:
 
     self._llm: LLM = llm
     self._target_language: Language = target_language
+    self._write_mode: TranslatedWriteMode = write_mode
     self._user_prompt: str | None = user_prompt
     self._max_chunk_tokens_count: int = max_chunk_tokens_count
     self._max_threads_count: int = max_threads_count
@@ -110,6 +119,7 @@ class _Translator:
     spine: tuple[Path, HTMLFile] | None = None
     translated_texts: list[str] = []
     translated_count: int = 0
+    append = (self._write_mode == TranslatedWriteMode.APPEND)
 
     for translated_text in _translate(
       llm=self._llm,
@@ -125,7 +135,7 @@ class _Translator:
 
       if spine and translated_count >= len(translated_texts):
         spine_path, spine_file = spine
-        spine_file.write_texts(translated_texts)
+        spine_file.write_texts(translated_texts, append)
         context.write_spine_file(spine_path, spine_file)
         spine = None
 
@@ -152,7 +162,7 @@ class _Translator:
     if spine:
       spine_path, spine_file = spine
       if translated_count > 0:
-        spine_file.write_texts(translated_texts)
+        spine_file.write_texts(translated_texts, append)
       context.write_spine_file(spine_path, spine_file)
 
 def _gen_fragments(context: ZipContext):

{epub_translator-0.0.6.dist-info → epub_translator-0.0.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: epub-translator
-Version: 0.0.6
+Version: 0.0.7
 Summary: Translate the epub book using LLM. The translated book will retain the original text and list the translated text side by side with the original text.
 License: MIT
 Keywords: epub,llm,translation,translator

{epub_translator-0.0.6.dist-info → epub_translator-0.0.7.dist-info}/RECORD

@@ -1,12 +1,12 @@
-epub_translator/__init__.py,sha256=69k-70xbqxLgTvLyJE_MPFyymmPgyUAFg4mCSfEn49k,106
+epub_translator/__init__.py,sha256=b3_R1I7W6bPYzF1T6O3X_osekX940V41sj14qJ62k6E,127
 epub_translator/data/format.jinja,sha256=1dXiMAhznuHWA2vpDx-968pm95FCXCRV3cKVmy3EXYA,3710
 epub_translator/data/translate.jinja,sha256=XWiMgeQS3tKCnh9cC6p1KQs9kqb1GO6m_j_cNeDq8lQ,1768
 epub_translator/epub/__init__.py,sha256=vqD29XWxgcfHw8LFCHQxupQeJufEEoS5_KHNphwfEYU,73
 epub_translator/epub/content_parser.py,sha256=Y-2fEcACCAl_ucA6VwwOoQ95BMHr9f-hryuxnsYNONM,4403
 epub_translator/epub/html/__init__.py,sha256=Yyg0TrWjecolaJP6qLzDnOlhi7SrkW0Is7wh9Fr0-Bg,26
-epub_translator/epub/html/dom_operator.py,sha256=Ryayv6hG0jEXv7RkXrZTbIP54P0fyPTbMVbymMtBUnU,1935
+epub_translator/epub/html/dom_operator.py,sha256=0Oa73gzT_Aby4NOZE5cBiDZIP0HHWmvoOZpI8fgCw80,2144
 epub_translator/epub/html/empty_tags.py,sha256=GSSe-CV4YkUhWv4F0fiiRsf2vz0ZBAsC21Ovnqo5oIA,601
-epub_translator/epub/html/file.py,sha256=AqUV-Tmptk5J2EYmw3oRVsLjGSqEMNz5rItnoRbGstc,2477
+epub_translator/epub/html/file.py,sha256=7hdNHX5RI-Rzh4aJkOSMlwQZHLoDB0EcDxmRCmX4bwg,2497
 epub_translator/epub/html/texts_searcher.py,sha256=vamO99pki6_sX2PeKCJk7mPwHdApZq1sOgSYDTPckx8,1376
 epub_translator/llm/__init__.py,sha256=wMBWLgh5iLNQBioniSOmWC83NS7RLM41hIs1V1uZiWI,21
 epub_translator/llm/error.py,sha256=fG0A3z69YoSNu0MNVWVFMtHCB_4fpOvAEb0Kajn9OHc,1401
@@ -21,7 +21,7 @@ epub_translator/translation/store.py,sha256=T6OYhQZETPz5p-ykmPC_wDLzmRet8o7jDY3s
 epub_translator/translation/translation.py,sha256=BBmaNME7Pd0zytK_7CCx10dsoWLootB3uggvYLW-rQM,7079
 epub_translator/translation/types.py,sha256=Pnzll2V4MnOsW_kIP0XU6Cj-QQkLTByB6dr_U4sdNH8,989
 epub_translator/translation/utils.py,sha256=G6Gqq6mot3lgFA-jqUD0UqtDS0GC1wrb9DnK7rTxJNs,223
-epub_translator/translator.py,sha256=t5vgiDPiYda1CiKtEB91hjqVrN2dDZ7sTBLTotIVv3w,5487
+epub_translator/translator.py,sha256=XbC90X4R4fX2nTaPmfMVkdF4ZcgDOWCmvYMPNdGhuqM,5855
 epub_translator/xml/__init__.py,sha256=o2_qwUYU_MUcyfmfKkiOQ-cKUQyl4PiRL8YHVzCTAZU,106
 epub_translator/xml/decoder.py,sha256=UlqgmEKQDzxt3lvBeNGHgZP6jznmnq_1HLJuAe5X0C4,2181
 epub_translator/xml/encoder.py,sha256=p4A7GRSOM2i0WOh1lLtEdTTg2gXSQrxDdzMgUqbiV18,2428
@@ -30,7 +30,7 @@ epub_translator/xml/tag.py,sha256=QLZImF0PtYyiASI7swrB8DL_qUwcYpU6cL68jEXDnvg,23
 epub_translator/xml/transform.py,sha256=vS_a4d_o2Qqf9B6k2CovQVLUknp6TyUi3FyLOu21Vio,1126
 epub_translator/xml/utils.py,sha256=KDNGWHwaIiFKS27sjZF0e-bBSjeTxzceae_aeuj4wzI,384
 epub_translator/zip_context.py,sha256=SwkqghokGLGx-tlZ_K7sJTF-g-cm6dErurSZCM-ZLeM,2414
-epub_translator-0.0.6.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
-epub_translator-0.0.6.dist-info/METADATA,sha256=D_355033oXlgbht7x16tq6eHP50NZnHtStSyTDfF0-o,8081
-epub_translator-0.0.6.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
-epub_translator-0.0.6.dist-info/RECORD,,
+epub_translator-0.0.7.dist-info/LICENSE,sha256=5RF32sL3LtMOJIErdDKp1ZEYPGXS8WPpsiSz_jMBnGI,1066
+epub_translator-0.0.7.dist-info/METADATA,sha256=tRB8jrDtc2V4X2qqCMEbOLDaImhjzB1WqD5YGPKNQIg,8081
+epub_translator-0.0.7.dist-info/WHEEL,sha256=b4K_helf-jlQoXBBETfwnf4B04YC67LOev0jo4fX5m8,88
+epub_translator-0.0.7.dist-info/RECORD,,