pdf2zh-next 2.0.0__py3-none-any.whl

pdf2zh_next/__init__.py

@@ -0,0 +1,69 @@
+from pdf2zh_next.config import AnythingLLMSettings
+from pdf2zh_next.config import AzureOpenAISettings
+from pdf2zh_next.config import AzureSettings
+from pdf2zh_next.config import BingSettings
+from pdf2zh_next.config import DeepLSettings
+from pdf2zh_next.config import DeepSeekSettings
+from pdf2zh_next.config import DifySettings
+from pdf2zh_next.config import GeminiSettings
+from pdf2zh_next.config import GoogleSettings
+from pdf2zh_next.config import GrokSettings
+from pdf2zh_next.config import GroqSettings
+from pdf2zh_next.config import ModelScopeSettings
+from pdf2zh_next.config import OllamaSettings
+from pdf2zh_next.config import OpenAISettings
+from pdf2zh_next.config import QwenMtSettings
+from pdf2zh_next.config import SiliconFlowSettings
+from pdf2zh_next.config import TencentSettings
+from pdf2zh_next.config import XinferenceSettings
+from pdf2zh_next.config import ZhipuSettings
+from pdf2zh_next.config.main import ConfigManager
+from pdf2zh_next.config.model import BasicSettings
+from pdf2zh_next.config.model import PDFSettings
+from pdf2zh_next.config.model import SettingsModel
+from pdf2zh_next.config.model import TranslationSettings
+from pdf2zh_next.config.model import WatermarkOutputMode
+from pdf2zh_next.high_level import create_babeldoc_config
+from pdf2zh_next.high_level import do_translate_async_stream
+from pdf2zh_next.high_level import do_translate_file
+from pdf2zh_next.high_level import do_translate_file_async
+
+# from pdf2zh_next.high_level import translate, translate_stream
+
+__version__ = "2.0.0"
+__author__ = "Byaidu, awwaawwa"
+__license__ = "AGPL-3.0"
+__maintainer__ = "awwaawwa"
+__email__ = "aw@funstory.ai"
+
+__all__ = [
+    "SettingsModel",
+    "BasicSettings",
+    "OpenAISettings",
+    "BingSettings",
+    "GoogleSettings",
+    "DeepLSettings",
+    "DeepSeekSettings",
+    "OllamaSettings",
+    "XinferenceSettings",
+    "AzureOpenAISettings",
+    "ModelScopeSettings",
+    "ZhipuSettings",
+    "SiliconFlowSettings",
+    "TencentSettings",
+    "GeminiSettings",
+    "AzureSettings",
+    "AnythingLLMSettings",
+    "DifySettings",
+    "GrokSettings",
+    "GroqSettings",
+    "QwenMtSettings",
+    "PDFSettings",
+    "TranslationSettings",
+    "WatermarkOutputMode",
+    "do_translate_file_async",
+    "do_translate_file",
+    "do_translate_async_stream",
+    "create_babeldoc_config",
+    "ConfigManager",
+]

pdf2zh_next/config/__init__.py

@@ -0,0 +1,55 @@
+from pdf2zh_next.config.main import ConfigManager
+from pdf2zh_next.config.model import BasicSettings
+from pdf2zh_next.config.model import PDFSettings
+from pdf2zh_next.config.model import SettingsModel
+from pdf2zh_next.config.model import TranslationSettings
+from pdf2zh_next.config.model import WatermarkOutputMode
+from pdf2zh_next.config.translate_engine_model import TRANSLATION_ENGINE_METADATA
+from pdf2zh_next.config.translate_engine_model import AnythingLLMSettings
+from pdf2zh_next.config.translate_engine_model import AzureOpenAISettings
+from pdf2zh_next.config.translate_engine_model import AzureSettings
+from pdf2zh_next.config.translate_engine_model import BingSettings
+from pdf2zh_next.config.translate_engine_model import DeepLSettings
+from pdf2zh_next.config.translate_engine_model import DeepSeekSettings
+from pdf2zh_next.config.translate_engine_model import DifySettings
+from pdf2zh_next.config.translate_engine_model import GeminiSettings
+from pdf2zh_next.config.translate_engine_model import GoogleSettings
+from pdf2zh_next.config.translate_engine_model import GrokSettings
+from pdf2zh_next.config.translate_engine_model import GroqSettings
+from pdf2zh_next.config.translate_engine_model import ModelScopeSettings
+from pdf2zh_next.config.translate_engine_model import OllamaSettings
+from pdf2zh_next.config.translate_engine_model import OpenAISettings
+from pdf2zh_next.config.translate_engine_model import QwenMtSettings
+from pdf2zh_next.config.translate_engine_model import SiliconFlowSettings
+from pdf2zh_next.config.translate_engine_model import TencentSettings
+from pdf2zh_next.config.translate_engine_model import XinferenceSettings
+from pdf2zh_next.config.translate_engine_model import ZhipuSettings
+
+__all__ = [
+    "ConfigManager",
+    "SettingsModel",
+    "BasicSettings",
+    "TranslationSettings",
+    "PDFSettings",
+    "WatermarkOutputMode",
+    "BingSettings",
+    "GoogleSettings",
+    "OpenAISettings",
+    "DeepLSettings",
+    "OllamaSettings",
+    "XinferenceSettings",
+    "AzureOpenAISettings",
+    "ModelScopeSettings",
+    "ZhipuSettings",
+    "SiliconFlowSettings",
+    "TencentSettings",
+    "GeminiSettings",
+    "AzureSettings",
+    "AnythingLLMSettings",
+    "DifySettings",
+    "GrokSettings",
+    "GroqSettings",
+    "QwenMtSettings",
+    "DeepSeekSettings",
+    "TRANSLATION_ENGINE_METADATA",
+]

