python-iterutils 0.2.10__py3-none-any.whl → 0.2.10.2__py3-none-any.whl

iterutils/__init__.py

@@ -12,17 +12,18 @@ __all__ = [
     "flatten", "async_flatten", "collect", "async_collect", 
     "group_collect", "async_group_collect", "iter_unique", 
     "async_iter_unique", "wrap_iter", "wrap_aiter", "peek_iter", "peek_aiter", 
-    "acc_step", "cut_iter", "bfs_gen", "context", "backgroud_loop", 
+    "acc_step", "cut_iter", "context", "backgroud_loop", "gen_startup", 
+    "async_gen_startup", "do_iter", "do_aiter", "bfs_iter", "bfs_gen", 
 ]
 
 from asyncio import create_task, sleep as async_sleep
 from builtins import map as _map, filter as _filter, zip as _zip
 from collections import defaultdict, deque
 from collections.abc import (
-    AsyncIterable, AsyncIterator, Awaitable, Buffer, Callable, 
-    Collection, Container, Coroutine, Generator, Iterable, Iterator, 
-    Mapping, MutableMapping, MutableSet, MutableSequence, Sequence, 
-    ValuesView, 
+    AsyncGenerator, AsyncIterable, AsyncIterator, Awaitable, Buffer,
+    Callable, Collection, Container, Coroutine, Generator, Iterable, 
+    Iterator, Mapping, MutableMapping, MutableSet, MutableSequence, 
+    Sequence, ValuesView, 
 )
 from contextlib import (
     asynccontextmanager, contextmanager, ExitStack, AsyncExitStack, 
@@ -30,6 +31,7 @@ from contextlib import (
 )
 from copy import copy
 from dataclasses import dataclass
+from functools import update_wrapper
 from itertools import batched, chain as _chain, pairwise
 from inspect import iscoroutinefunction, signature
 from sys import _getframe, exc_info
@@ -624,26 +626,52 @@ def chain[T](
 @overload
 def chain_from_iterable[T](
     iterables: Iterable[Iterable[T]], 
-    threaded: Literal[False] = False, 
+    threaded: bool = False, 
     *, 
     async_: Literal[False] = False, 
 ) -> Iterator[T]:
     ...
 @overload
 def chain_from_iterable[T](
-    iterables: Iterable[Iterable[T] | AsyncIterable[T]] | AsyncIterable[Iterable[T] | AsyncIterable[T]], 
+    iterables: (
+        AsyncIterable[Iterable[T]] | 
+        AsyncIterable[AsyncIterable[T]] | 
+        AsyncIterable[Iterable[T] | AsyncIterable[T]]
+    ), 
+    threaded: bool = False, 
+    *, 
+    async_: bool = False, 
+) -> AsyncIterator[T]:
+    ...
+@overload
+def chain_from_iterable[T](
+    iterables: (
+        Iterable[Iterable[T]] | 
+        Iterable[AsyncIterable[T]] | 
+        Iterable[Iterable[T] | AsyncIterable[T]] | 
+        AsyncIterable[Iterable[T]] | 
+        AsyncIterable[AsyncIterable[T]] | 
+        AsyncIterable[Iterable[T] | AsyncIterable[T]]
+    ), 
     threaded: bool = False, 
     *, 
     async_: Literal[True], 
 ) -> AsyncIterator[T]:
     ...
 def chain_from_iterable[T](
-    iterables: Iterable[Iterable[T]] | Iterable[Iterable[T] | AsyncIterable[T]] | AsyncIterable[Iterable[T] | AsyncIterable[T]], 
+    iterables: (
+        Iterable[Iterable[T]] | 
+        Iterable[AsyncIterable[T]] | 
+        Iterable[Iterable[T] | AsyncIterable[T]] | 
+        AsyncIterable[Iterable[T]] | 
+        AsyncIterable[AsyncIterable[T]] | 
+        AsyncIterable[Iterable[T] | AsyncIterable[T]]
+    ), 
     threaded: bool = False, 
     *, 
     async_: Literal[False, True] = False, 
 ) -> Iterator[T] | AsyncIterator[T]:
-    if async_ or threaded or not isinstance(iterables, Iterable):
+    if async_ or not isinstance(iterables, Iterable):
         if isinstance(iterables, Iterable):
             return async_chain.from_iterable(iterables, threaded=threaded)
         else:
@@ -1264,48 +1292,6 @@ def cut_iter(
         yield stop, stop - start
 
 
-@overload
-def bfs_gen[T](
-    initial: T, 
-    /, 
-    unpack_iterator: Literal[False] = False, 
-) -> Generator[T, T | None, None]:
-    ...
-@overload
-def bfs_gen[T](
-    initial: T | Iterator[T], 
-    /, 
-    unpack_iterator: Literal[True], 
-) -> Generator[T, T | None, None]:
-    ...
-def bfs_gen[T](
-    initial: T | Iterator[T], 
-    /, 
-    unpack_iterator: bool = False, 
-) -> Generator[T, T | None, None]:
-    """辅助函数，返回生成器，用来简化广度优先遍历
-    """
-    dq: deque[T] = deque()
-    push, pushmany, pop = dq.append, dq.extend, dq.popleft
-    if isinstance(initial, Iterator) and unpack_iterator:
-        pushmany(initial)
-    else:
-        push(initial) # type: ignore
-    while dq:
-        args: None | T = yield (val := pop())
-        if unpack_iterator:
-            while args is not None:
-                if isinstance(args, Iterator):
-                    pushmany(args)
-                else:
-                    push(args)
-                args = yield val
-        else:
-            while args is not None:
-                push(args)
-                args = yield val
-
-
 @overload
 def context[T](
     func: Callable[..., T], 
@@ -1445,3 +1431,70 @@ def backgroud_loop(
                     print("\r\x1b[K", end="")
         return ctx()
 
+
+def gen_startup[**Args, G: Generator](func: Callable[Args, G], /):
+    def wrapper(*args: Args.args, **kwds: Args.kwargs) -> G:
+        gen = func(*args, **kwds)
+        next(gen)
+        return gen
+    return update_wrapper(wrapper, func)
+
+
+def async_gen_startup[**Args, G: AsyncGenerator](func: Callable[Args, G], /):
+    async def wrapper(*args: Args.args, **kwds: Args.kwargs) -> G:
+        gen = func(*args, **kwds)
+        await anext(gen)
+        return gen
+    return update_wrapper(wrapper, func)
+
+
+def do_iter[T](
+    func: Callable[[], T] | Iterable[T], 
+    /, 
+    sentinel=undefined, 
+    sentinel_excs: type[BaseException] | tuple[type[BaseException], ...] = (), 
+) -> Iterator[T]:
+    try:
+        yield from iter(func if callable(func) else iter(func).__next__, sentinel)
+    except sentinel_excs:
+        pass
+
+
+async def do_aiter[T](
+    func: Callable[[], T] | Callable[[], Awaitable[T]] | Iterable[T] | AsyncIterable[T],  
+    /, 
+    sentinel=undefined, 
+    sentinel_excs: type[BaseException] | tuple[type[BaseException], ...] = (), 
+) -> AsyncIterator[T]:
+    if callable(func):
+        func = ensure_async(func)
+    else:
+        func = ensure_aiter(func).__anext__
+    try:
+        while True:
+            v = await func()
+            if v is sentinel:
+                break
+            yield v
+    except StopAsyncIteration:
+        pass
+    except sentinel_excs:
+        pass
+
+
+def bfs_iter[T](*initials: T) -> tuple[Iterator[T], Callable[[T], None]]:
+    dq = deque(initials)
+    return do_iter(dq.popleft, sentinel_excs=IndexError), dq.append
+
+
+@gen_startup
+def bfs_gen[T](*initials) -> Generator[None | T, T | None, None]:
+    dq = deque(initials)
+    push, pop = dq.append, dq.popleft
+    try:
+        p = yield None
+        while True:
+            p = yield (pop() if p is None else push(p))
+    except IndexError:
+        pass
+

{python_iterutils-0.2.10.dist-info → python_iterutils-0.2.10.2.dist-info}/METADATA

@@ -1,9 +1,9 @@
-Metadata-Version: 2.1
+Metadata-Version: 2.4
 Name: python-iterutils
-Version: 0.2.10
+Version: 0.2.10.2
 Summary: Python another itertools.
-Home-page: https://github.com/ChenyangGao/python-modules/tree/main/python-iterutils
 License: MIT
+License-File: LICENSE
 Keywords: iterable,iterutils
 Author: ChenyangGao
 Author-email: wosiwujm@gmail.com
@@ -16,6 +16,7 @@ Classifier: Programming Language :: Python
 Classifier: Programming Language :: Python :: 3
 Classifier: Programming Language :: Python :: 3.12
 Classifier: Programming Language :: Python :: 3.13
+Classifier: Programming Language :: Python :: 3.14
 Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
@@ -23,6 +24,7 @@ Classifier: Topic :: Software Development :: Libraries :: Python Modules
 Requires-Dist: python-asynctools (>=0.1.3.4)
 Requires-Dist: python-texttools (>=0.0.3)
 Requires-Dist: python-undefined (>=0.0.3)
+Project-URL: Homepage, https://github.com/ChenyangGao/python-modules/tree/main/python-iterutils
 Project-URL: Repository, https://github.com/ChenyangGao/python-modules/tree/main/python-iterutils
 Description-Content-Type: text/markdown
 

python_iterutils-0.2.10.2.dist-info/RECORD

@@ -0,0 +1,6 @@
+iterutils/__init__.py,sha256=vVWjJwlpdML_9QJOwIbZkOf5v68Ngvxq5PuPhaDmWMI,42422
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.2.10.2.dist-info/METADATA,sha256=739Kkv7nn-POyjfaNQTQk4n0oHjot_7Xni65P90a46Q,1485
+python_iterutils-0.2.10.2.dist-info/WHEEL,sha256=zp0Cn7JsFoX2ATtOhtaFYIiE2rmFAD4OcMhtUki8W3U,88
+python_iterutils-0.2.10.2.dist-info/licenses/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.2.10.2.dist-info/RECORD,,

{python_iterutils-0.2.10.dist-info → python_iterutils-0.2.10.2.dist-info}/WHEEL

@@ -1,4 +1,4 @@
 Wheel-Version: 1.0
-Generator: poetry-core 1.9.1
+Generator: poetry-core 2.2.1
 Root-Is-Purelib: true
 Tag: py3-none-any

python_iterutils-0.2.10.dist-info/LICENSE

@@ -1,21 +0,0 @@
-MIT License
-
-Copyright (c) 2024 ChenyangGao <https://github.com/ChenyangGao>
-
-Permission is hereby granted, free of charge, to any person obtaining a copy
-of this software and associated documentation files (the "Software"), to deal
-in the Software without restriction, including without limitation the rights
-to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
-copies of the Software, and to permit persons to whom the Software is
-furnished to do so, subject to the following conditions:
-
-The above copyright notice and this permission notice shall be included in all
-copies or substantial portions of the Software.
-
-THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
-IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
-FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
-AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
-LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
-OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
-SOFTWARE.

python_iterutils-0.2.10.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=7bUS_NH2raU-_Cw-pDM224pC7je1kP1K0dnRmO4kHYU,40941
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.2.10.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.2.10.dist-info/METADATA,sha256=yA2qW_0Ae5UtLUJp7u1OQzGup4hVpkKYxS4meiD-c8E,1398
-python_iterutils-0.2.10.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.2.10.dist-info/RECORD,,