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

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

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

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

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

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

@@ -6,6 +6,7 @@ import argparse
 import os
 import shutil
 import time
+from concurrent.futures import ThreadPoolExecutor
 from dataclasses import dataclass
 from pathlib import Path
 
@@ -14,6 +15,8 @@ from dotenv import load_dotenv
 
 from rclone_api import Config, Rclone, SizeSuffix
 
+os.environ["RCLONE_API_VERBOSE"] = "1"
+
 
 @dataclass
 class Args:
@@ -96,25 +99,12 @@ pass = {SRC_SFTP_PASS}
     return Config(config_text), creds
 
 
-def _init() -> None:
-    """Check if all required environment variables are set before running tests."""
-    required_vars = [
-        "BUCKET_NAME",
-        "BUCKET_KEY_SECRET",
-        "BUCKET_KEY_PUBLIC",
-        "BUCKET_URL",
-    ]
-    missing = [var for var in required_vars if not os.getenv(var)]
-    if missing:
-        print(f"Missing required environment variables: {', '.join(missing)}")
-    os.environ["RCLONE_API_VERBOSE"] = "1"
-
-
 def _run_profile(
     rclone: Rclone,
     src_file: str,
     transfers: int,
-    size: int,
+    offset: SizeSuffix,
+    size: SizeSuffix,
     log_dir: Path,
     direct_io: bool = True,
 ) -> None:
@@ -128,8 +118,8 @@ def _run_profile(
     start = time.time()
     bytes_or_err: bytes | Exception = rclone.copy_bytes(
         src=src_file,
-        offset=0,
-        length=size,
+        offset=offset.as_int(),
+        length=size.as_int(),
         direct_io=direct_io,
         transfers=transfers,
         mount_log=mount_log,
@@ -142,14 +132,14 @@ def _run_profile(
         assert False, f"Error: {bytes_or_err}\nStack trace:\n{stack_trace}"
     assert isinstance(bytes_or_err, bytes)
     # self.assertEqual(len(bytes_or_err), size)
-    assert len(bytes_or_err) == size, f"Length: {len(bytes_or_err)} != {size}"
+    assert len(bytes_or_err) == size.as_int(), f"Length: {len(bytes_or_err)} != {size}"
 
     # print io stats
     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 * 100
+    efficiency_100 = efficiency.as_int() * 100
     efficiency_str = f"{efficiency_100:.2f}"
 
     bytes_send_suffix = SizeSuffix(bytes_sent)
@@ -165,14 +155,7 @@ def _run_profile(
     print(f"Time: {diff:.1f} seconds")
 
 
-def test_profile_copy_bytes(args: Args) -> None:
-    print("Running test_profile_copy_bytes")
-    config, creds = _generate_rclone_config()
-    print("Config:")
-    print(config)
-    print("Credentials:")
-    print(creds)
-    rclone = Rclone(config)
+def test_profile_copy_bytes(args: Args, rclone: Rclone, offset: SizeSuffix) -> None:
 
     sizes = [
         1024 * 1024 * 1,
@@ -202,7 +185,8 @@ def test_profile_copy_bytes(args: Args) -> None:
                 rclone=rclone,
                 src_file=src_file,
                 transfers=transfers,
-                size=size,
+                offset=offset,
+                size=SizeSuffix(size),
                 direct_io=args.direct_io,
                 log_dir=mount_root_path,
             )
@@ -218,9 +202,24 @@ def _parse_args() -> Args:
 
 def main() -> None:
     """Main entry point."""
-    _init()
+    print("Running test_profile_copy_bytes")
+    config, creds = _generate_rclone_config()
+    print("Config:")
+    print(config)
+    print("Credentials:")
+    print(creds)
+    rclone = Rclone(config)
+
     args = _parse_args()
-    test_profile_copy_bytes(args)
+
+    with ThreadPoolExecutor() as executor:
+
+        def task(offset: SizeSuffix, args=args, rclone=rclone):
+            return test_profile_copy_bytes(args, rclone, offset)
+
+        executor.submit(task, SizeSuffix(0))
+        executor.submit(task, SizeSuffix("1G"))
+        executor.submit(task, SizeSuffix("2G"))
 
 
 if __name__ == "__main__":

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

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