python-filewrap 0.1.2__tar.gz → 0.1.3__tar.gz

{python_filewrap-0.1.2 → python_filewrap-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-filewrap
-Version: 0.1.2
+Version: 0.1.3
 Summary: Python file wrappers.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-filewrap
 License: MIT

{python_filewrap-0.1.2 → python_filewrap-0.1.3}/filewrap/__init__.py

@@ -2,12 +2,13 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 1, 2)
+__version__ = (0, 1, 3)
 __all__ = [
     "Buffer", "SupportsRead", "SupportsReadinto", 
     "SupportsWrite", "SupportsSeek", 
     "bio_chunk_iter", "bio_chunk_async_iter", 
     "bio_skip_iter", "bio_skip_async_iter", 
+    "bytes_iter", "bytes_async_iter", 
     "bytes_iter_skip", "bytes_async_iter_skip", 
     "bytes_iter_to_reader", "bytes_iter_to_async_reader", 
     "bytes_to_chunk_iter", "bytes_to_chunk_async_iter", 
@@ -350,6 +351,67 @@ async def bio_skip_async_iter(
         yield length
 
 
+def bytes_iter(
+    it: Iterable[Buffer], 
+    /, 
+    size: int = -1, 
+    callback: None | Callable[[int], Any] = None, 
+) -> Iterator[Buffer]:
+    it = iter(it)
+    if size < 0:
+        yield from it
+        return
+    elif size == 0:
+        return
+    for b in it:
+        l = len(b)
+        if l <= size:
+            yield b
+            if callback is not None:
+                callback(l)
+            if l < size:
+                size -= l
+            else:
+                break
+        else:
+            yield memoryview(b)[:size]
+            if callback is not None:
+                callback(size)
+            break
+
+
+async def bytes_async_iter(
+    it: Iterable[Buffer] | AsyncIterable[Buffer], 
+    /, 
+    size: int = -1, 
+    callback: None | Callable[[int], Any] = None, 
+    threaded: bool = False, 
+) -> AsyncIterator[Buffer]:
+    it = aiter(ensure_aiter(it, threaded=threaded))
+    if size < 0:
+        async for chunk in it:
+            yield chunk
+        return
+    elif size == 0:
+        return
+    callback = ensure_async(callback) if callable(callback) else None
+    async for b in it:
+        l = len(b)
+        if l <= size:
+            yield b
+            if callback is not None:
+                await callback(l)
+            if l < size:
+                size -= l
+            else:
+                break
+        else:
+            yield memoryview(b)[:size]
+            if callback is not None:
+                await callback(size)
+            break
+
+
 def bytes_iter_skip(
     it: Iterable[Buffer], 
     /, 
@@ -379,8 +441,9 @@ async def bytes_async_iter_skip(
     /, 
     size: int = -1, 
     callback: None | Callable[[int], Any] = None, 
+    threaded: bool = False, 
 ) -> AsyncIterator[Buffer]:
-    it = aiter(ensure_aiter(it))
+    it = aiter(ensure_aiter(it, threaded=threaded))
     if size == 0:
         return it
     callback = ensure_async(callback) if callable(callback) else None

{python_filewrap-0.1.2 → python_filewrap-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-filewrap"
-version = "0.1.2"
+version = "0.1.3"
 description = "Python file wrappers."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"