Iterator support

Author: smurfix

trio_asyncio/base.py

@@ -215,6 +215,23 @@ async def run_coroutine(self, coro):
     def wrap_generator(self, gen, *args):
         return run_generator(self, gen(*args))
 
+    def run_iterator(self, aiter):
+        """Call an asyncio iterator from Trio.
+
+        Example Usage::
+            async def slow_nums():
+                n = 0
+                while True:
+                    asyncio.sleep(1)
+                    yield n
+                    n += 1
+
+            async def trio_code(loop):
+                async for n in loop.run_iterator(slow_nums()):
+                    print(n)
+        """
+        return run_generator(self, aiter)
+
     async def run_asyncio(self, proc, *args):
         """Run an asyncio function or method from Trio.
 

trio_asyncio/loop.py

@@ -27,6 +27,7 @@
     'run_coroutine',
     'run_asyncio',
     'wrap_generator',
+    'run_iterator',
     'TrioChildWatcher',
     'TrioPolicy',
 ]
@@ -226,6 +227,12 @@ def wrap_generator(proc, *args):
         raise RuntimeError("Need to run in a trio_asyncio.open_loop() context")
     return loop.wrap_generator(proc, *args)
 
+def run_iterator(aiter):
+    loop = asyncio.get_event_loop()
+    if not isinstance(loop, TrioEventLoop):
+        raise RuntimeError("Need to run in a trio_asyncio.open_loop() context")
+    return loop.run_iterator(aiter)
+
 
 async def run_asyncio(proc, *args):
     """Run an asyncio function or method from Trio.