python-iterutils 0.2.2__tar.gz → 0.2.3__tar.gz

{python_iterutils-0.2.2 → python_iterutils-0.2.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.2.2
+Version: 0.2.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.2.2 → python_iterutils-0.2.3}/iterutils/__init__.py

@@ -2,7 +2,7 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 2, 2)
+__version__ = (0, 2, 3)
 __all__ = [
     "Return", "Yield", "YieldFrom", "iterable", "async_iterable", 
     "foreach", "async_foreach", "through", "async_through", "flatten", 
@@ -11,8 +11,8 @@ __all__ = [
     "iter_unique", "async_iter_unique", "wrap_iter", "wrap_aiter", 
     "acc_step", "cut_iter", "iter_gen_step", "iter_gen_step_async", 
     "run_gen_step", "run_gen_step_sync_iter", "run_gen_step_async_iter", 
-    "run_gen_step_iter", "as_gen_step", "bfs_gen", "with_iter_next", 
-    "backgroud_loop", 
+    "run_gen_step_iter", "as_gen_step", "as_gen_step_iter", "bfs_gen", 
+    "with_iter_next", "backgroud_loop", 
 ]
 
 from abc import ABC, abstractmethod
@@ -813,6 +813,18 @@ async def iter_gen_step_async(
         await close()
 
 
+@overload
+def run_gen_step(
+    gen_step: Generator | Callable[[], Generator], 
+    may_await: bool | Literal[1] = True, 
+    may_call: bool | Literal[1] = True, 
+    threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
+    *, 
+    async_: None = None, 
+):
+    ...
+@overload
 def run_gen_step(
     gen_step: Generator | Callable[[], Generator], 
     may_await: bool | Literal[1] = True, 
@@ -820,7 +832,28 @@ def run_gen_step(
     threaded: bool = False, 
     running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
     *, 
-    async_: None | bool = None, 
+    async_: Literal[False], 
+):
+    ...
+@overload
+def run_gen_step(
+    gen_step: Generator | Callable[[], Generator], 
+    may_await: bool | Literal[1] = True, 
+    may_call: bool | Literal[1] = True, 
+    threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
+    *, 
+    async_: Literal[True], 
+) -> Coroutine:
+    ...
+def run_gen_step(
+    gen_step: Generator | Callable[[], Generator], 
+    may_await: bool | Literal[1] = True, 
+    may_call: bool | Literal[1] = True, 
+    threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
+    *, 
+    async_: None | Literal[False, True] = None, 
 ):
     """
     """
@@ -1013,6 +1046,7 @@ def run_gen_step_iter(
     may_await: bool | Literal[1] = True, 
     may_call: bool | Literal[1] = True, 
     threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
     *, 
     async_: None = None, 
 ) -> Iterator | AsyncIterator:
@@ -1023,6 +1057,7 @@ def run_gen_step_iter(
     may_await: bool | Literal[1] = True, 
     may_call: bool | Literal[1] = True, 
     threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
     *, 
     async_: Literal[False], 
 ) -> Iterator:
@@ -1033,6 +1068,7 @@ def run_gen_step_iter(
     may_await: bool | Literal[1] = True, 
     may_call: bool | Literal[1] = True, 
     threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
     *, 
     async_: Literal[True], 
 ) -> AsyncIterator:
@@ -1042,8 +1078,9 @@ def run_gen_step_iter(
     may_await: bool | Literal[1] = True, 
     may_call: bool | Literal[1] = True, 
     threaded: bool = False, 
+    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
     *, 
-    async_: None | bool = None, 
+    async_: None | Literal[False, True] = None, 
 ) -> Iterator | AsyncIterator:
     """
     """
@@ -1062,38 +1099,41 @@ def run_gen_step_iter(
 
 @optional
 def as_gen_step(
-    func: Callable, 
+    func: Callable[..., Generator], 
     /, 
+    *deco_args, 
     iter: bool = False, 
-    may_await: bool | Literal[1] = True, 
-    may_call: bool | Literal[1] = True, 
-    threaded: bool = False, 
-    running_flag: None | SupportsBool | Callable[[], SupportsBool] | Callable[[], Awaitable[SupportsBool]] = None, 
-    *, 
-    async_: None | bool = None, 
+    **deco_kwds, 
 ) -> Callable:
-    """装饰器，构建一个 gen_step 函数
-    """
-    if async_ is None:
-        async_ = _get_async()
-    def wrapper(*args, **kwds):
-        if iter:
-            return run_gen_step_iter(
-                func(*args, **kwds), 
-                may_await=may_await, 
-                may_call=may_call, 
-                threaded=threaded, 
-                async_=async_, # type: ignore
-            )
-        else:
-            return run_gen_step(
-                func(*args, **kwds), 
-                may_await=may_await, 
-                may_call=may_call, 
-                threaded=threaded, 
-                running_flag=running_flag, 
-                async_=async_, 
-            )
+    """装饰器，把生成器封装成 gen_step 函数
+    """
+    call = run_gen_step_iter if iter else run_gen_step
+    def wrapper(*args, async_: Literal[False, True] = False, **kwds):
+        return call(
+            func(*args, async_=async_, **kwds), 
+            *deco_args, 
+            async_=async_, 
+            **deco_kwds, 
+        )
+    return wrapper
+
+
+@optional
+def as_gen_step_iter(
+    func: Callable[..., Generator], 
+    /, 
+    *deco_args, 
+    **deco_kwds, 
+) -> Callable:
+    """装饰器，把生成器封装成 gen_step 函数
+    """
+    def wrapper(*args, async_: Literal[False, True] = False, **kwds):
+        return run_gen_step_iter(
+            func(*args, async_=async_, **kwds), 
+            *deco_args, 
+            async_=async_, 
+            **deco_kwds, 
+        )
     return wrapper
 
 

{python_iterutils-0.2.2 → python_iterutils-0.2.3}/pyproject.toml

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