Replace asyncio.Timeout with async_timeout.timeout (elastic#38)

Author: ciscorn

elasticsearch_async/connection.py

@@ -2,6 +2,7 @@
 
 import aiohttp
 from aiohttp.client_exceptions import ServerFingerprintMismatch
+import async_timeout
 
 from elasticsearch.exceptions import ConnectionError, ConnectionTimeout, SSLError
 from elasticsearch.connection import Connection
@@ -50,7 +51,7 @@ async def perform_request(self, method, url, params=None, body=None, timeout=Non
         start = self.loop.time()
         response = None
         try:
-            with aiohttp.Timeout(timeout or self.timeout):
+            async with async_timeout.timeout(timeout or self.timeout):
                 response = yield from self.session.request(method, url, data=body)
                 raw_data = yield from response.text()
             duration = self.loop.time() - start

setup.py

@@ -12,6 +12,7 @@
 
 install_requires = [
     'aiohttp',
+    'async_timeout',
     'elasticsearch>=5.0.0',
 ]