rclone-api 1.0.98__tar.gz → 1.0.99__tar.gz

{rclone_api-1.0.98 → rclone_api-1.0.99}/PKG-INFO

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

{rclone_api-1.0.98 → rclone_api-1.0.99}/pyproject.toml

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

rclone_api-1.0.99/src/rclone_api/experimental/flags.py

@@ -0,0 +1,121 @@
+from dataclasses import dataclass, fields, is_dataclass
+from typing import Type, TypeVar
+
+T = TypeVar("T")
+
+
+def _merge(cls: Type[T], dataclass_a: T, dataclass_b: T) -> T:
+    if not is_dataclass(dataclass_a) or not is_dataclass(dataclass_b):
+        raise ValueError("Both inputs must be dataclass instances")
+    if type(dataclass_a) is not type(dataclass_b):
+        raise ValueError("Dataclass instances must be of the same type")
+
+    merged_kwargs = {}
+    for field in fields(dataclass_a):
+        a_value = getattr(dataclass_a, field.name)
+        b_value = getattr(dataclass_b, field.name)
+
+        if is_dataclass(a_value) and is_dataclass(b_value):
+            merged_kwargs[field.name] = _merge(type(a_value), a_value, b_value)
+        else:
+            merged_kwargs[field.name] = b_value if b_value is not None else a_value
+
+    return cls(**merged_kwargs)
+
+
+def _field_name_to_flag(field_name: str) -> str:
+    return f"--{field_name.replace('_', '-')}"
+
+
+@dataclass
+class BaseFlags:
+    def to_args(self) -> list[str]:
+        args = []
+        for field in fields(self):
+            value = getattr(self, field.name)
+            if value is None:
+                continue
+            # If the field value is a nested dataclass that supports to_args, use it.
+            if is_dataclass(value) and hasattr(value, "to_args"):
+                to_args = getattr(value, "to_args")
+                args.extend(to_args())
+            elif isinstance(value, bool):
+                # Only include the flag if the boolean is True.
+                if value:
+                    args.append(_field_name_to_flag(field.name))
+            else:
+                args.append(_field_name_to_flag(field.name))
+                if isinstance(value, list):
+                    # Join list values with a comma.
+                    args.append(",".join(map(str, value)))
+                else:
+                    args.append(str(value))
+        return args
+
+    def merge(self, other: "BaseFlags") -> "BaseFlags":
+        # Use the type of self, so merging CopyFlags returns a CopyFlags instance.
+        return _merge(type(self), self, other)
+
+    def __repr__(self):
+        return str(self.to_args())
+
+
+@dataclass(repr=False)
+class CopyFlags(BaseFlags):
+    check_first: bool | None = None
+    checksum: bool | None = False
+    compare_dest: list[str] | None = None
+    copy_dest: list[str] | None = None
+    cutoff_mode: str | None = None
+    ignore_case_sync: bool | None = None
+    ignore_checksum: bool | None = None
+    ignore_existing: bool | None = None
+    ignore_size: bool | None = None
+    ignore_times: bool | None = None
+    immutable: bool | None = None
+    inplace: bool | None = None
+    links: bool | None = None
+    max_backlog: int | None = None
+    max_duration: str | None = None
+    max_transfer: str | None = None
+    metadata: bool | None = None
+    modify_window: str | None = None
+    multi_thread_chunk_size: str | None = None
+    multi_thread_cutoff: str | None = None
+    multi_thread_streams: int | None = None
+    multi_thread_write_buffer_size: str | None = None
+    no_check_dest: bool | None = None
+    no_traverse: bool | None = None
+    no_update_dir_modtime: bool | None = None
+    no_update_modtime: bool | None = None
+    order_by: str | None = None
+    partial_suffix: str | None = None
+    refresh_times: bool | None = None
+    server_side_across_configs: bool | None = None
+    size_only: bool | None = None
+    streaming_upload_cutoff: str | None = None
+    update: bool | None = None
+
+
+@dataclass(repr=False)
+class Flags(BaseFlags):
+    copy: CopyFlags | None = None
+
+
+def unit_test() -> None:
+    copy_flags_a = CopyFlags(compare_dest=["a", "b"])
+    copy_flags_b = CopyFlags(checksum=False)
+    flags_a = copy_flags_a.merge(copy_flags_b)
+    print("A:", flags_a)
+
+    copy_flags_c = CopyFlags(checksum=True)
+    copy_flags_d = CopyFlags(checksum=False)
+
+    merged_c_d = copy_flags_c.merge(copy_flags_d)
+    print("B:", merged_c_d)
+    merged_d_c = copy_flags_d.merge(copy_flags_c)
+    print("C:", merged_d_c)
+
+
+if __name__ == "__main__":
+    unit_test()

