rclone-api 1.3.1__tar.gz → 1.3.2__tar.gz

{rclone_api-1.3.1 → rclone_api-1.3.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rclone_api
-Version: 1.3.1
+Version: 1.3.2
 Summary: rclone api in python
 Home-page: https://github.com/zackees/rclone-api
 License: BSD 3-Clause License

{rclone_api-1.3.1 → rclone_api-1.3.2}/pyproject.toml

@@ -23,7 +23,7 @@ dependencies = [
 ]
 
 # Change this with the version number bump.
-version = "1.3.1"
+version = "1.3.2"
 
 [tool.setuptools]
 package-dir = {"" = "src"}

{rclone_api-1.3.1 → rclone_api-1.3.2}/src/rclone_api/profile/mount_copy_bytes.py

@@ -164,9 +164,17 @@ def _run_profile(
     diff = (time.time() - start) / num
     net_io_end = psutil.net_io_counters()
     # self.assertEqual(len(bytes_or_err), size)
-    assert (
-        bytes_count == SizeSuffix(size * num).as_int()
-    ), f"Length: {SizeSuffix(bytes_count)} != {SizeSuffix(size* num)}"
+    # assert (
+    #     bytes_count == SizeSuffix(size * num).as_int()
+    # ), f"Length: {SizeSuffix(bytes_count)} != {SizeSuffix(size* num)}"
+
+    if bytes_count != SizeSuffix(size * num).as_int():
+        print("\n######################## ERROR ########################")
+        print(f"Error: Length: {SizeSuffix(bytes_count)} != {SizeSuffix(size* num)}")
+        print(f"  Bytes count: {bytes_count}")
+        print(f"  Size: {SizeSuffix(size * num)}")
+        print(f"  num: {num}")
+        print("########################################################\n")
 
     # print io stats
     bytes_sent = (net_io_end.bytes_sent - net_io_start.bytes_sent) // num
@@ -203,11 +211,11 @@ def test_profile_copy_bytes(
         sizes = [size.as_int()]
     else:
         sizes = [
-            # 1024 * 1024 * 1,
+            1024 * 1024 * 1,
             # 1024 * 1024 * 2,
             # 1024 * 1024 * 4,
             # 1024 * 1024 * 8,
-            1024 * 1024 * 16,
+            # 1024 * 1024 * 16,
             # 1024 * 1024 * 32,
             1024 * 1024 * 64,
             1024 * 1024 * 128,

{rclone_api-1.3.1 → rclone_api-1.3.2}/src/rclone_api/rclone.py

@@ -682,7 +682,7 @@ class Rclone:
         save_state_json: Path,
         chunk_size: SizeSuffix | None = None,
         read_threads: int = 8,
-        write_threads: int = 16,
+        write_threads: int = 8,
         retries: int = 3,
         verbose: bool | None = None,
         max_chunks_before_suspension: int | None = None,
@@ -829,74 +829,6 @@ class Rclone:
         finally:
             chunk_fetcher.shutdown()
 
-    def _copy_bytes(
-        self,
-        src: str,
-        offset: int,
-        length: int,
-        transfers: int = 1,  # Note, increasing transfers can result in devestating drop in performance.
-        # If outfile is supplied then bytes are written to this file and success returns bytes(0)
-        outfile: Path | None = None,
-        mount_log: Path | None = None,
-        direct_io: bool = True,
-    ) -> bytes | Exception:
-        """Copy bytes from a file to another file."""
-        from rclone_api.util import random_str
-
-        tmp_mnts = Path("tmp_mnts") / random_str(12)
-        src_parent_path = Path(src).parent.as_posix()
-        src_file = Path(src).name
-        other_args: list[str] = [
-            "--no-modtime",
-            # "--vfs-read-wait", "1s"
-        ]
-        vfs_read_chunk_size = SizeSuffix(length // transfers)
-        vfs_read_chunk_size_limit = SizeSuffix(length)
-        vfs_read_chunk_streams = transfers
-        vfs_disk_space_total_size = SizeSuffix(length)
-        # --max-read-ahead SizeSuffix
-        max_read_ahead = SizeSuffix(vfs_read_chunk_size.as_int())
-
-        # other_args += ["--vfs-read-chunk-size", str(vfs_read_chunk_size)]
-        other_args += ["--vfs-read-chunk-size", str(vfs_read_chunk_size)]
-        other_args += ["--vfs-read-chunk-size-limit", str(vfs_read_chunk_size_limit)]
-        other_args += ["--vfs-read-chunk-streams", str(vfs_read_chunk_streams)]
-        other_args += ["--vfs-disk-space-total-size", str(vfs_disk_space_total_size)]
-        other_args += ["--max-read-ahead", str(max_read_ahead)]
-        other_args += ["--read-only"]
-        if direct_io:
-            other_args += ["--direct-io"]
-
-        with TemporaryDirectory() as tmpdir:
-            cache_dir = Path(tmpdir) / "cache"
-            other_args += ["--cache-dir", str(cache_dir.absolute())]
-            try:
-                # use scoped mount to do the read, then write the bytes to the destination
-                with self.scoped_mount(
-                    src_parent_path,
-                    tmp_mnts,
-                    use_links=True,
-                    verbose=mount_log is not None,
-                    vfs_cache_mode="minimal",
-                    other_args=other_args,
-                    log=mount_log,
-                    cache_dir=cache_dir,
-                    cache_dir_delete_on_exit=True,
-                ):
-                    src_file_mnt = tmp_mnts / src_file
-                    with open(src_file_mnt, "rb") as f:
-                        f.seek(offset)
-                        data = f.read(length)
-                        if outfile is None:
-                            return data
-                        with open(outfile, "wb") as out:
-                            out.write(data)
-                            del data
-                        return bytes(0)
-
-            except Exception as e:
-                return e
-
     def get_multi_mount_file_chunker(
         self,
         src: str,
@@ -928,16 +860,19 @@ class Rclone:
         if direct_io:
             other_args += ["--direct-io"]
 
+        base_mount_dir = Path("tmp_mnts")
+        base_cache_dir = Path("cache")
+
         filename = Path(src).name
         with ThreadPoolExecutor(max_workers=threads) as executor:
             futures: list[Future] = []
             try:
                 for i in range(threads):
-                    tmp_mnts = Path("tmp_mnts") / random_str(12)
+                    tmp_mnts = base_mount_dir / random_str(12)
                     verbose = mount_log is not None
 
                     src_parent_path = Path(src).parent.as_posix()
-                    cache_dir = Path("cache") / random_str(12)
+                    cache_dir = base_cache_dir / random_str(12)
 
                     def task(
                         src_parent_path=src_parent_path,
@@ -992,7 +927,7 @@ class Rclone:
         )
         return filechunker
 
-    def copy_bytes(
+    def copy_bytes_multimount(
         self,
         src: str,
         offset: int,
@@ -1004,7 +939,7 @@ class Rclone:
         mount_log: Path | None = None,
         direct_io: bool = True,
     ) -> bytes | Exception:
-        """Copy bytes from a file to another file."""
+        """Copy a slice of bytes from the src file to dst. Parallelism is achieved through multiple mounted files."""
         from rclone_api.types import FilePart
 
         # determine number of threads from chunk size
@@ -1022,12 +957,13 @@ class Rclone:
             payload = fp.payload
             if isinstance(payload, Exception):
                 return payload
-            if outfile is None:
-                out = payload.read_bytes()
-                payload.unlink()
-                return out
-            shutil.move(payload, outfile)
-            return bytes(0)
+            try:
+                if outfile is None:
+                    return payload.read_bytes()
+                shutil.move(payload, outfile)
+                return bytes(0)
+            finally:
+                fp.close()
 
         except Exception as e:
             warnings.warn(f"Error copying bytes: {e}")

{rclone_api-1.3.1 → rclone_api-1.3.2}/src/rclone_api/s3/chunk_task.py

@@ -132,11 +132,6 @@ def file_chunker(
     upload_info = upload_state.upload_info
     file_path = upload_info.src_file_path
     chunk_size = upload_info.chunk_size
-    # src = Path(file_path)
-
-    # for p in upload_state.parts:
-    #     if not isinstance(p, EndOfStream):
-    #         part_tracker.add_done_part_number(p.part_number)
 
     done_part_numbers: set[int] = {
         p.part_number for p in upload_state.parts if not isinstance(p, EndOfStream)
@@ -173,9 +168,6 @@ def file_chunker(
 
             assert offset < file_size, f"Offset {offset} is greater than file size"
             fetch_size = max(0, min(chunk_size, file_size - offset))
-
-            # assert fetch_size > 0, f"Invalid fetch size: {fetch_size}"
-
             if fetch_size == 0:
                 logger.error(
                     f"Empty data for part {curr_part_number} of {file_path}, is this the last chunk?"
@@ -186,17 +178,7 @@ def file_chunker(
                         f"This should have been the last part, but it is not: {final_part_number} != {curr_part_number}"
                     )
 
-            # Open the file, seek, read the chunk, and close immediately.
-            # with open(file_path, "rb") as f:
-            #     f.seek(offset)
-            #     data = f.read(chunk_size)
-
-            # data = chunk_fetcher(offset, chunk_size).result()
-
             assert curr_part_number is not None
-            # cpn: int = curr_part_number
-
-            # offset = (curr_part_number - 1) * chunk_size
             logger.info(
                 f"Reading chunk {curr_part_number} of {num_parts} for {file_path}"
             )
@@ -208,7 +190,6 @@ def file_chunker(
             while queue_upload.full():
                 time.sleep(0.1)
     except Exception as e:
-
         logger.error(f"Error reading file: {e}", exc_info=True)
     finally:
         logger.info(f"Finishing FILE CHUNKER for {file_path} and adding EndOfStream")

{rclone_api-1.3.1 → rclone_api-1.3.2}/src/rclone_api.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: rclone_api
-Version: 1.3.1
+Version: 1.3.2
 Summary: rclone api in python
 Home-page: https://github.com/zackees/rclone-api
 License: BSD 3-Clause License

{rclone_api-1.3.1 → rclone_api-1.3.2}/tests/test_copy_bytes.py

@@ -72,7 +72,7 @@ class RcloneCopyBytesTester(unittest.TestCase):
 
     def test_copy_bytes(self) -> None:
         rclone = Rclone(_generate_rclone_config())
-        bytes_or_err: bytes | Exception = rclone.copy_bytes(
+        bytes_or_err: bytes | Exception = rclone.copy_bytes_multimount(
             src="dst:rclone-api-unit-test/zachs_video/breaking_ai_mind.mp4",
             offset=0,
             length=1024 * 1024,
@@ -92,7 +92,7 @@ class RcloneCopyBytesTester(unittest.TestCase):
             tmp = Path(tmpdir) / "tmp.mp4"
             log = Path(tmpdir) / "log.txt"
             rclone = Rclone(_generate_rclone_config())
-            bytes_or_err: bytes | Exception = rclone.copy_bytes(
+            bytes_or_err: bytes | Exception = rclone.copy_bytes_multimount(
                 src="dst:rclone-api-unit-test/zachs_video/breaking_ai_mind.mp4",
                 offset=0,
                 length=1024 * 1024,