python-iterutils 0.0.4.5__py3-none-any.whl → 0.0.5__py3-none-any.whl

iterutils/__init__.py

@@ -2,10 +2,11 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 0, 4)
+__version__ = (0, 0, 5)
 __all__ = [
     "iterable", "async_iterable", "foreach", "async_foreach", "through", "async_through", 
-    "wrap_iter", "wrap_aiter", "acc_step", "cut_iter", "run_gen_step", 
+    "wrap_iter", "wrap_aiter", "acc_step", "cut_iter", "run_gen_step", "run_gen_step_iter", 
+    "Yield", "YieldFrom", 
 ]
 
 from asyncio import to_thread
@@ -13,7 +14,7 @@ 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
 
@@ -21,6 +22,18 @@ from asynctools import async_zip, ensure_async, ensure_aiter
 T = TypeVar("T")
 
 
+class Yield:
+
+    def __init__(self, value, /):
+        self.value = value
+
+
+class YieldFrom:
+
+    def __init__(self, value, /):
+        self.value = value
+
+
 def iterable(it, /) -> bool:
     try:
         return isinstance(iter(it), Iterable)
@@ -238,7 +251,7 @@ def run_gen_step(
             func = send(None)
             while True:
                 try:
-                    ret = func() if callable(func) else ret
+                    ret = func() if callable(func) else func
                 except BaseException as e:
                     func = throw(e)
                 else:
@@ -252,3 +265,112 @@ def run_gen_step(
             if close is not None:
                 close()
 
+
+@overload
+def run_gen_step_iter(
+    gen_step: Generator | Callable[[], Generator], 
+    threaded: bool = False, 
+    *, 
+    async_: Literal[False] = False, 
+) -> Iterator:
+    ...
+@overload
+def run_gen_step_iter(
+    gen_step: Generator | Callable[[], Generator], 
+    threaded: bool = False, 
+    *, 
+    async_: Literal[True], 
+) -> AsyncIterator:
+    ...
+def run_gen_step_iter(
+    gen_step: Generator | Callable[[], Generator], 
+    threaded: bool = False, 
+    *, 
+    async_: bool = False, 
+) -> Iterator | AsyncIterator:
+    if callable(gen_step):
+        gen = gen_step()
+        close = gen.close
+    else:
+        gen = gen_step
+        close = None
+    send = gen.send
+    throw = gen.throw
+    if async_:
+        async def process():
+            try:
+                if threaded:
+                    func = await to_thread(send, None)
+                else:
+                    func = send(None)
+                while True:
+                    yield_type = 0
+                    if isinstance(func, Yield):
+                        yield_type = 1
+                    elif isinstance(func, YieldFrom):
+                        yield_type = 2
+                    if yield_type:
+                        func = func.value
+                    try:
+                        if isawaitable(func):
+                            ret = await func
+                        elif callable(func):
+                            ret = func()
+                            if isawaitable(ret):
+                                ret = await ret
+                        else:
+                            ret = func
+                    except BaseException as e:
+                        if threaded:
+                            func = await to_thread(throw, e)
+                        else:
+                            func = throw(e)
+                    else:
+                        if yield_type == 1:
+                            yield ret
+                        elif yield_type == 2:
+                            async for val in ensure_aiter(ret, threaded=threaded):
+                                yield val
+                        if threaded:
+                            func = await to_thread(send, ret)
+                        else:
+                            func = send(ret)
+            except (StopIteration, GeneratorExit):
+                pass
+            finally:
+                if close is not None:
+                    if threaded:
+                        await to_thread(close)
+                    else:
+                        close()
+    else:
+        def process():
+            try:
+                func = send(None)
+                while True:
+                    yield_type = 0
+                    if isinstance(func, Yield):
+                        yield_type = 1
+                    elif isinstance(func, YieldFrom):
+                        yield_type = 2
+                    if yield_type:
+                        func = func.value
+                    try:
+                        ret = func() if callable(func) else func
+                    except BaseException as e:
+                        func = throw(e)
+                    else:
+                        if yield_type == 1:
+                            yield ret
+                        elif yield_type == 2:
+                            yield from ret
+                        func = send(ret)
+            except StopIteration as e:
+                return e.value
+            except GeneratorExit:
+                pass
+            finally:
+                if close is not None:
+                    close()
+    return process()
+

{python_iterutils-0.0.4.5.dist-info → python_iterutils-0.0.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.0.4.5
+Version: 0.0.5
 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.5.dist-info/RECORD

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=2DJxe5pFy9Y8kePciX4wlkmhl4HgYaJKbBNvHk45J1M,11199
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.0.5.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.0.5.dist-info/METADATA,sha256=w5Iy3Izn-dzUcqWxpWzlrEMcGmCFT_aPU6htLB66wHk,1382
+python_iterutils-0.0.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+python_iterutils-0.0.5.dist-info/RECORD,,

python_iterutils-0.0.4.5.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=aUoT5_AkcRUYDM5M9WHckr_URTA01jKhbPDcufH90Ko,7381
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.0.4.5.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.0.4.5.dist-info/METADATA,sha256=feKv9AWPu1xSoVXmPjJN86CY3xUzMLjaUEPbPpW79LY,1384
-python_iterutils-0.0.4.5.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-python_iterutils-0.0.4.5.dist-info/RECORD,,