kleinkram 0.0.116__py3-none-any.whl → 0.0.118__py3-none-any.whl

kleinkram/auth.py

@@ -3,6 +3,8 @@ import os
 import urllib.parse
 import webbrowser
 from http.server import BaseHTTPRequestHandler, HTTPServer
+from typing import Optional
+
 from typing_extensions import Annotated
 
 from pathlib import Path
@@ -72,6 +74,7 @@ class OAuthCallbackHandler(BaseHTTPRequestHandler):
             self.end_headers()
             self.wfile.write(b"Authentication successful. You can close this window.")
             return
+        print("here")
 
     def log_message(self, format, *args):
         pass
@@ -90,8 +93,8 @@ class AuthenticatedClient(httpx.Client):
         try:
             self.tokenfile = TokenFile()
             self._load_cookies()
-        except:
-            print("Not authenticated. Please run 'GTD login'.")
+        except Exception as e:
+            print(f"{self.tokenfile.endpoint} is not authenticated. Please run 'klein login'.")
 
     def _load_cookies(self):
         if self.tokenfile.isCliToken():
@@ -107,7 +110,7 @@ class AuthenticatedClient(httpx.Client):
         if not refresh_token:
             print("No refresh token found. Please login again.")
             raise Exception("No refresh token found.")
-        self.cookies.set(REFRESH_TOKEN, refresh_token,)
+        self.cookies.set(REFRESH_TOKEN, refresh_token, )
         response = self.post(
             "/auth/refresh-token",
         )
@@ -122,7 +125,7 @@ class AuthenticatedClient(httpx.Client):
             method, self.tokenfile.endpoint + url, *args, **kwargs
         )
         if (
-            url == "/auth/refresh-token"
+                url == "/auth/refresh-token"
         ) and response.status_code == 401:
             print("Refresh token expired. Please login again.")
             response.status_code = 403
@@ -137,32 +140,95 @@ class AuthenticatedClient(httpx.Client):
 client = AuthenticatedClient()
 
 
-def login(key: Annotated[str, typer.Option()] = None):
+def login(
+        key: Optional[str] = typer.Option(None, help="CLI Key", hidden=True),
+        open_browser: Optional[bool] = typer.Option(True, help="Open browser for authentication"),
+):
+    """
+    Login into the currently set endpoint.\n
+    By default, it will open the browser for authentication. On machines without a browser, you can manually open the URL provided and paste the tokens back.
+    """
     tokenfile = TokenFile()
     if key:
         tokenfile.saveTokens(key)
     else:
-        print("Opening browser for authentication...")
-        webbrowser.open(tokenfile.endpoint + "/auth/google?state=cli")
+        url = tokenfile.endpoint + "/auth/google?state=cli"
 
-        print("Waiting for authentication to complete...")
-        auth_tokens = get_auth_tokens()
+        has_browser = True
+        try:
+            browser_available = webbrowser.get()
+            if not browser_available:
+                raise Exception("No web browser available.")
+        except Exception as e:
+            has_browser = False
 
-        if not auth_tokens:
-            print("Failed to get authentication tokens.")
-            return
+        if has_browser and open_browser:
+            webbrowser.open(url)
+            auth_tokens = get_auth_tokens()
 
-        tokenfile.saveTokens(auth_tokens)
+            if not auth_tokens:
+                print("Failed to get authentication tokens.")
+                return
 
-        print("Authentication complete. Tokens saved to tokens.json.")
+            tokenfile.saveTokens(auth_tokens)
+            print("Authentication complete. Tokens saved to ~/.kleinkram.json.")
 
+            return
 
