streamable 2.1.0b2__tar.gz → 2.1.0rc1__tar.gz

{streamable-2.1.0b2/streamable.egg-info → streamable-2.1.0rc1}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamable
-Version: 2.1.0b2
+Version: 2.1.0rc1
 Summary: sync/async iterable streams for Python
 Author-email: ebonnal <bonnal.enzo.dev@gmail.com>
 License: Apache-2.0

{streamable-2.1.0b2 → streamable-2.1.0rc1}/streamable/__init__.py

@@ -12,4 +12,4 @@ ClosableAsyncIterator.__module__ = __name__
 
 __all__ = ["stream", "star", "Observation", "ClosableAsyncIterator"]
 
-__version__ = "2.1.0b2"
+__version__ = "2.1.0rc1"

{streamable-2.1.0b2 → streamable-2.1.0rc1}/streamable/_aiterators.py

@@ -36,7 +36,7 @@ from streamable._tools._afuture import (
     FIFOFutureResults,
     FutureResults,
 )
-from streamable._tools._async import AsyncFunction, anext
+from streamable._tools._async import AsyncFunction, anext, wait_for
 from streamable._tools._context import (
     ExecutorAsyncContextManager,
     NoopAsyncContextManager,
@@ -211,6 +211,7 @@ class _BufferAsyncIterable(AsyncIterable[Union[T, ExceptionContainer]]):
             try:
                 await asyncio.gather(task, return_exceptions=True)
             finally:
+                base_exception.clear()
                 while not buffer.empty():
                     buffer.get_nowait()
 
@@ -438,7 +439,7 @@ class _GroupByWithinAsyncIterable(
                 try:
                     while True:
                         try:
-                            elem = await asyncio.wait_for(
+                            elem = await wait_for(
                                 next_elem.get(), timeout=self._timeout(groups)
                             )
                         except asyncio.TimeoutError:
@@ -485,6 +486,7 @@ class _GroupByWithinAsyncIterable(
             try:
                 await asyncio.gather(task, return_exceptions=True)
             finally:
+                base_exception.clear()
                 while not next_elem.empty():
                     next_elem.get_nowait()
 

{streamable-2.1.0b2 → streamable-2.1.0rc1}/streamable/_iterators.py

@@ -130,6 +130,7 @@ class _BufferIterable(Iterable[Union[T, ExceptionContainer]]):
             stopped.set()
             limit.release()
             thread.join()
+            base_exception.clear()
             buffer.queue.clear()
 
 
@@ -377,6 +378,7 @@ class _GroupByWithinIterable(Iterable[Union[ExceptionContainer, Tuple[U, List[T]
             stopped.set()
             limit.release()
             thread.join()
+            base_exception.clear()
             next_elem.queue.clear()
 
 

streamable-2.1.0rc1/streamable/_tools/_async.py

@@ -0,0 +1,39 @@
+import asyncio
+import sys
+from typing import AsyncIterator, Awaitable, Callable, Coroutine, Optional, TypeVar
+
+T = TypeVar("T")
+R = TypeVar("R")
+
+AsyncFunction = Callable[[T], Coroutine[object, object, R]]
+
+
+# `builtins.anext` for pre 3.10
+async def anext(aiterator: AsyncIterator[T], default: T) -> T:
+    try:
+        return await aiterator.__anext__()
+    except StopAsyncIteration:
+        return default
+
+
+# before 3.12, `asyncio.wait_for` swallowed cancellations when the result was available.
+if sys.version_info >= (3, 12):
+    from asyncio import wait_for
+else:  # pragma: no cover
+
+    async def wait_for(awaitable: Awaitable[T], timeout: Optional[float]) -> T:
+        if timeout is None:
+            return await awaitable
+        task = asyncio.ensure_future(awaitable)
+        try:
+            await asyncio.wait((task,), timeout=timeout)
+        except BaseException:
+            task.cancel()
+            await asyncio.wait((task,))
+            raise
+        if not task.done():
+            task.cancel()
+            await asyncio.wait((task,))
+            if task.cancelled():
+                raise asyncio.TimeoutError
+        return task.result()

{streamable-2.1.0b2 → streamable-2.1.0rc1/streamable.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: streamable
-Version: 2.1.0b2
+Version: 2.1.0rc1
 Summary: sync/async iterable streams for Python
 Author-email: ebonnal <bonnal.enzo.dev@gmail.com>
 License: Apache-2.0

streamable-2.1.0b2/streamable/_tools/_async.py

@@ -1,14 +0,0 @@
-from typing import AsyncIterator, Callable, Coroutine, TypeVar
-
-T = TypeVar("T")
-R = TypeVar("R")
-
-AsyncFunction = Callable[[T], Coroutine[object, object, R]]
-
-
-# `builtins.anext` for pre 3.10
-async def anext(aiterator: AsyncIterator[T], default: T) -> T:
-    try:
-        return await aiterator.__anext__()
-    except StopAsyncIteration:
-        return default