iscoroutine() now also supports asyncio.CoroWrapper

1 files changed, 8 insertions, 1 deletions

diff --git a/trollius/coroutines.py b/trollius/coroutines.py
index ee610c2..fd16ceb 100644
--- a/trollius/coroutines.py
+++ b/trollius/coroutines.py
@@ -149,9 +149,16 @@ def iscoroutinefunction(func):
     return getattr(func, '_is_coroutine', False)
 
 
+if asyncio is not None:
+    # Accept also asyncio Future objects for interoperability
+    _COROUTINE_TYPES = (CoroWrapper, asyncio.tasks.CoroWrapper)
+else:
+    _COROUTINE_TYPES = CoroWrapper
+
+
 def iscoroutine(obj):
     """Return True if obj is a coroutine object."""
-    return isinstance(obj, CoroWrapper) or inspect.isgenerator(obj)
+    return isinstance(obj, _COROUTINE_TYPES) or inspect.isgenerator(obj)
 
 class FromWrapper(object):
     __slots__ = ('obj',)