First pass at iterable Emissions objects

Fixes #67.
altendky · Jul 4, 2020 · b791c29 · b791c29
1 parent a6b5d78
commit b791c29
Show file tree

Hide file tree

Showing 4 changed files with 7 additions and 4 deletions.
diff --git a/qtrio/_core.py b/qtrio/_core.py
@@ -147,6 +147,9 @@ async def aclose(self):
         """
         return await self.send_channel.aclose()
 
+    def __aiter__(self):
+        return self.channel.__aiter__()
+
 
 @async_generator.asynccontextmanager
 async def open_emissions_channel(

diff --git a/qtrio/_tests/test_core.py b/qtrio/_tests/test_core.py
@@ -726,7 +726,7 @@ async def test(request):
             await emissions.aclose()
 
             async with emissions.channel:
-                async for emission in emissions.channel:
+                async for emission in emissions:
                     [value] = emission.args
                     results.append(value)
 
@@ -765,7 +765,7 @@ async def test(request):
             await emissions.aclose()
 
             async with emissions.channel:
-                async for emission in emissions.channel:
+                async for emission in emissions:
                     [value] = emission.args
                     results.append(value)
 

diff --git a/qtrio/examples/_tests/test_crossingpaths.py b/qtrio/examples/_tests/test_crossingpaths.py
@@ -27,7 +27,7 @@ def setText(self, *args, **kwargs):
 
         async def user():
             async with emissions.channel:
-                async for emission in emissions.channel:
+                async for emission in emissions:
                     [text] = emission.args
                     results.append(text)
 

diff --git a/qtrio/examples/emissions.py b/qtrio/examples/emissions.py
@@ -108,7 +108,7 @@ async def main(window=None):
         window.show()
 
         async with emissions.channel:
-            async for emission in emissions.channel:
+            async for emission in emissions:
                 if emission.is_from(window.decrement.clicked):
                     window.decrement_count()
                 elif emission.is_from(window.increment.clicked):