-def endpoint(endpoint: Annotated[str, typer.Argument()]):
+        print(f"Please open the following URL manually in your browser to authenticate: {url + '-no-redirect'}")
+        print("Enter the authentication token provided after logging in:")
+        manual_auth_token = input("Authentication Token: ")
+        manual_refresh_token = input("Refresh Token: ")
+        if manual_auth_token:
+            tokenfile.saveTokens({AUTH_TOKEN: manual_auth_token, REFRESH_TOKEN: manual_refresh_token})
+            print("Authentication complete. Tokens saved to tokens.json.")
+        else:
+            print("No authentication token provided.")
+        return
+
+
+def setEndpoint(
+        endpoint: Optional[str] = typer.Argument(None, help="API endpoint to use")
+):
+    """
+    Set the current endpoint
+
+    Use this command to switch between different API endpoints.\n
+    Standard endpoints are:\n
+    - http://localhost:3000\n
+    - https://api.datasets.leggedrobotics.com\n
+    - https://api.datasets.dev.leggedrobotics.com
+    """
     tokenfile = TokenFile()
     tokenfile.endpoint = endpoint
     tokenfile.writeToFile()
+    print("Endpoint set to: " + endpoint)
+    if tokenfile.endpoint not in tokenfile.tokens:
+        print("No tokens found for this endpoint.")
+
+
+def endpoint():
+    """
+    Get the current endpoint
+
+    Also displays all endpoints with saved tokens.
+    """
+    tokenfile = TokenFile()
+    print("Current: " + tokenfile.endpoint)
+    print("Saved Tokens found for:")
+    for _endpoint, _ in tokenfile.tokens.items():
+        print("- " + _endpoint)
+
+
+def setCliKey(
+        key: Annotated[str, typer.Argument(help="CLI Key")]
+):
+    """
+    Set the CLI key (Actions Only)
 
-def setCliKey(key: Annotated[str, typer.Argument()]):
+    Same as login with the --key option.
+    Should never be used by the user, only in docker containers launched from within Kleinkram.
+    """
     tokenfile = TokenFile()
     if not tokenfile.endpoint in tokenfile.tokens:
         tokenfile.tokens[tokenfile.endpoint] = {}

kleinkram/main.py

@@ -1,3 +1,4 @@
+import sys
 from datetime import datetime, timedelta
 import os
 
@@ -10,17 +11,17 @@ from rich.table import Table
 
 from .helper import uploadFiles, expand_and_match
 
-from .auth import login, client, endpoint, setCliKey
+from .auth import login, client, endpoint, setCliKey, setEndpoint
 
 app = typer.Typer()
-projects = typer.Typer(name="projects")
-missions = typer.Typer(name="missions")
-files = typer.Typer(name="files")
-topics = typer.Typer(name="topics")
-queue = typer.Typer(name="queue")
-user = typer.Typer(name="users")
-tagtypes = typer.Typer(name="tagtypes")
-tag = typer.Typer(name="tag")
+projects = typer.Typer(name="projects", help="Project operations")
+missions = typer.Typer(name="missions", help="Mission operations")
+files = typer.Typer(name="files", help="File operations")
+topics = typer.Typer(name="topics", help="Topic operations")
+queue = typer.Typer(name="queue", help="Queue operations")
+user = typer.Typer(name="users", help="User operations")
+tagtypes = typer.Typer(name="tagtypes", help="TagType operations")
+tag = typer.Typer(name="tag", help="Tag operations")
 
 
 app.add_typer(projects)
@@ -33,27 +34,39 @@ app.add_typer(tagtypes)
 app.add_typer(tag)
 app.command()(login)
 app.command()(endpoint)
