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

iterutils/__init__.py

@@ -2,30 +2,34 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 3)
+__version__ = (0, 1, 4)
 __all__ = [
     "Return", "Yield", "YieldFrom", "iterable", "async_iterable", "foreach", "async_foreach", 
-    "through", "async_through", "flatten", "async_flatten", "map", "filter", "reduce", "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", 
+    "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", 
 ]
 
 from abc import ABC, abstractmethod
-from asyncio import to_thread
+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, 
     Iterable, Iterator, MutableSet, Sequence, 
 )
-from contextlib import contextmanager
+from contextlib import asynccontextmanager, contextmanager
 from dataclasses import dataclass
 from itertools import batched, pairwise
 from inspect import isawaitable, iscoroutinefunction
-from typing import overload, Any, ContextManager, Literal
+from _thread import start_new_thread
+from time import sleep, time
+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 texttools import format_time
 from undefined import undefined, Undefined
 
 
@@ -76,7 +80,7 @@ def foreach(
     if not (isinstance(iterable, Iterable) and all(isinstance(it, Iterable) for it in iterables)):
         return async_foreach(value, iterable, *iterables)
     if iterables:
-        for args in zip(iterable, *iterables):
+        for args in _zip(iterable, *iterables):
             value(*args)
     else:
         for arg in iterable:
@@ -110,8 +114,7 @@ def through(
         for _ in iterable:
             pass
     else:
-        from builtins import map
-        for v in map(take_while, iterable):
+        for v in _map(take_while, iterable):
             if not v:
                 break
 
@@ -181,30 +184,38 @@ async def async_flatten(
 
 
 def map(
-    function: Callable, 
+    function: None | Callable, 
     iterable: Iterable | AsyncIterable, 
     /, 
     *iterables: Iterable | AsyncIterable, 
-):
+):      
     if (
         iscoroutinefunction(function) or 
         isinstance(iterable, AsyncIterable) or 
         any(isinstance(i, AsyncIterable) for i in iterables)
     ):
+        if function is None:
+            if iterables:
+                return async_zip(iterable, *iterables)
+            else:
+                return iterable
         return async_map(function, iterable, *iterables)
-    from builtins import map
-    return map(function, iterable, *iterables)
+    if function is None:
+        if iterables:
+            return _zip(iterable, *iterables)
+        else:
+            return iterable
+    return _map(function, iterable, *iterables)
 
 
 def filter(
-    function: Callable, 
+    function: None | Callable, 
     iterable: Iterable | AsyncIterable, 
     /, 
 ):
     if iscoroutinefunction(function) or isinstance(iterable, AsyncIterable):
         return async_filter(function, iterable)
-    from builtins import filter
-    return filter(function, iterable)
+    return _filter(function, iterable)
 
 
 def reduce(
@@ -221,6 +232,16 @@ def reduce(
     return reduce(function, iterable, initial)
 
 
+def zip(
+    iterable: Iterable | AsyncIterable, 
+    /, 
+    *iterables: Iterable | AsyncIterable, 
+):
+    if isinstance(iterable, AsyncIterable) or any(isinstance(i, AsyncIterable) for i in iterables):
+        return async_zip(iterable, *iterables)
+    return _zip(iterable, *iterables)
+
+
 @overload
 def chunked[T](
     iterable: Iterable[T], 
@@ -747,3 +768,69 @@ def with_iter_next[T](
         except StopIteration:
             pass
 
+
+@overload
+def backgroud_loop(
+    call: None | Callable = None, 
+    /, 
+    interval: int | float = 0.05, 
+    *, 
+    async_: Literal[False] = False, 
+) -> ContextManager:
+    ...
+@overload
+def backgroud_loop(
+    call: None | Callable = None, 
+    /, 
+    interval: int | float = 0.05, 
+    *, 
+    async_: Literal[True], 
+) -> AsyncContextManager:
+    ...
+def backgroud_loop(
+    call: None | Callable = None, 
+    /, 
+    interval: int | float = 0.05, 
+    *, 
+    async_: Literal[False, True] = False, 
+) -> ContextManager | AsyncContextManager:
+    use_default_call = not callable(call)
+    if use_default_call:
+        start = time()
+        def call():
+            print(f"\r\x1b[K{format_time(time() - start)}", end="")
+    def run():
+        while running:
+            try:
+                yield call
+            except Exception:
+                pass
+            if interval > 0:
+                if async_:
+                    yield async_sleep(interval)
+                else:
+                    sleep(interval)
+    running = True
+    if async_:
+        async def actx():
+            nonlocal running
+            try:
+                task = create_task(run())
+                yield task
+            finally:
+                running = False
+                task.cancel()
+                if use_default_call:
+                    print("\r\x1b[K", end="")
+        return asynccontextmanager(actx)()
+    else:
+        def ctx():
+            nonlocal running
+            try:
+                yield start_new_thread(run, ())
+            finally:
+                running = False
+                if use_default_call:
+                    print("\r\x1b[K", end="")
+        return contextmanager(ctx)()
+

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-iterutils
-Version: 0.1.3
+Version: 0.1.4
 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-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
 Description-Content-Type: text/markdown

python_iterutils-0.1.4.dist-info/RECORD

@@ -0,0 +1,7 @@
+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,,

python_iterutils-0.1.3.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=5oKgPEE_IkYDJs6oz-EbZl4sbjil2KIpif-m9C--7So,22262
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.1.3.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.1.3.dist-info/METADATA,sha256=pjYgGntBIyLr1k2Z2dftvgpWrHHHFmCXSPTS0gJdHIg,1384
-python_iterutils-0.1.3.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.1.3.dist-info/RECORD,,