feat(translations): support for missing params (#161)

crowdin · May 27, 2024 · a850ae3 · a850ae3
1 parent c4e4d9e
commit a850ae3
Show file tree

Hide file tree

Showing 4 changed files with 27 additions and 0 deletions.
diff --git a/crowdin_api/api_resources/translations/enums.py b/crowdin_api/api_resources/translations/enums.py
@@ -4,6 +4,7 @@
 class PreTranslationApplyMethod(Enum):
     TM = "tm"
     MT = "mt"
+    AI = "ai"
 
 
 class PreTranslationAutoApproveOption(Enum):

diff --git a/crowdin_api/api_resources/translations/resource.py b/crowdin_api/api_resources/translations/resource.py
@@ -2,6 +2,7 @@
 
 from crowdin_api.api_resources.abstract.resources import BaseResource
 from crowdin_api.api_resources.enums import ExportProjectTranslationFormat
+from crowdin_api.api_resources.translations.types import FallbackLanguages
 from crowdin_api.api_resources.translations.enums import (
     CharTransformation,
     PreTranslationApplyMethod,
@@ -54,10 +55,13 @@ def apply_pre_translation(
         projectId: Optional[int] = None,
         method: Optional[PreTranslationApplyMethod] = None,
         engineId: Optional[int] = None,
+        aiPromptId: Optional[int] = None,
         autoApproveOption: Optional[PreTranslationAutoApproveOption] = None,
         duplicateTranslations: Optional[bool] = None,
+        skipApprovedTranslations: Optional[bool] = None,
         translateUntranslatedOnly: Optional[bool] = None,
         translateWithPerfectMatchOnly: Optional[bool] = None,
+        fallbackLanguages: Optional[Iterable[FallbackLanguages]] = None,
         labelIds: Optional[Iterable[int]] = None,
         excludeLabelIds: Optional[Iterable[int]] = None,
     ):
@@ -67,6 +71,9 @@ def apply_pre_translation(
         Link to documentation:
         https://developer.crowdin.com/api/v2/#operation/api.projects.pre-translations.post
         """
+        if fallbackLanguages is None:
+            fallbackLanguages = []
+
         if labelIds is None:
             labelIds = []
 
@@ -83,10 +90,13 @@ def apply_pre_translation(
                 "fileIds": fileIds,
                 "method": method,
                 "engineId": engineId,
+                "aiPromptId": aiPromptId,
                 "autoApproveOption": autoApproveOption,
                 "duplicateTranslations": duplicateTranslations,
+                "skipApprovedTranslations": skipApprovedTranslations,
                 "translateUntranslatedOnly": translateUntranslatedOnly,
                 "translateWithPerfectMatchOnly": translateWithPerfectMatchOnly,
+                "fallbackLanguages": fallbackLanguages,
                 "labelIds": labelIds,
                 "excludeLabelIds": excludeLabelIds,
             },

diff --git a/crowdin_api/api_resources/translations/tests/test_translations_resources.py b/crowdin_api/api_resources/translations/tests/test_translations_resources.py
@@ -48,10 +48,13 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "fileIds": [1, 2],
                     "method": None,
                     "engineId": None,
+                    "aiPromptId": None,
                     "autoApproveOption": None,
                     "duplicateTranslations": None,
+                    "skipApprovedTranslations": None,
                     "translateUntranslatedOnly": None,
                     "translateWithPerfectMatchOnly": None,
+                    "fallbackLanguages": [],
                     "labelIds": [],
                     "excludeLabelIds": [],
                 },
@@ -62,10 +65,13 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "fileIds": [1, 2],
                     "method": PreTranslationApplyMethod.MT,
                     "engineId": 3,
+                    "aiPromptId": 0,
                     "autoApproveOption": PreTranslationAutoApproveOption.ALL,
                     "duplicateTranslations": False,
+                    "skipApprovedTranslations": False,
                     "translateUntranslatedOnly": False,
                     "translateWithPerfectMatchOnly": False,
+                    "fallbackLanguages": ["lang"],
                     "labelIds": [1],
                     "excludeLabelIds": [1],
                 },
@@ -74,10 +80,13 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "fileIds": [1, 2],
                     "method": PreTranslationApplyMethod.MT,
                     "engineId": 3,
+                    "aiPromptId": 0,
                     "autoApproveOption": PreTranslationAutoApproveOption.ALL,
                     "duplicateTranslations": False,
+                    "skipApprovedTranslations": False,
                     "translateUntranslatedOnly": False,
                     "translateWithPerfectMatchOnly": False,
+                    "fallbackLanguages": ["lang"],
                     "labelIds": [1],
                     "excludeLabelIds": [1],
                 },

diff --git a/crowdin_api/api_resources/translations/types.py b/crowdin_api/api_resources/translations/types.py
@@ -0,0 +1,7 @@
+from typing import Iterable
+
+from crowdin_api.typing import TypedDict
+
+
+class FallbackLanguages(TypedDict):
+    languageId: Iterable[str]