-app.command()(setCliKey)
+app.command()(setEndpoint)
+app.command(hidden=True)(setCliKey)
 
 
 @files.command("list")
 def list_files(
-        project: Annotated[str, typer.Option()] = None,
-        mission: Annotated[str, typer.Option()] = None,
-        topics: Annotated[List[str], typer.Option()] = None,
+        project: Optional[str] = typer.Option(None, help="Name of Project"),
+        mission: Optional[str] = typer.Option(None, help="Name of Mission"),
+        topics: Optional[List[str]] = typer.Option(None, help="Comma separated list of topics")
 ):
     """
     List all files with optional filters for project, mission, or topics.
+
+    Can list files of a project, mission, or with specific topics (Logical AND).
+    Examples:\n
+        - 'klein files list'\n
+        - 'klein files list --project "Project 1"'\n
+        - 'klein files list --mission "Mission 1"'\n
+        - 'klein files list --topics "/elevation_mapping/semantic_map,/elevation_mapping/elevation_map_raw"'\n
+        - 'klein files list --topics "/elevation_mapping/semantic_map,/elevation_mapping/elevation_map_raw" --mission "Mission A"'
     """
     try:
         url = f"/file/filteredByNames"
+        params = {}
+        if project:
+            params["projectName"] = project
+        if mission:
+            params["missionName"] = mission
+        if topics:
+            params["topics"] = topics
         response = client.get(
             url,
-            params={
-                "projectName": project,
-                "missionName": mission,
-                "topics": topics,
-            },
+            params=params,
         )
         response.raise_for_status()
         data = response.json()
@@ -102,11 +115,12 @@ def list_projects():
 
 @missions.command("list")
 def list_missions(
-        project: Annotated[str, typer.Option()] = None,
-        verbose: Annotated[bool, typer.Option()] = False,
+        project: Optional[str] = typer.Option(None, help="Name of Project"),
+        verbose: Optional[bool] = typer.Option(False, help="Outputs a table with more information"),
 ):
     """
     List all missions with optional filter for project.
+
     """
     try:
         url = "/mission"
