fix(asyncio): use asyncio.async when possible

h2non · Jan 28, 2017 · 254a255 · 254a255
1 parent 170c128
commit 254a255
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/paco/interval.py b/paco/interval.py
@@ -3,6 +3,11 @@
 from .decorator import decorate
 from .assertions import assert_corofunction
 
+try:
+    ensure_future = asyncio.ensure_future
+except:
+    ensure_future = getattr(asyncio, 'async')
+
 
 @decorate
 def interval(coro, interval=1, times=None):
@@ -62,6 +67,6 @@ def schedule(times, *args, **kw):
             yield from asyncio.sleep(interval)
 
     def wrapper(*args, **kw):
-        return asyncio.ensure_future(schedule(times, *args, **kw))
+        return ensure_future(schedule(times, *args, **kw))
 
     return wrapper