Merge pull request #98 from xxnuo/main

feat(cache folder): Change cache path to ~/.cache/babeldoc

babeldoc/document_il/midend/il_translator.py

@@ -2,6 +2,7 @@
 import json
 import logging
 from pathlib import Path
+from typing import List, Union
 
 from tqdm import tqdm
 
@@ -178,7 +179,7 @@ class TranslateInput:
         def __init__(
             self,
             unicode: str,
-            placeholders: [RichTextPlaceholder | FormulaPlaceholder],
+            placeholders: List[Union[RichTextPlaceholder, FormulaPlaceholder]],
             base_style: PdfStyle = None,
         ):
             self.unicode = unicode
@@ -487,7 +488,9 @@ def remove_placeholder(text: str):
                     PdfSameStyleCharacters,
                 ) and text.replace(" ", "") == "".join(
                     x.char_unicode for x in placeholder.composition.pdf_character
-                ).replace(" ", ""):
+                ).replace(
+                    " ", ""
+                ):
                     comp = PdfParagraphComposition(
                         pdf_same_style_characters=placeholder.composition,
                     )

babeldoc/document_il/translator/cache.py

@@ -8,6 +8,8 @@
 from peewee import SqliteDatabase
 from peewee import TextField
 
+from babeldoc.const import CACHE_FOLDER as cache_folder
+
 # we don't init the database here
 db = SqliteDatabase(None)
 
@@ -92,7 +94,6 @@ def set(self, original_text: str, translation: str):
 
 
 def init_db(remove_exists=False):
-    cache_folder = Path.home() / ".cache" / "pdf2zh"
     cache_folder.mkdir(parents=True, exist_ok=True)
     # The current version does not support database migration, so add the version number to the file name.
     cache_db_path = cache_folder / "cache.v1.db"