feat(python): add bridge to transformation on search

Author: shortcuts

templates/python/api.mustache

@@ -9,6 +9,10 @@ from algoliasearch.search.models.operation_type import OperationType
 from algoliasearch.search.models.replace_all_objects_response import ReplaceAllObjectsResponse
 from algoliasearch.search.models.scope_type import ScopeType
 from algoliasearch.search.models.secured_api_key_restrictions import SecuredApiKeyRestrictions
+
+from algoliasearch.ingestion.models.watch_response import WatchResponse
+from algoliasearch.ingestion.config import IngestionConfig
+from algoliasearch.ingestion.client import (IngestionClient, IngestionClientSync)
 {{/isSearchClient}}
 
 {{#operations}}{{#operation}}{{#imports}}
@@ -37,6 +41,9 @@ class {{classname}}{{#isSyncClient}}Sync{{/isSyncClient}}:
     """
 
     _transporter: Transporter{{#isSyncClient}}Sync{{/isSyncClient}}
+    {{#isSearchClient}}
+    _ingestion_transporter: Optional[IngestionClient{{#isSyncClient}}Sync{{/isSyncClient}}]
+    {{/isSearchClient}}
     _config: BaseConfig
     _request_options: RequestOptions
 
@@ -49,6 +56,9 @@ class {{classname}}{{#isSyncClient}}Sync{{/isSyncClient}}:
         self._config = config
         self._request_options = RequestOptions(config)
 
+        if config.region is not None:
+            self._ingestion_transporter = IngestionClient{{#isSyncClient}}Sync{{/isSyncClient}}.create_with_config(IngestionConfig(config.app_id, config.api_key, config.region))
+
         if transporter is None:
             transporter = Transporter{{#isSyncClient}}Sync{{/isSyncClient}}(config)
         self._transporter = transporter

templates/python/config.mustache

@@ -77,3 +77,9 @@ class {{#lambda.pascalcase}}{{client}}{{/lambda.pascalcase}}Config(BaseConfig):
             {{/hostWithAppID}}
         )
         {{/hasRegionalHost}}
+
+    {{#isSearchClient}}
+    def with_transformation(self, region: str = ""):
+        "This method is required to be called with the appropriate region of your Algolia application if you wish to leverage the *_with_transformation methods."
+        self.region = region
+    {{/isSearchClient}}

templates/python/search_helpers.mustache

@@ -292,6 +292,31 @@
         """
         return {{^isSyncClient}}await {{/isSyncClient}}self.chunked_batch(index_name=index_name, objects=objects, action=Action.ADDOBJECT, wait_for_tasks=wait_for_tasks, batch_size=batch_size, request_options=request_options)
 
+    {{^isSyncClient}}async {{/isSyncClient}}def save_objects_with_transformation(
+        self,
+        index_name: str,
+        objects: List[Dict[str, Any]],
+        wait_for_tasks: bool = False,
+        batch_size: int = 1000,
+        request_options: Optional[Union[dict, RequestOptions]] = None,
+    ) -> WatchResponse:
+        """
+        Helper: Similar to the `save_objects` method but requires a Push connector (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/) to be created first, in order to transform records before indexing them to Algolia. The `region` must've been passed to the client's config at instantiation.
+        """
+        if self._ingestion_transporter is None:
+            raise ValueError("`region` must be provided at client instantiation before calling this method.")
+
+        return {{^isSyncClient}}await {{/isSyncClient}}self._ingestion_transporter.push(
+            index_name=index_name,
+            push_task_payload={
+                "action": Action.ADDOBJECT,
+                "records": objects,
+            },
+            watch=wait_for_tasks,
+            request_options=request_options,
+        )
+
+
     {{^isSyncClient}}async {{/isSyncClient}}def delete_objects(
         self,
         index_name: str,
@@ -319,6 +344,32 @@
         """
         return {{^isSyncClient}}await {{/isSyncClient}}self.chunked_batch(index_name=index_name, objects=objects, action=Action.PARTIALUPDATEOBJECT if create_if_not_exists else Action.PARTIALUPDATEOBJECTNOCREATE, wait_for_tasks=wait_for_tasks, batch_size=batch_size, request_options=request_options)
 
+    {{^isSyncClient}}async {{/isSyncClient}}def partial_update_objects_with_transformation(
+        self,
+        index_name: str,
+        objects: List[Dict[str, Any]],
+        create_if_not_exists: bool = False,
+        wait_for_tasks: bool = False,
+        batch_size: int = 1000,
+        request_options: Optional[Union[dict, RequestOptions]] = None,
+    ) -> WatchResponse:
+        """
+        Helper: Similar to the `partial_update_objects` method but requires a Push connector (https://www.algolia.com/doc/guides/sending-and-managing-data/send-and-update-your-data/connectors/push/) to be created first, in order to transform records before indexing them to Algolia. The `region` must've been passed to the client instantiation method.
+        """
+        if self._ingestion_transporter is None:
+            raise ValueError("`region` must be provided at client instantiation before calling this method.")
+
+        return {{^isSyncClient}}await {{/isSyncClient}}self._ingestion_transporter.push(
+            index_name=index_name,
+            push_task_payload={
+                "action": Action.PARTIALUPDATEOBJECT if create_if_not_exists else Action.PARTIALUPDATEOBJECTNOCREATE,
+                "records": objects,
+            },
+            watch=wait_for_tasks,
+            request_options=request_options,
+        )
+
+
     {{^isSyncClient}}async {{/isSyncClient}}def chunked_batch(
         self,
         index_name: str,