latch 2.76.2__tar.gz → 2.76.4__tar.gz

{latch-2.76.2 → latch-2.76.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: latch
-Version: 2.76.2
+Version: 2.76.4
 Summary: The Latch SDK
 Project-URL: Homepage, https://latch.bio
 Project-URL: Documentation, https://latch.wiki

{latch-2.76.2 → latch-2.76.4}/pyproject.toml

@@ -12,7 +12,7 @@ include = ["src/**/*.py", "src/**/*.graphql", "src/**/py.typed", "src/latch_cli/
 
 [project]
 name = "latch"
-version = "2.76.2"
+version = "2.76.4"
 description = "The Latch SDK"
 authors = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]
 maintainers = [{ name = "Kenny Workman", email = "kenny@latch.bio" }]

{latch-2.76.2 → latch-2.76.4}/src/latch_cli/main.py

@@ -150,8 +150,15 @@ def login(connection: Optional[str]):
 
 
 @latch.command("workspace")
+@click.option(
+    "--id",
+    "workspace_id",
+    type=str,
+    default=None,
+    help="Workspace ID to switch to without opening the interactive selector.",
+)
 @requires_login
-def workspace():
+def workspace(workspace_id: Optional[str]):
     """Spawns an interactive terminal prompt allowing users to choose what workspace they want to work in."""
 
     crash_handler.message = "Unable to fetch workspaces"
@@ -159,7 +166,7 @@ def workspace():
 
     from latch_cli.services.workspace import workspace
 
-    workspace()
+    workspace(workspace_id)
 
 
 """

{latch-2.76.2 → latch-2.76.4}/src/latch_cli/services/workspace.py

@@ -1,7 +1,6 @@
-from typing import List, TypedDict
+from typing import Optional
 
 import click
-from latch_sdk_config.user import user_config
 
 from latch.utils import (
     NoWorkspaceSelectedError,
@@ -10,11 +9,11 @@ from latch.utils import (
     get_workspaces,
 )
 from latch_cli.menus import SelectOption, select_tui
+from latch_sdk_config.user import user_config
 
 
-def workspace():
-    """Opens a terminal user interface in which a user can select the workspace
-    the want to switch to.
+def workspace(workspace_id: Optional[str] = None):
+    """Opens a terminal user interface in which a user can select the workspace they want to switch to.
 
     Like `get_executions`, this function should only be called from the CLI.
     """
@@ -26,9 +25,32 @@ def workspace():
     except NoWorkspaceSelectedError:
         old_id = None
 
+    if workspace_id is not None:
+        selected_option = data.get(workspace_id)
+        if selected_option is None:
+            click.secho(
+                f"Workspace {workspace_id} does not exist or you do not have permission to access it.",
+                fg="red",
+                bold=True,
+            )
+            raise click.exceptions.Exit(1)
+
+        user_config.update_workspace(
+            selected_option["workspace_id"], selected_option["name"]
+        )
+
+        if old_id != selected_option["workspace_id"]:
+            click.secho(
+                f"Successfully switched to context {selected_option['name']}",
+                fg="green",
+            )
+        else:
+            click.secho(f"Already in context {selected_option['name']}.", fg="green")
+        return
+
     selected_marker = "\x1b[3m\x1b[2m (currently selected) \x1b[22m\x1b[23m"
 
-    options: List[SelectOption[WSInfo]] = []
+    options: list[SelectOption[WSInfo]] = []
     for _, info in sorted(
         data.items(), key=lambda x: "" if x[1]["default"] else x[1]["name"]
     ):