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

iterutils/__init__.py

@@ -2,13 +2,15 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 0, 3)
+__version__ = (0, 0, 4)
 __all__ = [
     "iterable", "async_iterable", "foreach", "async_foreach", "through", "async_through", 
-    "wrap_iter", "wrap_aiter", "acc_step", "cut_iter", 
+    "wrap_iter", "wrap_aiter", "acc_step", "cut_iter", "run_gen_step", 
 ]
 
-from collections.abc import AsyncIterable, AsyncIterator, Callable, Iterable, Iterator
+from asyncio import to_thread
+from collections.abc import AsyncIterable, AsyncIterator, Callable, Generator, Iterable, Iterator
+from inspect import isawaitable
 from typing import Any, TypeVar
 
 from asynctools import async_zip, ensure_async, ensure_aiter
@@ -162,3 +164,52 @@ def cut_iter(
     if start != stop:
         yield stop, stop - start
 
+
+def run_gen_step(
+    gen_step: Generator[Callable, Any, T] | Callable[[], Generator[Callable, Any, T]], 
+    *, 
+    async_: bool = False, 
+    threaded: bool = False, 
+) -> T:
+    if callable(gen_step):
+        gen = gen_step()
+        close = gen.close
+    else:
+        gen = gen_step
+        close = None
+    send = gen.send
+    if async_:
+        async def process():
+            try:
+                if threaded:
+                    func = await to_thread(send, None)
+                else:
+                    func = send(None)
+                while True:
+                    ret = func()
+                    if isawaitable(ret):
+                        ret = await ret
+                    if threaded:
+                        func = await to_thread(send, ret)
+                    else:
+                        func = send(ret)
+            except StopIteration as e:
+                return e.value
+            finally:
+                if close is not None:
+                    if threaded:
+                        await to_thread(close)
+                    else:
+                        close()
+        return process()
+    else:
+        try:
+            func = send(None)
+            while True:
+                func = send(func())
+        except StopIteration as e:
+            return e.value
+        finally:
+            if close is not None:
+                close()
+

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

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

@@ -0,0 +1,8 @@
+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,,

python_iterutils-0.0.3.1.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=OkA4DRvWukHvkyV4K7kU3PcRFPNcpnvcwEx3aQJ7two,4565
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.0.3.1.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.0.3.1.dist-info/METADATA,sha256=Gh6Hp95ieNcahFQO0oJbEnTXeo3Bydz5vgo1D-cVV7w,1384
-python_iterutils-0.0.3.1.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-python_iterutils-0.0.3.1.dist-info/RECORD,,