{rclone_api-1.0.98 → rclone_api-1.0.99}/src/rclone_api/rclone.py

@@ -919,6 +919,7 @@ class Rclone:
                 if outdir.exists():
                     print(f"{outdir} mount still exists, attempting to remove")
                     if not _IS_WINDOWS:
+
                         def exec(cmd: str) -> int:
                             if verbose:
                                 print(f"Executing: {cmd}")

{rclone_api-1.0.98 → rclone_api-1.0.99}/src/rclone_api/s3/chunk_uploader.py

@@ -260,6 +260,20 @@ def _get_chunk_tmpdir() -> Path:
         return out
 
 
+def _get_file_size(file_path: Path, timeout: int = 60) -> int:
+    sleep_time = timeout / 60 if timeout > 0 else 1
+    start = time.time()
+    while True:
+        try:
+            if file_path.exists():
+                return file_path.stat().st_size
+        except FileNotFoundError:
+            pass
+        if time.time() - start > timeout:
+            raise TimeoutError(f"File {file_path} not found after {timeout} seconds")
+        time.sleep(sleep_time)
+
+
 def file_chunker(
     upload_state: UploadState, max_chunks: int | None, output: Queue[FileChunk | None]
 ) -> None:
@@ -279,7 +293,8 @@ def file_chunker(
     file_path = upload_info.src_file_path
     chunk_size = upload_info.chunk_size
     src = Path(file_path)
-    file_size = os.path.getsize(file_path)
+    # Mounted files may take a while to appear, so keep retrying.
+    file_size = _get_file_size(src, timeout=60)
     part_number = 1
     done_part_numbers: set[int] = {
         p.part_number for p in upload_state.parts if p is not None

{rclone_api-1.0.98 → rclone_api-1.0.99}/src/rclone_api.egg-info/PKG-INFO

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

rclone_api-1.0.98/src/rclone_api/experimental/flags.py

@@ -1,93 +0,0 @@
-from dataclasses import dataclass, field
-
-
-@dataclass
-class CopyFlags:
-    check_first: bool = False
-    checksum: bool = False
-    ignore_existing: bool = False
-    ignore_times: bool = False
-    immutable: bool = False
-    inplace: bool = False
-    links: bool = False
-    metadata: bool = False
-
-
-@dataclass
-class SyncFlags:
-    backup_dir: str | None = None
-    delete_after: bool = False
-    delete_before: bool = False
-    delete_during: bool = False
-    ignore_errors: bool = False
-    track_renames: bool = False
-
-
-@dataclass
-class ImportantFlags:
-    dry_run: bool = False
-    interactive: bool = False
-    verbose: int = 0  # 0 = default, 1 = -v, 2 = -vv, etc.
-
-
-@dataclass
-class NetworkingFlags:
-    bwlimit: str | None = None
-    timeout: str | None = "5m0s"
-    tpslimit: float | None = None
-    user_agent: str | None = "rclone/v1.69.1"
-
-
-@dataclass
-class PerformanceFlags:
-    buffer_size: str | None = "16MiB"
-    checkers: int = 8
-    transfers: int = 4
-
-
-@dataclass
-class ConfigFlags:
-    config: str | None = None
-    ask_password: bool = True
-    auto_confirm: bool = False
-
-
-@dataclass
-class DebuggingFlags:
-    cpuprofile: str | None = None
-    memprofile: str | None = None
-
-
-@dataclass
-class FilterFlags:
-    exclude: list[str] = field(default_factory=list)
-    include: list[str] = field(default_factory=list)
-    max_age: str | None = None
-    min_size: str | None = None
-
-
-@dataclass
-class ListingFlags:
-    fast_list: bool = False
-
-
-@dataclass
-class LoggingFlags:
-    log_file: str | None = None
-    log_level: str = "NOTICE"  # Options: DEBUG, INFO, NOTICE, ERROR
-    stats: str | None = "1m0s"
-    progress: bool = False
-
-
-@dataclass
-class Flags:
-    copy: CopyFlags | None = None
-    sync: SyncFlags | None = None
-    important: ImportantFlags | None = None
-    networking: NetworkingFlags | None = None
-    performance: PerformanceFlags | None = None
-    config: ConfigFlags | None = None
-    debugging: DebuggingFlags | None = None
-    filter: FilterFlags | None = None
-    listing: ListingFlags | None = None
-    logging: LoggingFlags | None = None