python-filewrap 0.0.6__tar.gz → 0.0.7__tar.gz

{python_filewrap-0.0.6 → python_filewrap-0.0.7}/PKG-INFO

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

{python_filewrap-0.0.6 → python_filewrap-0.0.7}/filewrap/__init__.py

@@ -2,11 +2,12 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 0, 6)
+__version__ = (0, 0, 7)
 __all__ = [
     "SupportsRead", "SupportsWrite", 
     "bio_chunk_iter", "bio_chunk_async_iter", 
     "bio_skip_iter", "bio_skip_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", 
     "bytes_ensure_part_iter", "bytes_ensure_part_async_iter", 
@@ -16,6 +17,7 @@ from asyncio import to_thread, Lock as AsyncLock
 from collections.abc import Awaitable, AsyncIterable, Iterable
 from functools import update_wrapper
 from inspect import isawaitable, iscoroutinefunction
+from itertools import chain
 from collections.abc import AsyncIterator, Callable, Iterator
 from shutil import COPY_BUFSIZE # type: ignore
 from threading import Lock
@@ -26,7 +28,7 @@ try:
 except ImportError:
     Buffer = Any
 
-from asynctools import ensure_async, ensure_aiter
+from asynctools import async_chain, ensure_async, ensure_aiter
 
 
 _T_co = TypeVar("_T_co", covariant=True)
@@ -242,6 +244,57 @@ async def bio_skip_async_iter(
         yield length
 
 
+def bytes_iter_skip(
+    it: Iterable[Buffer], 
+    /, 
+    size: int = -1, 
+    callback: None | Callable[[int], Any] = None, 
+) -> Iterator[memoryview | Buffer]:
+    it = iter(it)
+    if size == 0:
+        return it
+    if not callable(callback):
+        callback = None
+    for m in map(memoryview, it):
+        l = len(m)
+        if callback:
+            callback(min(l, size))
+        if l == size:
+            return it
+        elif l > size:
+            return chain((m[size:],), it)
+        else:
+            size -= l
+    return iter(())
+
+
+async def bytes_async_iter_skip(
+    it: Iterable[Buffer] | AsyncIterator[Buffer], 
+    /, 
+    size: int = -1, 
+    callback: None | Callable[[int], Any] = None, 
+) -> AsyncIterator[memoryview | Buffer]:
+    it = aiter(ensure_aiter(it))
+    if size == 0:
+        return it
+    callback = ensure_async(callback) if callable(callback) else None
+    async for b in it:
+        m = memoryview(b)
+        l = len(m)
+        if callback:
+            await callback(min(l, size))
+        if l == size:
+            return it
+        elif l > size:
+            return async_chain((m[size:],), it)
+        else:
+            size -= l
+    async def make_iter():
+        if False:
+            yield
+    return make_iter()
+
+
 def bytes_iter_to_reader(
     it: Iterable[Buffer], 
     /, 

{python_filewrap-0.0.6 → python_filewrap-0.0.7}/pyproject.toml

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