Fall back to asyncio.async when asyncio.ensure_future is not available. (2)

jonathanslenders · jonathanslenders · commit 69f05f7e4732 · 2016-03-17T21:31:41.000+01:00
diff --git a/asyncio_redis/replies.py b/asyncio_redis/replies.py
@@ -14,6 +14,12 @@
 )
 
 
+try:
+    ensure_future = asyncio.ensure_future
+except AttributeError:
+    ensure_future = asyncio.async
+
+
 class StatusReply:
     """
     Wrapper for Redis status replies.
@@ -62,7 +68,7 @@ def getter(f):
 
         for _ in range(self._result.count // 2):
             read_future = self._result._read(count=2)
-            yield asyncio.ensure_future(getter(read_future), loop=self._result._loop)
+            yield ensure_future(getter(read_future), loop=self._result._loop)
 
     @asyncio.coroutine
     def asdict(self):
