Add Async I/O Support

elasticsearch_dsl/__init__.py

@@ -15,6 +15,8 @@
 #  specific language governing permissions and limitations
 #  under the License.
 
+import sys
+
 from . import connections
 from .aggs import A
 from .analysis import analyzer, char_filter, normalizer, token_filter, tokenizer
@@ -159,3 +161,31 @@
     "token_filter",
     "tokenizer",
 ]
+
+
+try:
+    # Asyncio only supported in Python 3.6+
+    if sys.version_info < (3, 6):
+        raise ImportError
+
+    from elasticsearch_dsl._async.faceted_search import AsyncFacetedSearch
+    from elasticsearch_dsl._async.index import AsyncIndex, AsyncIndexTemplate
+    from elasticsearch_dsl._async.mapping import AsyncMapping
+    from elasticsearch_dsl._async.search import AsyncMultiSearch, AsyncSearch
+    from elasticsearch_dsl._async.update_by_query import AsyncUpdateByQuery
+
+    __all__ = sorted(
+        __all__
+        + [
+            "AsyncFacetedSearch",
+            "AsyncIndex",
+            "AsyncIndexTemplate",
+            "AsyncMapping",
+            "AsyncMultiSearch",
+            "AsyncSearch",
+            "AsyncUpdateByQuery",
+        ]
+    )
+
+except (ImportError, SyntaxError):
+    pass

elasticsearch_dsl/_async/faceted_search.py

@@ -0,0 +1,11 @@
+from elasticsearch_dsl.faceted_search import FacetedSearch
+
+
+class AsyncFacetedSearch(FacetedSearch):
+    async def execute(self):
+        """
+        Asynchronously execute the search and return the response.
+        """
+        r = await self._s.execute()
+        r._faceted_search = self
+        return r