python-filewrap 0.2__tar.gz → 0.2.2__tar.gz

{python_filewrap-0.2 → python_filewrap-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: python-filewrap
-Version: 0.2
+Version: 0.2.2
 Summary: Python file wrappers.
 Home-page: https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-filewrap
 License: MIT
@@ -21,7 +21,7 @@ Classifier: Programming Language :: Python :: 3 :: Only
 Classifier: Topic :: Software Development
 Classifier: Topic :: Software Development :: Libraries
 Classifier: Topic :: Software Development :: Libraries :: Python Modules
-Requires-Dist: python-asynctools (>=0.0.4)
+Requires-Dist: python-asynctools (>=0.0.5)
 Requires-Dist: python-property (>=0.0.2)
 Project-URL: Repository, https://github.com/ChenyangGao/web-mount-packs/tree/main/python-module/python-filewrap
 Description-Content-Type: text/markdown

{python_filewrap-0.2 → python_filewrap-0.2.2}/filewrap/__init__.py

@@ -2,7 +2,7 @@
 # encoding: utf-8
 
 __author__ = "ChenyangGao <https://chenyanggao.github.io>"
-__version__ = (0, 2)
+__version__ = (0, 2, 2)
 __all__ = [
     "Buffer", "SupportsRead", "SupportsReadinto", "SupportsWrite", "SupportsSeek", 
     "AsyncBufferedReader", "AsyncTextIOWrapper", 
@@ -14,9 +14,10 @@ __all__ = [
     "bytes_to_chunk_iter", "bytes_to_chunk_async_iter", 
     "bytes_ensure_part_iter", "bytes_ensure_part_async_iter", 
     "progress_bytes_iter", "progress_bytes_async_iter", 
+    "copyfileobj", "copyfileobj_async", 
 ]
 
-from asyncio import run as run_async, to_thread, Lock as AsyncLock
+from asyncio import to_thread, Lock as AsyncLock
 from collections.abc import Awaitable, AsyncIterable, AsyncIterator, Callable, Iterable, Iterator
 from functools import update_wrapper
 from io import BufferedIOBase, BufferedReader, BytesIO, RawIOBase, TextIOWrapper
@@ -45,7 +46,7 @@ except ImportError:
     Buffer.register(_SimpleCData)
     Buffer.register(array)
 
-from asynctools import async_chain, ensure_async, ensure_aiter, ensure_coroutine
+from asynctools import async_chain, ensure_async, ensure_aiter, run_async
 from property import staticproperty
 
 
@@ -164,7 +165,7 @@ class AsyncBufferedReader(BufferedReader):
         except (AttributeError, TypeError):
             ret = getattr(raw, "close")()
         if isawaitable(ret):
-            run_async(ensure_coroutine(ret))
+            run_async(ret)
 
     async def flush(self, /):
         return await ensure_async(self.raw.flush, threaded=True)()
@@ -588,7 +589,7 @@ class AsyncTextIOWrapper(TextIOWrapper):
         except (AttributeError, TypeError):
             ret = getattr(buffer, "close")()
         if isawaitable(ret):
-            run_async(ensure_coroutine(ret))
+            run_async(ret)
 
     async def flush(self, /):
         return await ensure_async(self.buffer.flush, threaded=True)()
@@ -1476,14 +1477,17 @@ def bytes_iter_to_async_reader(
         except:
             pass
     def close():
+        nonlocal at_end
         try:
             method = getattr(it, "aclose")
         except AttributeError:
             method = getattr(it, "close")
         ret = method()
         if isawaitable(ret):
-            run_async(ensure_coroutine(ret))
+            run_async(ret)
+        at_end = True
     async def aclose():
+        nonlocal at_end
         try:
             method = getattr(it, "aclose")
         except AttributeError:
@@ -1491,6 +1495,7 @@ def bytes_iter_to_async_reader(
         ret = method()
         if isawaitable(ret):
             await ret
+        at_end = True
     def peek(n: int = 0, /) -> bytearray:
         if n <= 0:
             return unconsumed[:]
@@ -1789,3 +1794,57 @@ async def progress_bytes_async_iter(
         if callable(close_progress):
             await ensure_async(close_progress)()
 
+
+def copyfileobj(
+    fsrc, 
+    fdst: SupportsWrite[Buffer], 
+    /, 
+    chunksize: int = COPY_BUFSIZE, 
+):
+    if chunksize <= 0:
+        chunksize = COPY_BUFSIZE
+    fdst_write = fdst.write
+    fsrc_read = getattr(fsrc, "read", None)
+    fsrc_readinto = getattr(fsrc, "readinto", None)
+    if callable(fsrc_readinto):
+        buf = bytearray(chunksize)
+        view = memoryview(buf)
+        while size := fsrc_readinto(buf):
+            fdst_write(view[:size])
+    elif callable(fsrc_read):
+        while chunk := fsrc_read(chunksize):
+            fdst_write(chunk)
+    else:
+        for chunk in fsrc:
+            if chunk:
+                fdst_write(chunk)
+
+
+async def copyfileobj_async(
+    fsrc, 
+    fdst: SupportsWrite[Buffer], 
+    /, 
+    chunksize: int = COPY_BUFSIZE, 
+    threaded: bool = True, 
+):
+    if chunksize <= 0:
+        chunksize = COPY_BUFSIZE
+    fdst_write = ensure_async(fdst.write, threaded=threaded)
+    fsrc_read = getattr(fsrc, "read", None)
+    fsrc_readinto = getattr(fsrc, "readinto", None)
+    if callable(fsrc_readinto):
+        fsrc_readinto = ensure_async(fsrc_readinto, threaded=threaded)
+        buf = bytearray(chunksize)
+        view = memoryview(buf)
+        while size := await fsrc_readinto(buf):
+            await fdst_write(view[:size])
+    elif callable(fsrc_read):
+        fsrc_read = ensure_async(fsrc_read, threaded=threaded)
+        while chunk := await fsrc_read(chunksize):
+            await fdst_write(chunk)
+    else:
+        chunkiter = ensure_aiter(fsrc, threaded=threaded)
+        async for chunk in chunkiter:
+            if chunk:
+                await fdst_write(chunk)
+

{python_filewrap-0.2 → python_filewrap-0.2.2}/pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "python-filewrap"
-version = "0.2"
+version = "0.2.2"
 description = "Python file wrappers."
 authors = ["ChenyangGao <wosiwujm@gmail.com>"]
 license = "MIT"
@@ -27,7 +27,7 @@ include = [
 
 [tool.poetry.dependencies]
 python = "^3.10"
-python-asynctools = ">=0.0.4"
+python-asynctools = ">=0.0.5"
 python-property = ">=0.0.2"
 
 [build-system]