tinybird 0.0.1.dev161__tar.gz → 0.0.1.dev162__tar.gz

{tinybird-0.0.1.dev161 → tinybird-0.0.1.dev162}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev161
+Version: 0.0.1.dev162
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird

{tinybird-0.0.1.dev161 → tinybird-0.0.1.dev162}/tinybird/tb/__cli__.py

@@ -4,5 +4,5 @@ __description__ = 'Tinybird Command Line Tool'
 __url__ = 'https://www.tinybird.co/docs/forward/commands'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev161'
-__revision__ = '29ac536'
+__version__ = '0.0.1.dev162'
+__revision__ = 'ccf549a'

{tinybird-0.0.1.dev161 → tinybird-0.0.1.dev162}/tinybird/tb/modules/local_common.py

@@ -36,31 +36,7 @@ async def get_tinybird_local_config(config_obj: Dict[str, Any], test: bool = Fal
     """
     path = config_obj.get("path")
     config = CLIConfig.get_project_config()
-
-    try:
-        # ruff: noqa: ASYNC210
-        tokens = requests.get(f"{TB_LOCAL_ADDRESS}/tokens").json()
-    except Exception:
-        try:
-            # Check if tinybird-local is running with docker, in case it's a config issue
-            output = subprocess.check_output(["docker", "ps"], text=True)  # noqa: ASYNC221
-            header_row = next((line for line in output.splitlines() if "CONTAINER ID" in line), "")
-            tb_local_row = next(
-                (line for line in output.splitlines() if TB_CONTAINER_NAME in line),
-                f"{TB_CONTAINER_NAME} not found in output",
-            )
-            add_telemetry_event(
-                "docker_debug",
-                data={
-                    "docker_ps_output": header_row + tb_local_row,
-                },
-            )
-        except Exception:
-            pass
-        raise CLILocalException(
-            FeedbackManager.error(message="Tinybird local is not running. Please run `tb local start` first.")
-        )
-
+    tokens = get_local_tokens()
     user_token = tokens["user_token"]
     admin_token = tokens["admin_token"]
     default_token = tokens["workspace_admin_token"]
@@ -118,3 +94,29 @@ def get_build_workspace_name(path: str) -> str:
 def get_test_workspace_name(path: str) -> str:
     folder_hash = hashlib.sha256(path.encode()).hexdigest()
     return f"Tinybird_Local_Test_{folder_hash}"
+
+
+def get_local_tokens() -> Dict[str, str]:
+    try:
+        # ruff: noqa: ASYNC210
+        return requests.get(f"{TB_LOCAL_ADDRESS}/tokens").json()
+    except Exception:
+        try:
+            # Check if tinybird-local is running with docker, in case it's a config issue
+            output = subprocess.check_output(["docker", "ps"], text=True)
+            header_row = next((line for line in output.splitlines() if "CONTAINER ID" in line), "")
+            tb_local_row = next(
+                (line for line in output.splitlines() if TB_CONTAINER_NAME in line),
+                f"{TB_CONTAINER_NAME} not found in output",
+            )
+            add_telemetry_event(
+                "docker_debug",
+                data={
+                    "docker_ps_output": header_row + tb_local_row,
+                },
+            )
+        except Exception:
+            pass
+        raise CLILocalException(
+            FeedbackManager.error(message="Tinybird local is not running. Please run `tb local start` first.")
+        )

{tinybird-0.0.1.dev161 → tinybird-0.0.1.dev162}/tinybird/tb/modules/workspace.py

@@ -6,9 +6,10 @@
 from typing import Any, Dict, List, Optional
 
 import click
+import requests
 from click import Context
 
-from tinybird.tb.client import TinyB
+from tinybird.tb.client import AuthNoTokenException, TinyB
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import (
     _get_workspace_plan_name,
@@ -26,6 +27,10 @@ from tinybird.tb.modules.common import (
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import CLIWorkspaceException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.local_common import (
+    TB_LOCAL_ADDRESS,
+    get_local_tokens,
+)
 
 
 @cli.group()
@@ -221,3 +226,57 @@ async def delete_workspace(
             click.echo(FeedbackManager.success_workspace_deleted(workspace_name=workspace_to_delete["name"]))
         except Exception as e:
             raise CLIWorkspaceException(FeedbackManager.error_exception(error=str(e)))
+
+
+@workspace.command(name="clear", short_help="Clear a workspace. Only available against Tinybird Local.")
+@click.option("--yes", is_flag=True, default=False, help="Don't ask for confirmation")
+@click.pass_context
+@coro
+async def workspace_clear(ctx: Context, yes: bool) -> None:
+    """Delete a workspace where you are an admin."""
+    is_cloud = ctx.ensure_object(dict)["env"] == "cloud"
+    if is_cloud:
+        raise CLIWorkspaceException(
+            FeedbackManager.error(
+                message="`tb workspace clear` is not available against Tinybird Cloud. Use `tb --cloud deploy` instead."
+            )
+        )
+    yes = yes or click.confirm(
+        FeedbackManager.warning(message="Are you sure you want to clear the workspace? [y/N]:"),
+        show_default=False,
+        prompt_suffix="",
+    )
+    if yes:
+        await clear_workspace()
+
+
+async def clear_workspace() -> None:
+    config = CLIConfig.get_project_config()
+    tokens = get_local_tokens()
+
+    user_token = tokens["user_token"]
+    admin_token = tokens["admin_token"]
+    user_client = config.get_client(host=TB_LOCAL_ADDRESS, token=user_token)
+    ws_name = config.get("name")
+    if not ws_name:
+        raise AuthNoTokenException()
+
+    user_workspaces = requests.get(  # noqa: ASYNC210
+        f"{TB_LOCAL_ADDRESS}/v1/user/workspaces?with_organization=true&token={admin_token}"
+    ).json()
+    user_org_id = user_workspaces.get("organization_id", {})
+    local_workspaces = user_workspaces.get("workspaces", [])
+
+    ws = next((ws for ws in local_workspaces if ws["name"] == ws_name), None)
+
+    if ws:
+        requests.delete(f"{TB_LOCAL_ADDRESS}/v1/workspaces/{ws['id']}?token={user_token}&hard_delete_confirmation=yes")  # noqa: ASYNC210
+        ws = None
+
+    if not ws:
+        await user_client.create_workspace(ws_name, assign_to_organization_id=user_org_id, version="v1")
+        user_workspaces = requests.get(f"{TB_LOCAL_ADDRESS}/v1/user/workspaces?token={admin_token}").json()  # noqa: ASYNC210
+        ws = next((ws for ws in user_workspaces["workspaces"] if ws["name"] == ws_name), None)
+
+    if ws:
+        click.echo(FeedbackManager.success(message=f"\n✓ Workspace '{ws_name}' cleared"))

{tinybird-0.0.1.dev161 → tinybird-0.0.1.dev162}/tinybird.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev161
+Version: 0.0.1.dev162
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird