python-iterutils 0.1.4__py3-none-any.whl → 0.1.5__py3-none-any.whl

iterutils/__init__.py

@@ -2,7 +2,7 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 4)
+__version__ = (0, 1, 5)
 __all__ = [
     "Return", "Yield", "YieldFrom", "iterable", "async_iterable", "foreach", "async_foreach", 
     "through", "async_through", "flatten", "async_flatten", "map", "filter", "reduce", "zip", 
@@ -15,10 +15,10 @@ from asyncio import create_task, sleep as async_sleep, to_thread
 from builtins import map as _map, filter as _filter, zip as _zip
 from collections import deque
 from collections.abc import (
-    AsyncIterable, AsyncIterator, Awaitable, Buffer, Callable, Generator, 
+    AsyncIterable, AsyncIterator, Awaitable, Buffer, Callable, Coroutine, Generator, 
     Iterable, Iterator, MutableSet, Sequence, 
 )
-from contextlib import asynccontextmanager, contextmanager
+from contextlib import asynccontextmanager, contextmanager, ExitStack, AsyncExitStack
 from dataclasses import dataclass
 from itertools import batched, pairwise
 from inspect import isawaitable, iscoroutinefunction
@@ -769,6 +769,48 @@ def with_iter_next[T](
             pass
 
 
+@overload
+def context[T](
+    func: Callable[..., T], 
+    *ctxs: ContextManager, 
+    async_: Literal[False] = False, 
+) -> T:
+    ...
+@overload
+def context[T](
+    func: Callable[..., T] | Callable[..., Awaitable[T]], 
+    *ctxs: ContextManager | AsyncContextManager, 
+    async_: Literal[True], 
+) -> Coroutine[Any, Any, T]:
+    ...
+def context[T](
+    func: Callable[..., T] | Callable[..., Awaitable[T]], 
+    *ctxs: ContextManager | AsyncContextManager, 
+    async_: Literal[False, True] = False, 
+) -> T | Coroutine[Any, Any, T]:
+    if async_:
+        async def call():
+            args: list = []
+            add_arg = args.append
+            with ExitStack() as stack:
+                async with AsyncExitStack() as async_stack:
+                    enter = stack.enter_context
+                    async_enter = async_stack.enter_async_context
+                    for ctx in ctxs:
+                        if isinstance(ctx, AsyncContextManager):
+                            add_arg(await async_enter(ctx))
+                        else:
+                            add_arg(enter(ctx))
+                    ret = func(*args)
+                    if isawaitable(ret):
+                        ret = await ret
+                    return ret
+        return call()
+    else:
+        with ExitStack() as stack:
+            return func(*map(stack.enter_context, ctxs)) # type: ignore
+
+
 @overload
 def backgroud_loop(
     call: None | Callable = None, 

{python_iterutils-0.1.4.dist-info → python_iterutils-0.1.5.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=VRL_byJ8rSDBFDYPZFnxmG7AzDJ7EXC_i1AjYi_9RZM,26174
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.1.5.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.1.5.dist-info/METADATA,sha256=tLN1tAcz_WipgczszAbxMcKFImyyiBehvVnjwY_8_sk,1426
+python_iterutils-0.1.5.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+python_iterutils-0.1.5.dist-info/RECORD,,

python_iterutils-0.1.4.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=aYU53GoLQ7CYy-rcts8b47yo6dp0lJncxs5UPRHb_YA,24752
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.1.4.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.1.4.dist-info/METADATA,sha256=1siMQz1thjXxHUIQpTkxst6IY2Y3kfBRdeLeQIWRFOA,1426
-python_iterutils-0.1.4.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.1.4.dist-info/RECORD,,