python-iterutils 0.1.5__tar.gz → 0.1.6__tar.gz

{python_iterutils-0.1.5 → python_iterutils-0.1.6}/PKG-INFO

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

{python_iterutils-0.1.5 → python_iterutils-0.1.6}/iterutils/__init__.py

@@ -2,12 +2,13 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 5)
+__version__ = (0, 1, 6)
 __all__ = [
     "Return", "Yield", "YieldFrom", "iterable", "async_iterable", "foreach", "async_foreach", 
     "through", "async_through", "flatten", "async_flatten", "map", "filter", "reduce", "zip", 
     "chunked", "iter_unique", "async_iter_unique", "wrap_iter", "wrap_aiter", "acc_step", 
-    "cut_iter", "run_gen_step", "run_gen_step_iter", "bfs_gen", "with_iter_next", "backgroud_loop", 
+    "cut_iter", "run_gen_step", "run_gen_step_iter", "as_gen_step", "bfs_gen", "with_iter_next", 
+    "backgroud_loop", 
 ]
 
 from abc import ABC, abstractmethod
@@ -29,6 +30,7 @@ from typing import overload, Any, AsyncContextManager, ContextManager, Literal
 from asynctools import (
     async_filter, async_map, async_reduce, async_zip, async_batched, ensure_async, ensure_aiter, 
 )
+from decotools import optional
 from texttools import format_time
 from undefined import undefined, Undefined
 
@@ -681,6 +683,21 @@ def run_gen_step_iter(
     return process()
 
 
+@optional
+def as_gen_step(
+    func: Callable, 
+    /, 
+    async_: bool = False, 
+    iter: bool = False, 
+) -> Callable:
+    def wrapper(*args, **kwds):
+        if iter:
+            return run_gen_step_iter(func(*args, **kwds), async_=async_) # type: ignore
+        else:
+            return run_gen_step(func(*args, **kwds), async_=async_)
+    return wrapper
+
+
 @overload
 def bfs_gen[T](
     initial: T, 

{python_iterutils-0.1.5 → python_iterutils-0.1.6}/pyproject.toml

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