python-iterutils 0.2.5__py3-none-any.whl → 0.2.5.2__py3-none-any.whl

iterutils/__init__.py

@@ -7,8 +7,9 @@ __all__ = [
     "Yield", "YieldFrom", "iterable", "async_iterable", 
     "run_gen_step", "run_gen_step_iter", "as_gen_step", 
     "as_gen_step_iter", "split_cm, ""with_iter_next", 
-    "map", "filter", "reduce", "zip", "chain", "chunked", 
-    "foreach", "async_foreach", "through", "async_through", 
+    "map", "filter", "reduce", "zip", "chain", 
+    "chain_from_iterable", "chunked", "foreach", 
+    "async_foreach", "through", "async_through", 
     "flatten", "async_flatten", "collect", "async_collect", 
     "group_collect", "async_group_collect", "iter_unique", 
     "async_iter_unique", "wrap_iter", "wrap_aiter", "acc_step", 
@@ -74,24 +75,27 @@ def async_iterable(obj, /) -> bool:
     return isinstance(obj, AsyncIterable)
 
 
-def _get_async(back: int = 2, /) -> bool:
+def _coalesce(vals, default=None):
+    for val in vals:
+        if val is not None:
+            return val
+    return default
+
+
+@overload
+def _get_async(back: int = 2, /, *, default: Literal[False] = False) -> bool:
+    ...
+@overload
+def _get_async[T](back: int = 2, /, *, default: T) -> bool | T:
+    ...
+def _get_async[T](back: int = 2, /, *, default: Literal[False] | T = False) -> bool | T:
     """往上查找，从最近的调用栈的命名空间中获取 `async_` 的值
     """
-    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 None and f_locals is not f_globals:
-        if "async_" in f_locals:
-            if f_locals["async_"] is not None:
-                return f_locals["async_"]
-        f = f.f_back
-        if f is None:
-            break
-        f_locals = f.f_locals
-    return False
+    def iter_frams(f: None | FrameType = _getframe(back)):
+        while f:
+            yield f.f_locals.get("async_")
+            f = f.f_back
+    return _coalesce(iter_frams(), default)
 
 
 def _run_gen_step(gen: Generator, /):
@@ -270,10 +274,14 @@ def as_gen_step[**Args](
     gen_step: Callable[Args, Generator], 
     /, 
 ) -> Callable[Args, Any]:
+    default_async = _get_async(default=None)
     def wrapper(*args: Args.args, **kwds: Args.kwargs):
         return run_gen_step(
             gen_step, 
-            cast(None | Literal[False, True], kwds.pop("async_", None)), 
+            _coalesce((
+                kwds.pop("async_", None), 
+                _get_async(default=default_async), 
+            )), 
             *args, 
             **kwds, 
         )
@@ -284,10 +292,14 @@ def as_gen_step_iter[**Args](
     gen_step: Callable[Args, Generator], 
     /, 
 ) -> Callable[Args, Iterable | AsyncIterable]:
+    default_async = _get_async(default=None)
     def wrapper(*args: Args.args, **kwds: Args.kwargs):
         return run_gen_step_iter(
             gen_step, 
-            cast(None | Literal[False, True], kwds.pop("async_", None)), 
+            _coalesce((
+                kwds.pop("async_", None), 
+                _get_async(default=default_async), 
+            )), 
             *args, 
             **kwds, 
         )
@@ -533,6 +545,8 @@ def chain_from_iterable[T](
             return async_chain.from_iterable(iterables, threaded=threaded)
     return _chain.from_iterable(iterables) # type: ignore
 
+setattr(chain, "from_iterable", chain_from_iterable)
+
 
 @overload
 def chunked[T](

{python_iterutils-0.2.5.dist-info → python_iterutils-0.2.5.2.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=PCXCgTYvEOJ5GCXeZEILDFYfCk9O5elECgTLJXWYpxw,33637
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.2.5.2.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.2.5.2.dist-info/METADATA,sha256=JM7iFXIRu01IwlLEMQsJq6NZeW56MNPHH3kl8WseTas,1429
+python_iterutils-0.2.5.2.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+python_iterutils-0.2.5.2.dist-info/RECORD,,

python_iterutils-0.2.5.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=GJWB55mT2qVPXmSKIR0qfVQXiH2_7wZwpQTLVSNmqwg,33272
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.2.5.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.2.5.dist-info/METADATA,sha256=boKzeiFl7sWgEEk0XU9MShqgHHzbUbqAn37UWikXmrE,1427
-python_iterutils-0.2.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.2.5.dist-info/RECORD,,