Fix python 3.8 compatibility

vxgmichel · Feb 5, 2020 · 385558f · 385558f
1 parent 21e549a
commit 385558f
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/aiostream/aiter_utils.py b/aiostream/aiter_utils.py
@@ -172,11 +172,12 @@ async def __aexit__(self, typ, value, traceback):
                 # Look for an aclose method
                 aclose = getattr(self._aiterator, 'aclose', None)
 
-                # The ag_running attribute only exists for python >= 3.8
+                # The ag_running attribute has been introduced with python 3.8
                 running = getattr(self._aiterator, 'ag_running', False)
+                closed = not getattr(self._aiterator, 'ag_frame', True)
 
-                # A RuntimeError is raised if aiterator is already running
-                if aclose and not running:
+                # A RuntimeError is raised if aiterator is running or closed
+                if aclose and not running and not closed:
                     try:
                         await aclose()