python-iterutils 0.1.0__py3-none-any.whl → 0.1.1__py3-none-any.whl

iterutils/__init__.py

@@ -2,16 +2,17 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 0)
+__version__ = (0, 1, 1)
 __all__ = [
-    "iterable", "async_iterable", "foreach", "async_foreach", "through", "async_through", 
-    "flatten", "async_flatten", "chunked", "wrap_iter", "wrap_aiter", "acc_step", 
-    "cut_iter", "run_gen_step", "run_gen_step_iter", "Return", "Yield", "YieldFrom", 
+    "Return", "Yield", "YieldFrom", "iterable", "async_iterable", "foreach", "async_foreach", 
+    "through", "async_through", "flatten", "async_flatten", "chunked", "wrap_iter", "wrap_aiter", 
+    "acc_step", "cut_iter", "run_gen_step", "run_gen_step_iter", "bfs_gen", 
 ]
 
 
 from abc import ABC, abstractmethod
 from asyncio import to_thread
+from collections import deque
 from collections.abc import (
     AsyncIterable, AsyncIterator, Buffer, Callable, Generator, 
     Iterable, Iterator, Sequence, 
@@ -565,3 +566,45 @@ def run_gen_step_iter(
                     close()
     return process()
 
+
+@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
+

{python_iterutils-0.1.0.dist-info → python_iterutils-0.1.1.dist-info}/METADATA

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

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+iterutils/__init__.py,sha256=Wrlg5UDuOta7ixKS3Mh5FT9EZkqKT7sBRXNeujoanDE,18639
+iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_iterutils-0.1.1.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_iterutils-0.1.1.dist-info/METADATA,sha256=DjSEacOQgDGD7CZDcm-uzPLfWz-Gayb5agE0pkZjRUM,1341
+python_iterutils-0.1.1.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
+python_iterutils-0.1.1.dist-info/RECORD,,

python_iterutils-0.1.0.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-iterutils/__init__.py,sha256=m6CldG12romSrteHnbHufKlmbRjzd6LfnsxqrqCETwM,17487
-iterutils/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_iterutils-0.1.0.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_iterutils-0.1.0.dist-info/METADATA,sha256=a1p4ibqdGfFKWEhsNSv80GJwd3PiNr7JNCmVOBEAb2U,1341
-python_iterutils-0.1.0.dist-info/WHEEL,sha256=Nq82e9rUAnEjt98J6MlVmMCZb-t9cYE2Ir1kpBmnWfs,88
-python_iterutils-0.1.0.dist-info/RECORD,,