python-filewrap 0.2__py3-none-any.whl → 0.2.2__py3-none-any.whl

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.dist-info → python_filewrap-0.2.2.dist-info}/METADATA

@@ -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.2.dist-info/RECORD

@@ -0,0 +1,7 @@
+LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+filewrap/__init__.py,sha256=kQ3EL_pPx5jkImO5tPuhAx6PiQnocXeHwVmBz3_o60k,62700
+filewrap/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+python_filewrap-0.2.2.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
+python_filewrap-0.2.2.dist-info/METADATA,sha256=hOJOU4Q01T3SkJQ0VQ7qAVOZSUcvwlEe3q2uHeo_xZ4,1413
+python_filewrap-0.2.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+python_filewrap-0.2.2.dist-info/RECORD,,

python_filewrap-0.2.dist-info/RECORD

@@ -1,7 +0,0 @@
-LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-filewrap/__init__.py,sha256=WKOOUCXO47ZmDRyROPcLkKIFiwsb5eDBhvbItCJ8zm4,61007
-filewrap/py.typed,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-python_filewrap-0.2.dist-info/LICENSE,sha256=o5242_N2TgDsWwFhPn7yr8YJNF7XsJM5NxUMtcT97bc,1100
-python_filewrap-0.2.dist-info/METADATA,sha256=mSMywqyvrqatQcKls-K7-2HQP02yE9ux2jMk6Gxuvu4,1411
-python_filewrap-0.2.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-python_filewrap-0.2.dist-info/RECORD,,