feat: add labels to apply pre-translation (#118)

crowdin · Oct 3, 2023 · dd0c0e7 · dd0c0e7
1 parent 55ada53
commit dd0c0e7
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 0 deletions.
diff --git a/crowdin_api/api_resources/translations/resource.py b/crowdin_api/api_resources/translations/resource.py
@@ -54,13 +54,20 @@ def apply_pre_translation(
         duplicateTranslations: Optional[bool] = None,
         translateUntranslatedOnly: Optional[bool] = None,
         translateWithPerfectMatchOnly: Optional[bool] = None,
+        labelIds: Optional[Iterable[int]] = None,
+        excludeLabelIds: Optional[Iterable[int]] = None,
     ):
         """
         Apply Pre-Translation.
 
         Link to documentation:
         https://developer.crowdin.com/api/v2/#operation/api.projects.pre-translations.post
         """
+        if labelIds is None:
+            labelIds = []
+
+        if excludeLabelIds is None:
+            excludeLabelIds = []
 
         return self.requester.request(
             method="post",
@@ -74,6 +81,8 @@ def apply_pre_translation(
                 "duplicateTranslations": duplicateTranslations,
                 "translateUntranslatedOnly": translateUntranslatedOnly,
                 "translateWithPerfectMatchOnly": translateWithPerfectMatchOnly,
+                "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
@@ -45,6 +45,8 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "duplicateTranslations": None,
                     "translateUntranslatedOnly": None,
                     "translateWithPerfectMatchOnly": None,
+                    "labelIds": [],
+                    "excludeLabelIds": [],
                 },
             ),
             (
@@ -57,6 +59,8 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "duplicateTranslations": False,
                     "translateUntranslatedOnly": False,
                     "translateWithPerfectMatchOnly": False,
+                    "labelIds": [1],
+                    "excludeLabelIds": [1],
                 },
                 {
                     "languageIds": ["some", "language"],
@@ -67,6 +71,8 @@ def test_list_project_branches(self, m_request, base_absolut_url):
                     "duplicateTranslations": False,
                     "translateUntranslatedOnly": False,
                     "translateWithPerfectMatchOnly": False,
+                    "labelIds": [1],
+                    "excludeLabelIds": [1],
                 },
             ),
         ),