@@ -150,26 +164,47 @@ def list_missions(
 @missions.command("byUUID")
 def mission_by_uuid(
         uuid: Annotated[str, typer.Argument()],
+        json: Optional[bool] = typer.Option(False, help="Output as JSON"),
 ):
+    """
+    Get mission name, project name, creator and table of its files given a Mission UUID
+
+    Use the JSON flag to output the full JSON response instead.
+
+    Can be run with API Key or with login.
+    """
     try:
         url = "/mission/byUUID"
         response = client.get(url, params={"uuid": uuid})
         response.raise_for_status()
         data = response.json()
-        print(f"mission: {data['name']}")
-        print(f"Creator: {data['creator']['name']}")
-        table = Table("Filename", "Size", "date")
-        for file in data["files"]:
-            table.add_row(file["filename"], f"{file['size']}", file["date"])
+        if json:
+            print(data)
+        else:
+            print(f"mission: {data['name']}")
+            print(f"Creator: {data['creator']['name']}")
+            print("Project: " + data["project"]["name"])
+            table = Table("Filename", "Size", "date")
+            for file in data["files"]:
+                table.add_row(file["filename"], f"{file['size']}", file["date"])
+            print(table)
     except httpx.HTTPError as e:
         print(f"Failed to fetch missions: {e}")
 
 
 @topics.command("list")
 def topics(
-        file: Annotated[str, typer.Option()] = None,
-        full: Annotated[bool, typer.Option()] = False,
+        file: Annotated[str, typer.Option(help="Name of File")],
+        full: Annotated[bool, typer.Option(help="As a table with additional parameters")] = False,
+        # Todo add mission / project as optional argument as filenames are not unique
 ):
+    """
+    List topics for a file
+
+    Only makes sense with MCAP files as we don't associate topics with BAGs.
+    """
+    if file.endswith(".bag"):
+        print("BAG files generally do not have topics")
     try:
         url = "/file/byName"
         response = client.get(url, params={"name": file})
@@ -195,11 +230,19 @@ def topics(
 
 
 @projects.command("create")
-def create_project(name: Annotated[str, typer.Option()]):
+def create_project(
+        name: Annotated[str, typer.Option(help="Name of Project")],
+        description: Annotated[str, typer.Option(help="Description of Project")],
+):
+    """
+    Create a new project
+    """
     try:
         url = "/project/create"
-        response = client.post(url, json={"name": name})
-        response.raise_for_status()
+        response = client.post(url, json={"name": name, "description": description, "requiredTags": []}) # TODO: Add required tags as option
+        if response.status_code >= 400:
+            print(f"Failed to create project: {response.json()["message"]}")
+            return
         print("Project created")
 
     except httpx.HTTPError as e:
@@ -208,12 +251,23 @@ def create_project(name: Annotated[str, typer.Option()]):
 
 @app.command("upload")
 def upload(
-        path: Annotated[str, typer.Option(prompt=True)],
-        project: Annotated[str, typer.Option(prompt=True)],
-        mission: Annotated[str, typer.Option(prompt=True)],
+        path: Annotated[str, typer.Option(prompt=True, help="Path to files to upload, Regex supported")],
+        project: Annotated[str, typer.Option(prompt=True, help="Name of Project")],
+        mission: Annotated[str, typer.Option(prompt=True, help="Name of Mission to create")],
 ):
+    """
+    Upload files matching the path to a mission in a project.
+
+    The mission name must be unique within the project and not yet created.\n
+    Examples:\n
+        - 'klein upload --path "~/data/**/*.bag" --project "Project 1" --mission "Mission 1"'\n
+
+    """
     files = expand_and_match(path)
     filenames = list(map(lambda x: x.split("/")[-1], filter(lambda x: not os.path.isdir(x),files)))
+    if not filenames:
+        print("No files found")
+        return
     filepaths = {}
     for path in files:
         if not os.path.isdir(path):
@@ -245,7 +299,7 @@ def upload(
 
         create_mission_url = "/mission/create"
         new_mission = client.post(
-            create_mission_url, json={"name": mission, "projectUUID": project_json["uuid"]}
+            create_mission_url, json={"name": mission, "projectUUID": project_json["uuid"], "tags": []}
         )
         new_mission.raise_for_status()
         new_mission_data = new_mission.json()
@@ -320,7 +374,7 @@ def clear_queue():
         print("Operation cancelled.")
 
 
-@app.command("wipe")
+@app.command("wipe", hidden=True)
 def wipe():
     """Wipe all data"""
     # Prompt the user for confirmation
@@ -360,8 +414,13 @@ def wipe():
         print("Operation cancelled.")
 
 
-@app.command("claim")
+@app.command("claim", hidden=True)
 def claim():
+    """
+    Claim admin rights as the first user
+
+    Only works if no other user has claimed admin rights before.
+    """
     response = client.post("/user/claimAdmin")
     response.raise_for_status()
     print("Admin claimed.")
@@ -369,6 +428,7 @@ def claim():
 
 @user.command("list")
 def users():
+    """List all users"""
     response = client.get("/user/all")
     response.raise_for_status()
     data = response.json()
@@ -380,6 +440,7 @@ def users():
 
 @user.command("info")
 def user_info():
+    """Get logged in user info"""
     response = client.get("/user/me")
     response.raise_for_status()
     data = response.json()
@@ -388,6 +449,7 @@ def user_info():
 
 @user.command("promote")
 def promote(email: Annotated[str, typer.Option()]):
+    """Promote another user to admin"""
     response = client.post("/user/promote", json={"email": email})
     response.raise_for_status()
     print("User promoted.")
@@ -395,6 +457,7 @@ def promote(email: Annotated[str, typer.Option()]):
 
 @user.command("demote")
 def demote(email: Annotated[str, typer.Option()]):
+    """Demote another user from admin"""
     response = client.post("/user/demote", json={"email": email})
     response.raise_for_status()
     print("User demoted.")
@@ -404,6 +467,7 @@ def demote(email: Annotated[str, typer.Option()]):
 def download(
         missionuuid: Annotated[str, typer.Argument()],
 ):
+    """Download file"""
     try:
         response = client.get("/file/downloadWithToken", params={"uuid": missionuuid})
         response.raise_for_status()
@@ -417,6 +481,7 @@ def addTag(
         tagtypeuuid: Annotated[str, typer.Argument()],
         value: Annotated[str, typer.Argument()],
 ):
+    """Tag a mission"""
     try:
         response = client.post("/tag/addTag", json={"mission": missionuuid, "tagType": tagtypeuuid, "value": value})
         if response.status_code < 400:
@@ -426,13 +491,15 @@ def addTag(
             print("Failed to tag mission")
     except httpx.HTTPError as e:
         print(e)
-        print("Failed to tag mission"
-)
+        print("Failed to tag mission")
+        sys.exit(1)
+
 
 @tagtypes.command('list')
 def tagTypes(
         verbose: Annotated[bool, typer.Option()] = False,
 ):
+    """List all tagtypes"""
     try:
         response = client.get("/tag/all")
         response.raise_for_status()
@@ -453,6 +520,7 @@ def tagTypes(
 def deleteTag(
         taguuid: Annotated[str, typer.Argument()],
 ):
+    """Delete a tag"""
     try:
         response = client.delete("/tag/deleteTag", params={"uuid": taguuid})
         if response.status_code < 400:

{kleinkram-0.0.116.dist-info → kleinkram-0.0.118.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: kleinkram
-Version: 0.0.116
+Version: 0.0.118
 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.0.118.dist-info/RECORD

@@ -0,0 +1,10 @@
+kleinkram/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+kleinkram/auth.py,sha256=s6Lek1vLRs2cXSwfGmEWuuzmszMhaTAJbh9icxWPSkQ,7628
+kleinkram/consts.py,sha256=8kCOeSdMqEyB89hT55w5yd1fksUwbj52a2nYkZTZI_0,88
+kleinkram/helper.py,sha256=mP6AohHWU8kckAi1Oevgs0gfGPH1Qzkp0f7jiQF9u5I,2198
+kleinkram/main.py,sha256=O1VOT8YFOc6YcyLhd723iOScvO1ILUG2LW1ggkUZVJY,18071
+kleinkram-0.0.118.dist-info/METADATA,sha256=8eKAgwE0t4W6ecNaQ4r3eOnzgbCa-6ArE_aetX_otHk,733
+kleinkram-0.0.118.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
+kleinkram-0.0.118.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
+kleinkram-0.0.118.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
+kleinkram-0.0.118.dist-info/RECORD,,

kleinkram-0.0.116.dist-info/RECORD

@@ -1,10 +0,0 @@
-kleinkram/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-kleinkram/auth.py,sha256=0pu0RQQrtTFoW07Rgni7FAMJS3e0BZ9Pz9QrFPsbfxg,5343
-kleinkram/consts.py,sha256=8kCOeSdMqEyB89hT55w5yd1fksUwbj52a2nYkZTZI_0,88
-kleinkram/helper.py,sha256=mP6AohHWU8kckAi1Oevgs0gfGPH1Qzkp0f7jiQF9u5I,2198
-kleinkram/main.py,sha256=dS0gOy_KRqrzbX5xynROi6cnbDb_9Aww2ER4U4bpfU4,15106
-kleinkram-0.0.116.dist-info/METADATA,sha256=ANiDjnZ0f-h4P6d6pUmUaeBDOroCDfojKMpDL_3d5Yk,733
-kleinkram-0.0.116.dist-info/WHEEL,sha256=1yFddiXMmvYK7QYTqtRNtX66WJ0Mz8PYEiEUoOUUxRY,87
-kleinkram-0.0.116.dist-info/entry_points.txt,sha256=RHXtRzcreVHImatgjhQwZQ6GdJThElYjHEWcR1BPXUI,45
-kleinkram-0.0.116.dist-info/licenses/LICENSE,sha256=ixuiBLtpoK3iv89l7ylKkg9rs2GzF9ukPH7ynZYzK5s,35148
-kleinkram-0.0.116.dist-info/RECORD,,