python-iterutils 0.1.6__py3-none-any.whl → 0.1.7__py3-none-any.whl

iterutils/__init__.py

@@ -2,7 +2,7 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 6)
+__version__ = (0, 1, 7)
 __all__ = [
     "Return", "Yield", "YieldFrom", "iterable", "async_iterable", "foreach", "async_foreach", 
     "through", "async_through", "flatten", "async_flatten", "map", "filter", "reduce", "zip", 
@@ -23,8 +23,10 @@ from contextlib import asynccontextmanager, contextmanager, ExitStack, AsyncExit
 from dataclasses import dataclass
 from itertools import batched, pairwise
 from inspect import isawaitable, iscoroutinefunction
+from sys import _getframe
 from _thread import start_new_thread
 from time import sleep, time
+from types import FrameType
 from typing import overload, Any, AsyncContextManager, ContextManager, Literal
 
 from asynctools import (
@@ -452,13 +454,32 @@ def cut_iter(
         yield stop, stop - start
 
 
+def _get_async(back: int = 2) -> bool:
+    f: None | FrameType
+    f = _getframe(back)
+    f_globals = f.f_globals
+    f_locals  = f.f_locals
+    if f_locals is f_globals:
+        return f_locals.get("async_") or False
+    while f_locals is not f_globals:
+        if "async_" in f_locals:
+            return f_locals["async_"] or False
+        f = f.f_back
+        if f is None:
+            break
+        f_locals = f.f_locals
+    return False
+
+
 def run_gen_step[T](
     gen_step: Generator[Any, Any, T] | Callable[[], Generator[Any, Any, T]], 
     *, 
-    async_: bool = False, 
+    async_: None | bool = None, 
     threaded: bool = False, 
     as_iter: bool = False, 
 ) -> T:
+    if async_ is None:
+        async_ = _get_async()
     if callable(gen_step):
         gen = gen_step()
         close = gen.close
@@ -561,7 +582,15 @@ def run_gen_step_iter(
     gen_step: Generator | Callable[[], Generator], 
     threaded: bool = False, 
     *, 
-    async_: Literal[False] = False, 
+    async_: None = None, 
+) -> Iterator | AsyncIterator:
+    ...
+@overload
+def run_gen_step_iter(
+    gen_step: Generator | Callable[[], Generator], 
+    threaded: bool = False, 
+    *, 
+    async_: Literal[False], 
 ) -> Iterator:
     ...
 @overload
@@ -576,8 +605,10 @@ def run_gen_step_iter(
     gen_step: Generator | Callable[[], Generator], 
     threaded: bool = False, 
     *, 
-    async_: bool = False, 
+    async_: None | bool = None, 
 ) -> Iterator | AsyncIterator:
+    if async_ is None:
+        async_ = _get_async()
     if callable(gen_step):
         gen = gen_step()
         close = gen.close
@@ -687,9 +718,11 @@ def run_gen_step_iter(
 def as_gen_step(
     func: Callable, 
     /, 
-    async_: bool = False, 
+    async_: None | bool = None, 
     iter: bool = False, 
 ) -> Callable:
+    if async_ is None:
+        async_ = _get_async()
     def wrapper(*args, **kwds):
         if iter:
             return run_gen_step_iter(func(*args, **kwds), async_=async_) # type: ignore

{python_iterutils-0.1.6.dist-info → python_iterutils-0.1.7.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.1.6
+Version: 0.1.7
 Summary: Python another itertools.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-iterutils
 License: MIT
@@ -21,6 +21,7 @@ Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Dist: python-asynctools (>=0.0.10)
+Requires-Dist: python-decotools (>=0.0.2)
 Requires-Dist: python-texttools (>=0.0.3)
 Requires-Dist: python-undefined (>=0.0.3)
 Project-URL: Repository, https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-iterutils

python_iterutils-0.1.7.dist-info/RECORD

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=BR4ZaZouhl3o9e6ZrBLLEBRfNJzKik8Et-B38Lw_yf0,27444
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.1.7.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.1.7.dist-info/METADATA,sha256=DkEirEtm5-ycdVDBZwb_o-0poNqQDA8Gv5QfJ5o4SwA,1468
+python_iterutils-0.1.7.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+python_iterutils-0.1.7.dist-info/RECORD,,

python_iterutils-0.1.6.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=q5UPgJ2fSDIMQIRFyKlgLqWDiy-b_G8Ft6BSFz_lH3Y,26588
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.1.6.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.1.6.dist-info/METADATA,sha256=XfsSpEENF-pQdLvFDu4TwX60uxwvfqYIWFpdYNqlqkU,1426
-python_iterutils-0.1.6.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.1.6.dist-info/RECORD,,