kleinkram 0.30.2.dev20241010132702__py3-none-any.whl → 0.30.4.dev20241010181849__py3-none-any.whl

kleinkram/mission/mission.py

@@ -156,8 +156,8 @@ def mission_by_uuid(
 
 @missionCommands.command("download")
 def download(
-    mission_uuid: Annotated[str, typer.Argument()],
-    local_path: Annotated[str, typer.Argument()],
+    mission_uuid: Annotated[List[str], typer.Option(help="UUIDs of Mission to download")],
+    local_path: Annotated[str, typer.Option()],
 ):
     """
 
@@ -172,34 +172,40 @@ def download(
         raise ValueError(f"Local path '{local_path}' is not empty, but must be empty.")
 
     client = AuthenticatedClient()
-    response = client.get("/mission/download", params={"uuid": mission_uuid})
+    for single_mission_uuid in mission_uuid:
+        response = client.get("/mission/download", params={"uuid": single_mission_uuid})
+        try:
+            response.raise_for_status()
+        except httpx.HTTPError as e:
+            raise AccessDeniedException(
+                f"Failed to download file."
+                f"Consider using the following command to list all missions: 'klein mission list --verbose'\n",
+                f"{response.json()['message']} ({response.status_code})",
+            )
 
-    try:
-        response.raise_for_status()
-    except httpx.HTTPError as e:
-        raise AccessDeniedException(
-            f"Failed to download file."
-            f"Consider using the following command to list all missions: 'klein mission list --verbose'\n",
-            f"{response.json()['message']} ({response.status_code})",
-        )
+        paths = response.json()
+        if(len(paths) == 0):
+            continue
+        mission_name = paths[0].split("/")[5] # Trust me bro
 
-    paths = response.json()
+        local_mission_path = os.path.join(local_path, mission_name)
+        os.mkdir(local_mission_path)
 
-    print(f"Downloading files to {local_path}:")
-    for path in paths:
+        print(f"Downloading files to {local_mission_path}:")
+        for path in paths:
 
-        filename = path.split("/")[-1].split("?")[0]
-        print(f" - {filename}")
+            filename = path.split("/")[-1].split("?")[0]
+            print(f" - {filename}")
 
-        response = requests.get(path, stream=True)  # Enable streaming mode
-        chunk_size = 1024 * 100  # 100 KB chunks, adjust size if needed
+            response = requests.get(path, stream=True)  # Enable streaming mode
+            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:
-            for chunk in response.iter_content(chunk_size=chunk_size):
-                if chunk:  # Filter out keep-alive new chunks
-                    f.write(chunk)
-            print(f"   Downloaded {filename}")
+            # Open the file for writing in binary mode
+            with open(os.path.join(local_mission_path, filename), "wb") as f:
+                for chunk in response.iter_content(chunk_size=chunk_size):
+                    if chunk:  # Filter out keep-alive new chunks
+                        f.write(chunk)
+                print(f"   Downloaded {filename}")
 
 
 @missionCommands.command("upload")

{kleinkram-0.30.2.dev20241010132702.dist-info → kleinkram-0.30.4.dev20241010181849.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kleinkram
-Version: 0.30.2.dev20241010132702
+Version: 0.30.4.dev20241010181849
 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.30.2.dev20241010132702.dist-info → kleinkram-0.30.4.dev20241010181849.dist-info}/RECORD

@@ -7,14 +7,14 @@ kleinkram/main.py,sha256=9u_msMlPhphfaL92GNrm-cRUeSHxc_mOFPeXLDzv-ok,9642
 kleinkram/auth/auth.py,sha256=w3-TsxWxURzLQ3_p43zgV4Rlh4dVL_WqI6HG2aes-b4,4991
 kleinkram/endpoint/endpoint.py,sha256=uez5UrAnP7L5rVHUysA9tFkN3dB3dG1Ojt9g3w-UWuQ,1441
 kleinkram/file/file.py,sha256=gLCZDHHgQWq25OmeG-lwkIh4aRZaLK12xxLkbhZ_m-g,5390
-kleinkram/mission/mission.py,sha256=wI83aQbWj4P6zydVThMJiAdQHAVQWE98_-VWE5K12HQ,8684
+kleinkram/mission/mission.py,sha256=gdkR6IWgyBGf0IgnEfcRBgRlNUSKqm_J25yX1V0dY9Q,9093
 kleinkram/project/project.py,sha256=yDygz9JJ4Td5VsoCoCLm36HccRyd7jl65Hq05uxEGts,1602
 kleinkram/queue/queue.py,sha256=MaLBjAu8asi9BkPvbbT-5AobCcpy3ex5rxM1kHpRINA,181
 kleinkram/tag/tag.py,sha256=JSHbDPVfsvP34MuQhw__DPQk-Bah5G9BgwYsj_K_JGc,1805
 kleinkram/topic/topic.py,sha256=IaXhrIHcJ3FSIr0WC-N7u9fkz-lAvSBgQklTX67t0Yc,1641
 kleinkram/user/user.py,sha256=hDrbWeFPPnh2sswDd445SwcIFGyAbfXXWpYq1VqrK0g,1379
-kleinkram-0.30.2.dev20241010132702.dist-info/METADATA,sha256=HlmaCi4njfo4z508YFE89BQDHDls5gW7uOTIO1DNSfo,845
-kleinkram-0.30.2.dev20241010132702.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-kleinkram-0.30.2.dev20241010132702.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
-kleinkram-0.30.2.dev20241010132702.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-kleinkram-0.30.2.dev20241010132702.dist-info/RECORD,,
+kleinkram-0.30.4.dev20241010181849.dist-info/METADATA,sha256=0bO3d5o03qfhq1xTCwd4MahxeJhpGrAL22HEjafD7FQ,845
+kleinkram-0.30.4.dev20241010181849.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+kleinkram-0.30.4.dev20241010181849.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
+kleinkram-0.30.4.dev20241010181849.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
+kleinkram-0.30.4.dev20241010181849.dist-info/RECORD,,