kleinkram 0.26.1.dev20241008105540__tar.gz → 0.27.0.dev20241008141236__tar.gz

{kleinkram-0.26.1.dev20241008105540 → kleinkram-0.27.0.dev20241008141236}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kleinkram
-Version: 0.26.1.dev20241008105540
+Version: 0.27.0.dev20241008141236
 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.dev20241008105540 → kleinkram-0.27.0.dev20241008141236}/pyproject.toml

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

{kleinkram-0.26.1.dev20241008105540 → kleinkram-0.27.0.dev20241008141236}/src/kleinkram/error_handling.py

@@ -1,3 +1,4 @@
+import sys
 import typing
 
 import typer
@@ -20,7 +21,7 @@ class AccessDeniedException(Exception):
 
 
 def not_yet_implemented_handler(e: Exception):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         "This feature is not yet implemented, please check for updates.",
         title="Not Yet Implemented",
@@ -34,7 +35,7 @@ def not_yet_implemented_handler(e: Exception):
 
 
 def not_authenticated_handler(e: NotAuthenticatedException):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         f"{e.message}\n » Please run 'klein login' to authenticate.",
         title="Not Authenticated",
@@ -48,7 +49,7 @@ def not_authenticated_handler(e: NotAuthenticatedException):
 
 
 def access_denied_handler(e: AccessDeniedException):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         f"{e.message}\n » API Response: {e.api_error}",
         title="Access Denied",
@@ -62,7 +63,7 @@ def access_denied_handler(e: AccessDeniedException):
 
 
 def value_error_handler(e: Exception):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         str(e),
         title="Invalid Argument",
@@ -76,7 +77,7 @@ def value_error_handler(e: Exception):
 
 
 def http_status_error_handler(e: HTTPStatusError):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         f"An HTTP error occurred: {e}\n\n » Please report this error to the developers.",
         title="HTTP Status Error",
@@ -90,7 +91,7 @@ def http_status_error_handler(e: HTTPStatusError):
 
 
 def remote_down_handler(e: Exception):
-    console = Console()
+    console = Console(file=sys.stderr)
     panel = Panel(
         f"An error occurred while communicating with the remote server: {e}\n"
         f"\n » The server may be down or unreachable; please try again.",

{kleinkram-0.26.1.dev20241008105540 → kleinkram-0.27.0.dev20241008141236}/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
@@ -15,6 +15,54 @@ file = typer.Typer(
     context_settings={"help_option_names": ["-h", "--help"]},
 )
 
+@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(

{kleinkram-0.26.1.dev20241008105540 → kleinkram-0.27.0.dev20241008141236}/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: