rclone-api 1.1.54__tar.gz → 1.1.56__tar.gz

{rclone_api-1.1.54 → rclone_api-1.1.56}/PKG-INFO

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

{rclone_api-1.1.54 → rclone_api-1.1.56}/pyproject.toml

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

{rclone_api-1.1.54 → rclone_api-1.1.56}/src/rclone_api/profile/mount_copy_bytes.py

@@ -6,7 +6,7 @@ import argparse
 import os
 import shutil
 import time
-from concurrent.futures import ThreadPoolExecutor
+from concurrent.futures import ThreadPoolExecutor, Future
 from dataclasses import dataclass
 from pathlib import Path
 
@@ -138,8 +138,8 @@ def _run_profile(
     bytes_sent = net_io_end.bytes_sent - net_io_start.bytes_sent
     bytes_recv = net_io_end.bytes_recv - net_io_start.bytes_recv
     packets_sent = net_io_end.packets_sent - net_io_start.packets_sent
-    efficiency = size / (bytes_recv)
-    efficiency_100 = efficiency.as_int() * 100
+    efficiency = size.as_int() / (bytes_recv)
+    efficiency_100 = efficiency * 100
     efficiency_str = f"{efficiency_100:.2f}"
 
     bytes_send_suffix = SizeSuffix(bytes_sent)
@@ -155,7 +155,7 @@ def _run_profile(
     print(f"Time: {diff:.1f} seconds")
 
 
-def test_profile_copy_bytes(args: Args, rclone: Rclone, offset: SizeSuffix) -> None:
+def test_profile_copy_bytes(args: Args, rclone: Rclone, offset: SizeSuffix, transfer_list: list[int] | None, mount_root_path: Path) -> None:
 
     sizes = [
         1024 * 1024 * 1,
@@ -169,15 +169,13 @@ def test_profile_copy_bytes(args: Args, rclone: Rclone, offset: SizeSuffix) -> N
         # 1024 * 1024 * 256,
     ]
     # transfer_list = [1, 2, 4, 8, 16]
-    transfer_list = [1, 2, 4]
+    transfer_list = transfer_list or [1, 2, 4]
 
     # src_file = "dst:rclone-api-unit-test/zachs_video/internaly_ai_alignment.mp4"
     # sftp mount
     src_file = "src:aa_misc_data/aa_misc_data/world_lending_library_2024_11.tar.zst"
 
-    mount_root_path = Path("rclone_logs") / "mount"
-    if mount_root_path.exists():
-        shutil.rmtree(mount_root_path)
+
 
     for size in sizes:
         for transfers in transfer_list:
@@ -210,16 +208,24 @@ def main() -> None:
     print(creds)
     rclone = Rclone(config)
 
-    args = _parse_args()
-
-    with ThreadPoolExecutor() as executor:
-
-        def task(offset: SizeSuffix, args=args, rclone=rclone):
-            return test_profile_copy_bytes(args, rclone, offset)
+    mount_root_path = Path("rclone_logs") / "mount"
+    if mount_root_path.exists():
+        shutil.rmtree(mount_root_path)
 
-        executor.submit(task, SizeSuffix(0))
-        executor.submit(task, SizeSuffix("1G"))
-        executor.submit(task, SizeSuffix("2G"))
+    args = _parse_args()
+    transfer_list = [1]
+    parallel_workers = 4
+
+    def task(offset: SizeSuffix, args=args, rclone=rclone, transfer_list=transfer_list, mount_root_path=mount_root_path):
+        return test_profile_copy_bytes(args, rclone, offset, transfer_list, mount_root_path)
+    
+
+    with ThreadPoolExecutor(max_workers=parallel_workers) as executor:
+        tasks: list[Future] = []
+        for i in range(parallel_workers):
+            offset = SizeSuffix(i * 1024 * 1024 * 256)
+            future = ThreadPoolExecutor().submit(lambda: task(offset=offset))
+            tasks.append(future)
 
 
 if __name__ == "__main__":

{rclone_api-1.1.54 → rclone_api-1.1.56}/src/rclone_api.egg-info/PKG-INFO

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