python-iterutils 0.0.4__py3-none-any.whl → 0.0.4.2__py3-none-any.whl

iterutils/__init__.py

@@ -9,9 +9,11 @@ __all__ = [
 ]
 
 from asyncio import to_thread
-from collections.abc import AsyncIterable, AsyncIterator, Callable, Generator, Iterable, Iterator
+from collections.abc import (
+    AsyncIterable, AsyncIterator, Awaitable, Callable, Generator, Iterable, Iterator, 
+)
 from inspect import isawaitable
-from typing import Any, TypeVar
+from typing import overload, Any, Literal, TypeVar
 
 from asynctools import async_zip, ensure_async, ensure_aiter
 
@@ -165,12 +167,31 @@ def cut_iter(
         yield stop, stop - start
 
 
+@overload
 def run_gen_step(
     gen_step: Generator[Callable, Any, T] | Callable[[], Generator[Callable, Any, T]], 
     *, 
-    async_: bool = False, 
+    async_: Literal[False] = False, 
     threaded: bool = False, 
+    as_iter: bool = False, 
 ) -> T:
+    ...
+@overload
+def run_gen_step(
+    gen_step: Generator[Callable, Any, T] | Callable[[], Generator[Callable, Any, T]], 
+    *, 
+    async_: Literal[True], 
+    threaded: bool = False, 
+    as_iter: bool = False, 
+) -> Awaitable[T]:
+    ...
+def run_gen_step(
+    gen_step: Generator[Callable, Any, T] | Callable[[], Generator[Callable, Any, T]], 
+    *, 
+    async_: bool = False, 
+    threaded: bool = False, 
+    as_iter: bool = False, 
+) -> T | Awaitable[T]:
     if callable(gen_step):
         gen = gen_step()
         close = gen.close
@@ -178,6 +199,7 @@ def run_gen_step(
         gen = gen_step
         close = None
     send = gen.send
+    throw = gen.throw
     if async_:
         async def process():
             try:
@@ -186,15 +208,33 @@ def run_gen_step(
                 else:
                     func = send(None)
                 while True:
-                    ret = func()
-                    if isawaitable(ret):
-                        ret = await ret
-                    if threaded:
-                        func = await to_thread(send, ret)
+                    try:
+                        ret = func()
+                        if isawaitable(ret):
+                            ret = await ret
+                    except BaseException as e:
+                        if threaded:
+                            func = await to_thread(throw, e)
+                        else:
+                            func = throw(e)
                     else:
-                        func = send(ret)
+                        if threaded:
+                            func = await to_thread(send, ret)
+                        else:
+                            func = send(ret)
             except StopIteration as e:
-                return e.value
+                result = e.value
+                if as_iter:
+                    if isawaitable(result):
+                        result = await result
+                    try:
+                        result = aiter(result)
+                    except TypeError:
+                        async def wrap(it):
+                            for val in iter(it):
+                                yield val
+                        result = wrap(result)
+                return result
             finally:
                 if close is not None:
                     if threaded:
@@ -206,9 +246,17 @@ def run_gen_step(
         try:
             func = send(None)
             while True:
-                func = send(func())
+                try:
+                    ret = func()
+                except BaseException as e:
+                    func = throw(e)
+                else:
+                    func = send(ret)
         except StopIteration as e:
-            return e.value
+            result = e.value
+            if as_iter:
+                result = iter(result)
+            return result
         finally:
             if close is not None:
                 close()

{python_iterutils-0.0.4.dist-info → python_iterutils-0.0.4.2.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.0.4
+Version: 0.0.4.2
 Summary: Python another itertools.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-iterutils
 License: MIT

python_iterutils-0.0.4.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=gB8PhcSVF_WfAHRPfHvjTLbO9-A843pA1vdahfKBRpw,7593
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.0.4.2.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.0.4.2.dist-info/METADATA,sha256=54xnC2-fhIS14RNphRtI_c_xnLUuT0CDPmjgUPZnZ30,1384
+python_iterutils-0.0.4.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+python_iterutils-0.0.4.2.dist-info/RECORD,,

python_iterutils-0.0.4.dist-info/RECORD

@@ -1,8 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/.DS_Store,sha256=Kj3OcVIE2j02GFjUj8pr1YvdK4YSjlslELr_iRcRv68,6148
-iterutils/__init__.py,sha256=M1KlG0BMF4oZIP4QU18m5riMnWO0oJKNP_SqVDlWKJo,6033
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.0.4.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.0.4.dist-info/METADATA,sha256=3CGonLEb1lXVVrsrd-08p1vMexFfRTdE6MxeCneF4vU,1382
-python_iterutils-0.0.4.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-python_iterutils-0.0.4.dist-info/RECORD,,