python-iterutils 0.0.4.1__tar.gz → 0.0.4.3__tar.gz

{python_iterutils-0.0.4.1 → python_iterutils-0.0.4.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.0.4.1
+Version: 0.0.4.3
 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.1 → python_iterutils-0.0.4.3}/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
@@ -209,7 +230,22 @@ def run_gen_step(
                         await to_thread(close)
                     else:
                         close()
-        return process()
+        result = process()
+        if as_iter:
+            async def wrap(result):
+                it = await result
+                try:
+                    it = aiter(it)
+                except TypeError:
+                    for val in iter(it):
+                        if isawaitable(val):
+                            val = await val
+                        yield val
+                else:
+                    async for val in it:
+                        yield val
+            result = wrap(result)
+        return result
     else:
         try:
             func = send(None)
@@ -221,7 +257,10 @@ def run_gen_step(
                 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.1 → python_iterutils-0.0.4.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-iterutils"
-version = "0.0.4.1"
+version = "0.0.4.3"
 description = "Python another itertools."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"