kleinkram 0.21.8.dev20240920104547__py3-none-any.whl → 0.21.9.dev20240920112301__py3-none-any.whl

kleinkram/helper.py

@@ -2,8 +2,10 @@ import glob
 import os
 import queue
 import threading
+from datetime import datetime
 from functools import partial
 
+import typer
 from botocore.config import Config
 from typing_extensions import Dict
 import boto3
@@ -117,7 +119,7 @@ def uploadFiles(files: Dict[str, str], credentials: Dict[str, str], nrThreads: i
     else:
         minio_endpoint = api_endpoint.replace("api", "minio")
 
-    config = Config(retries = {
+    config = Config(retries={
         'max_attempts': 10,
         'mode': 'standard'
     })
@@ -168,6 +170,7 @@ def uploadFile(_queue: queue.Queue, s3: BaseClient, transferCallback: TransferCa
             print(f"Error uploading {filename}: {e}")
             _queue.task_done()
 
+
 def canUploadMission(client: AuthenticatedClient, project_uuid: str):
     permissions = client.get("/user/permissions")
     permissions.raise_for_status()
@@ -177,6 +180,35 @@ def canUploadMission(client: AuthenticatedClient, project_uuid: str):
     return max_for_project >= 10
 
 
+def promptForTags(setTags: Dict[str, str], requiredTags: Dict[str, str]):
+    for required_tag in requiredTags:
+        if required_tag["name"] not in setTags:
+            while True:
+                if required_tag["datatype"] in ["LOCATION", "STRING", "LINK"]:
+                    tag_value = typer.prompt("Provide value for required tag " + required_tag["name"])
+                    if tag_value != "":
+                        break
+                elif required_tag["datatype"] == "BOOLEAN":
+                    tag_value = typer.confirm("Provide (y/N) for required tag " + required_tag["name"])
+                    break
+                elif required_tag["datatype"] == "NUMBER":
+                    tag_value = typer.prompt("Provide number for required tag " + required_tag["name"])
+                    try:
+                        tag_value = float(tag_value)
+                        break
+                    except ValueError:
+                        typer.echo("Invalid number format. Please provide a number.")
+                elif required_tag["datatype"] == "DATE":
+                    tag_value = typer.prompt("Provide date for required tag " + required_tag["name"])
+                    try:
+                        tag_value = datetime.strptime(tag_value, "%Y-%m-%d %H:%M:%S")
+                        break
+                    except ValueError:
+                        print("Invalid date format. Please use 'YYYY-MM-DD HH:MM:SS'")
+
+            setTags[required_tag["uuid"]] = tag_value
+
+
 if __name__ == "__main__":
     res = expand_and_match(
         "~/Downloads/dodo_mission_2024_02_08-20240408T074313Z-003/**.bag"

kleinkram/main.py

@@ -22,7 +22,7 @@ from kleinkram.queue.queue import queue
 from kleinkram.tag.tag import tag
 from kleinkram.topic.topic import topic
 from kleinkram.user.user import user
-from .helper import uploadFiles, expand_and_match, canUploadMission
+from .helper import uploadFiles, expand_and_match, canUploadMission, promptForTags
 
 
 class CommandPanel(str, Enum):
@@ -168,32 +168,7 @@ def upload(
             tags = []
         tags_dict = {item.split(":")[0]: item.split(":")[1] for item in tags}
 
-        for required_tag in project_json["requiredTags"]:
-            if required_tag["name"] not in tags_dict:
-                while True:
-                    if required_tag["datatype"] in ["LOCATION", "STRING", "LINK"]:
-                        tag_value = typer.prompt("Provide value for required tag " + required_tag["name"])
-                        if tag_value != "":
-                            break
-                    elif required_tag["datatype"] == "BOOLEAN":
-                        tag_value = typer.confirm("Provide (y/N) for required tag " + required_tag["name"])
-                        break
-                    elif required_tag["datatype"] == "NUMBER":
-                        tag_value = typer.prompt("Provide number for required tag " + required_tag["name"])
-                        try:
-                            tag_value = float(tag_value)
-                            break
-                        except ValueError:
-                            typer.echo("Invalid number format. Please provide a number.")
-                    elif required_tag["datatype"] == "DATE":
-                        tag_value = typer.prompt("Provide date for required tag " + required_tag["name"])
-                        try:
-                            tag_value = datetime.strptime(tag_value, "%Y-%m-%d %H:%M:%S")
-                            break
-                        except ValueError:
-                            print("Invalid date format. Please use 'YYYY-MM-DD HH:MM:SS'")
-
-                tags_dict[required_tag["uuid"]] = tag_value
+        promptForTags(tags_dict, project_json["requiredTags"])
 
         get_mission_url = "/mission/byName"
         mission_response = client.get(get_mission_url, params={"name": mission})

{kleinkram-0.21.8.dev20240920104547.dist-info → kleinkram-0.21.9.dev20240920112301.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kleinkram
-Version: 0.21.8.dev20240920104547
+Version: 0.21.9.dev20240920112301
 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.21.8.dev20240920104547.dist-info → kleinkram-0.21.9.dev20240920112301.dist-info}/RECORD

@@ -2,8 +2,8 @@ kleinkram/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 kleinkram/api_client.py,sha256=1GPsM-XFbPYEKP7RfWmzMTwxRqnVh4wtHVuW25KT8kA,2264
 kleinkram/consts.py,sha256=pm_6OuQcO-tYcRhwauTtyRRsuYY0y0yb6EGuIl49LnI,50
 kleinkram/error_handling.py,sha256=Nm3tUtc4blQylJLNChSLgkDLFDHg3Xza0CghRrd_SYE,4015
-kleinkram/helper.py,sha256=t-6Il0mLAuEei8m37alJ8kkI5UARisRcXjcVSD42V6E,6126
-kleinkram/main.py,sha256=C27P6hUcY12W3G8avYQBTftICP4stDjZTkhB3Jft9FE,10734
+kleinkram/helper.py,sha256=1DoQpJsYF1tWjUIwBPLkorxtFiaewOr2ipcnFWPM1nA,7673
+kleinkram/main.py,sha256=tArawTfNU88vL7ClW4xs8bXDjBviTZ7Gr9hRKzVuWuk,9263
 kleinkram/auth/auth.py,sha256=w3-TsxWxURzLQ3_p43zgV4Rlh4dVL_WqI6HG2aes-b4,4991
 kleinkram/endpoint/endpoint.py,sha256=neyW1lR7RNiRj8nwCl77StQrofwRCbnCY0mqLJkqW84,1419
 kleinkram/file/file.py,sha256=8CCYu9MdG24Rh0aEcNB3OEMUzpLMSoV4VIhdeq2n8iA,3206
@@ -13,8 +13,8 @@ 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.21.8.dev20240920104547.dist-info/METADATA,sha256=QPwfxdoZjBkmrQv4laK-VH1K-eTMMNZIHABqSBXVRtg,845
-kleinkram-0.21.8.dev20240920104547.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-kleinkram-0.21.8.dev20240920104547.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
-kleinkram-0.21.8.dev20240920104547.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-kleinkram-0.21.8.dev20240920104547.dist-info/RECORD,,
+kleinkram-0.21.9.dev20240920112301.dist-info/METADATA,sha256=k8Nutmz8CqMktCt62m7IwBfioE8j4cnRlWqA9EXXt2M,845
+kleinkram-0.21.9.dev20240920112301.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+kleinkram-0.21.9.dev20240920112301.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
+kleinkram-0.21.9.dev20240920112301.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
+kleinkram-0.21.9.dev20240920112301.dist-info/RECORD,,