Fix confusing traceback when calling wait_stream on a empty stream

Issue #76, PR #77
vxgmichel · May 20, 2021 · 871d1b1 · 871d1b1
2 parents bc9181c + 0d823f5
commit 871d1b1
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/aiostream/core.py b/aiostream/core.py
@@ -29,12 +29,12 @@ async def wait_stream(aiterable):
     A StreamEmpty exception is raised if the sequence is empty.
     """
     async with streamcontext(aiterable) as streamer:
+        last_item = unnassigned = object()
         async for item in streamer:
-            item
-        try:
-            return item
-        except NameError:
+            last_item = item
+        if last_item is unnassigned:
             raise StreamEmpty()
+        return last_item
 
 
 # Core objects