Rewrite coroutine to async def

This allows using async-timeout as a context manager

Author: pquentin

elasticsearch_async/connection.py

@@ -41,8 +41,7 @@ def __init__(self, host='localhost', port=9200, http_auth=None,
     def close(self):
         return self.session.close()
 
-    @asyncio.coroutine
-    def perform_request(self, method, url, params=None, body=None, timeout=None, ignore=()):
+    async def perform_request(self, method, url, params=None, body=None, timeout=None, ignore=()):
         url_path = url
         if params:
             url_path = '%s?%s' % (url, urlencode(params or {}))

elasticsearch_async/transport.py

@@ -58,8 +58,7 @@ def mark_dead(self, connection):
         if self.sniff_on_connection_fail:
             self.initiate_sniff()
 
-    @asyncio.coroutine
-    def _get_sniff_data(self, initial=False):
+    async def _get_sniff_data(self, initial=False):
         previous_sniff = self.last_sniff
 
         # reset last_sniff timestamp
@@ -102,8 +101,7 @@ def _get_sniff_data(self, initial=False):
             for t in chain(done, tasks):
                 t.cancel()
 
-    @asyncio.coroutine
-    def sniff_hosts(self, initial=False):
+    async def sniff_hosts(self, initial=False):
         """
         Obtain a list of nodes from the cluster and create a new connection
         pool using the information retrieved.
@@ -130,8 +128,7 @@ def sniff_hosts(self, initial=False):
             if c not in self.connection_pool.connections:
                 yield from c.close()
 
-    @asyncio.coroutine
-    def main_loop(self, method, url, params, body, ignore=(), timeout=None):
+    async def main_loop(self, method, url, params, body, ignore=(), timeout=None):
         for attempt in range(self.max_retries + 1):
             connection = self.get_connection()