pdf2zh_next/config/cli_env_model.py

@@ -0,0 +1,96 @@
+from __future__ import annotations
+
+import logging
+
+from pydantic import Field
+from pydantic import create_model
+
+from pdf2zh_next.config.model import SettingsModel
+from pdf2zh_next.config.translate_engine_model import _DEFAULT_TRANSLATION_ENGINE
+from pdf2zh_next.config.translate_engine_model import TRANSLATION_ENGINE_METADATA
+
+logger = logging.getLogger(__name__)
+
+# The following is magic code,
+# if you need to modify it,
+# please contact the maintainer!
+
+__translation_flag_fields = {
+    x.cli_flag_name: (
+        bool,
+        Field(
+            default=False, description=f"Use {x.translate_engine_type} for translation"
+        ),
+    )
+    for x in TRANSLATION_ENGINE_METADATA
+}
+
+__translation_flag_fields.update(
+    {
+        x.cli_detail_field_name: (
+            x.setting_model_type,
+            Field(default_factory=x.setting_model_type),
+        )
+        for x in TRANSLATION_ENGINE_METADATA
+        if x.cli_detail_field_name
+    }
+)
+
+__exclude_fields = list(__translation_flag_fields.keys())
+
+# If you want to use more field parameters in `pdf2zh_next/config/model.py`
+# please add the corresponding forwarding here!
+
+__cli_env_settings_model_fields = {
+    k: (
+        v.annotation,
+        Field(
+            default=v.default,
+            description=v.description,
+            default_factory=v.default_factory,
+            alias=v.alias,
+            discriminator=v.discriminator,
+        ),
+    )
+    for k, v in SettingsModel.model_fields.items()
+    if k != "translate_engine_settings"
+}
+__cli_env_settings_model_fields.update(__translation_flag_fields)
+
+CLIEnvSettingsModel = create_model(
+    "CLIEnvSettingsModel",
+    **__cli_env_settings_model_fields,
+)
+
+
+def to_settings_model(self) -> SettingsModel:
+    for metadata in TRANSLATION_ENGINE_METADATA:
+        if getattr(self, metadata.cli_flag_name):
+            if metadata.cli_detail_field_name:
+                translate_engine_settings = metadata.setting_model_type(
+                    **getattr(self, metadata.cli_detail_field_name).model_dump()
+                )
+            else:
+                translate_engine_settings = metadata.setting_model_type()
+            break
+    else:
+        logger.warning("No translation engine selected, using Bing")
+        translate_engine_settings = _DEFAULT_TRANSLATION_ENGINE()
+
+    return SettingsModel(
+        **self.model_dump(exclude=__exclude_fields),
+        translate_engine_settings=translate_engine_settings,
+    )
+
+
+def validate_settings(self) -> None:
+    self.to_settings_model().validate_settings()
+
+
+def clone(self):
+    return self.model_copy(deep=True)
+
+
+CLIEnvSettingsModel.to_settings_model = to_settings_model
+CLIEnvSettingsModel.validate_settings = validate_settings
+CLIEnvSettingsModel.clone = clone