kleinkram 0.26.1.dev20241008115122__tar.gz → 0.28.0__tar.gz

{kleinkram-0.26.1.dev20241008115122 → kleinkram-0.28.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kleinkram
-Version: 0.26.1.dev20241008115122
+Version: 0.28.0
 Summary: A CLI for the ETH project kleinkram
 Project-URL: Homepage, https://github.com/leggedrobotics/kleinkram
 Project-URL: Issues, https://github.com/leggedrobotics/kleinkram/issues

{kleinkram-0.26.1.dev20241008115122 → kleinkram-0.28.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "hatchling.build"
 
 [project]
 name = "kleinkram"
-version = "0.26.1-dev20241008115122"
+version = "0.28.0"
 description = "A CLI for the ETH project kleinkram"
 readme = "README.md"
 requires-python = ">=3.7"

{kleinkram-0.26.1.dev20241008115122 → kleinkram-0.28.0}/src/kleinkram/file/file.py

@@ -1,5 +1,5 @@
 import os
-from typing_extensions import Optional, Annotated
+from typing_extensions import Optional, Annotated, List
 
 import httpx
 import requests
@@ -16,6 +16,61 @@ file = typer.Typer(
 )
 
 
+@file.command("download")
+def download_file(
+    file_uuid: Annotated[List[str], typer.Option(help="UUIDs of the files")],
+    local_path: Annotated[
+        str,
+        typer.Option(
+            prompt=True,
+            help="Local path to save the file",
+        ),
+    ],
+):
+    """
+    Download files by UUIDs to a local path.\n
+    Examples:\n
+    klein file download --file-uuid="9d5a9..."  --file-uuid="9833f..." --local-path="~/Downloads" \n
+    klein file download --file-uuid="9d5a9..."  --local-path="~/Downloads/example.bag"
+
+    """
+    client = AuthenticatedClient()
+    url = f"/file/download"
+
+    fixed_local_path = os.path.expanduser(local_path)
+
+    isDir = os.path.isdir(fixed_local_path)
+    chunk_size = 1024 * 100  # 100 KB chunks, adjust size if needed
+
+    for file in file_uuid:
+        response = client.get(
+            url,
+            params={"uuid": file, "expires": True},
+        )
+        if response.status_code >= 400:
+            raise AccessDeniedException(
+                f"Failed to download file: {response.json()['message']}",
+                "Status Code: " + str(response.status_code),
+            )
+        download_url = response.text
+        if isDir:
+            filename = download_url.split("/")[6].split("?")[0]  # Trust me bro
+            filepath = os.path.join(fixed_local_path, filename)
+        elif not isDir and len(file_uuid) == 1:
+            filepath = fixed_local_path
+        else:
+            raise ValueError("Multiple files can only be downloaded to a directory")
+        if os.path.exists(filepath):
+            raise FileExistsError(f"File already exists: {filepath}")
+        print(f"Downloading to: {filepath}")
+        filestream = requests.get(download_url, stream=True)
+        with open(filepath, "wb") as f:
+            for chunk in filestream.iter_content(chunk_size=chunk_size):
+                if chunk:  # Filter out keep-alive new chunks
+                    f.write(chunk)
+            print(f"Completed")
+
+
 @file.command("list")
 def list_files(
     project: Optional[str] = typer.Option(None, help="Name of Project"),

{kleinkram-0.26.1.dev20241008115122 → kleinkram-0.28.0}/src/kleinkram/mission/mission.py

@@ -192,7 +192,7 @@ def download(
         print(f" - {filename}")
 
         response = requests.get(path, stream=True)  # Enable streaming mode
-        chunk_size = 1024  # 1 KB chunks, adjust size if needed
+        chunk_size = 1024 * 100  # 100 KB chunks, adjust size if needed
 
         # Open the file for writing in binary mode
         with open(os.path.join(local_path, filename), "wb") as f: