Generate async API

Author: sethmlarson

elasticsearch/_async/client/__init__.py

@@ -7,7 +7,7 @@
 
 class AsyncSearchClient(NamespacedClient):
     @query_params()
-    def delete(self, id, params=None, headers=None):
+    async def delete(self, id, params=None, headers=None):
         """
         Deletes an async search by ID. If the search is still running, the search
         request will be cancelled. Otherwise, the saved search results are deleted.
@@ -18,12 +18,12 @@ def delete(self, id, params=None, headers=None):
         if id in SKIP_IN_PATH:
             raise ValueError("Empty value passed for a required argument 'id'.")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "DELETE", _make_path("_async_search", id), params=params, headers=headers
         )
 
     @query_params("keep_alive", "typed_keys", "wait_for_completion_timeout")
-    def get(self, id, params=None, headers=None):
+    async def get(self, id, params=None, headers=None):
         """
         Retrieves the results of a previously submitted async search request given its
         ID.
@@ -40,7 +40,7 @@ def get(self, id, params=None, headers=None):
         if id in SKIP_IN_PATH:
             raise ValueError("Empty value passed for a required argument 'id'.")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "GET", _make_path("_async_search", id), params=params, headers=headers
         )
 
@@ -87,7 +87,7 @@ def get(self, id, params=None, headers=None):
         "version",
         "wait_for_completion_timeout",
     )
-    def submit(self, body=None, index=None, params=None, headers=None):
+    async def submit(self, body=None, index=None, params=None, headers=None):
         """
         Executes a search request asynchronously.
         `<https://www.elastic.co/guide/en/elasticsearch/reference/master/async-search.html>`_
@@ -182,7 +182,7 @@ def submit(self, body=None, index=None, params=None, headers=None):
         if "from_" in params:
             params["from"] = params.pop("from_")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "POST",
             _make_path(index, "_async_search"),
             params=params,

elasticsearch/_async/client/async_search.py

@@ -7,18 +7,18 @@
 
 class AutoscalingClient(NamespacedClient):
     @query_params()
-    def get_autoscaling_decision(self, params=None, headers=None):
+    async def get_autoscaling_decision(self, params=None, headers=None):
         """
         Gets the current autoscaling decision based on the configured autoscaling
         policy, indicating whether or not autoscaling is needed.
         `<https://www.elastic.co/guide/en/elasticsearch/reference/master/autoscaling-get-autoscaling-decision.html>`_
         """
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "GET", "/_autoscaling/decision", params=params, headers=headers
         )
 
     @query_params()
-    def delete_autoscaling_policy(self, name, params=None, headers=None):
+    async def delete_autoscaling_policy(self, name, params=None, headers=None):
         """
         Deletes an autoscaling policy.
         `<https://www.elastic.co/guide/en/elasticsearch/reference/master/autoscaling-delete-autoscaling-policy.html>`_
@@ -28,15 +28,15 @@ def delete_autoscaling_policy(self, name, params=None, headers=None):
         if name in SKIP_IN_PATH:
             raise ValueError("Empty value passed for a required argument 'name'.")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "DELETE",
             _make_path("_autoscaling", "policy", name),
             params=params,
             headers=headers,
         )
 
     @query_params()
-    def put_autoscaling_policy(self, name, body, params=None, headers=None):
+    async def put_autoscaling_policy(self, name, body, params=None, headers=None):
         """
         Creates a new autoscaling policy.
         `<https://www.elastic.co/guide/en/elasticsearch/reference/master/autoscaling-put-autoscaling-policy.html>`_
@@ -48,7 +48,7 @@ def put_autoscaling_policy(self, name, body, params=None, headers=None):
             if param in SKIP_IN_PATH:
                 raise ValueError("Empty value passed for a required argument.")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "PUT",
             _make_path("_autoscaling", "policy", name),
             params=params,
@@ -57,7 +57,7 @@ def put_autoscaling_policy(self, name, body, params=None, headers=None):
         )
 
     @query_params()
-    def get_autoscaling_policy(self, name, params=None, headers=None):
+    async def get_autoscaling_policy(self, name, params=None, headers=None):
         """
         Retrieves an autoscaling policy.
         `<https://www.elastic.co/guide/en/elasticsearch/reference/master/autoscaling-get-autoscaling-policy.html>`_
@@ -67,7 +67,7 @@ def get_autoscaling_policy(self, name, params=None, headers=None):
         if name in SKIP_IN_PATH:
             raise ValueError("Empty value passed for a required argument 'name'.")
 
-        return self.transport.perform_request(
+        return await self.transport.perform_request(
             "GET",
             _make_path("_autoscaling", "policy", name),
             params=params,