tinybird 0.0.1.dev161__py3-none-any.whl → 0.0.1.dev163__py3-none-any.whl

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.dev163'
+__revision__ = 'dff38d5'

tinybird/tb/modules/cli.py

@@ -317,7 +317,23 @@ def __unpatch_click_output():
 
 
 async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, staging: bool):
-    commands_without_ctx_client = ["auth", "check", "local", "login", "logout", "update", "upgrade", "create", "info"]
+    commands_without_ctx_client = [
+        "auth",
+        "check",
+        "local",
+        "login",
+        "logout",
+        "update",
+        "upgrade",
+        "create",
+        "info",
+        "tag",
+        "push",
+        "branch",
+        "diff",
+        "fmt",
+        "init",
+    ]
     command = ctx.invoked_subcommand
     if command in commands_without_ctx_client:
         return None

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/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.dist-info → tinybird-0.0.1.dev163.dist-info}/METADATA

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

{tinybird-0.0.1.dev161.dist-info → tinybird-0.0.1.dev163.dist-info}/RECORD

@@ -12,14 +12,14 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=BZuPM7MFS7vaEKK5tOMR2bwSAgJudPrJt27uVEwZmTY,40512
-tinybird/tb/__cli__.py,sha256=E8aXjTvNw5qa6fTVCqpIk7TJE7KKkGYHZ4NJ1NbwTmw,247
+tinybird/tb/__cli__.py,sha256=PeR2Y103-abS6voCkOb0Nk66NulYrLNJHp96nj-rAhg,247
 tinybird/tb/check_pypi.py,sha256=rW4QmDRbtgKdUUwJCnBkVjmTjZSZGN-XgZhx7vMkC0w,1009
 tinybird/tb/cli.py,sha256=u3eGOhX0MHkuT6tiwaZ0_3twqLmqKXDAOxF7yV_Nn9Q,1075
 tinybird/tb/client.py,sha256=CSBl_JRuioPyY0H8Ac96dJ9wQXDXfrvK2lwqlOxKGoY,55715
 tinybird/tb/config.py,sha256=jT9xndpeCY_g0HdB5qE2EquC0TFRRnkPnQFWZWd04jo,3998
 tinybird/tb/modules/build.py,sha256=zakH5812Lop-XHjGmDRdOPeofPtoeyb_2un_T6e50xw,19177
 tinybird/tb/modules/cicd.py,sha256=MnShTTJzKBYeajswF2jg7p7ZzupaeCgSriAN05MeEdg,7330
-tinybird/tb/modules/cli.py,sha256=jJ7vF05vppwqetaHiEyk_x2YFb6S53PS-PmYlNMRcTc,14274
+tinybird/tb/modules/cli.py,sha256=vADGfvo8lOp9-VIYD_d3KVIlcYfQk0ciJawthn-LFaw,14449
 tinybird/tb/modules/common.py,sha256=_mNLBzC7zkveYXgJ02aMJ9L3LrxsAELx84GwEYdWNa0,82955
 tinybird/tb/modules/config.py,sha256=ziqW_t_mRVvWOd85VoB4vKyvgMkEfpXDf9H4v38p2xc,11422
 tinybird/tb/modules/connection.py,sha256=7oOR7x4PhBcm1ETFFCH2YJ_3oeGXjAbmx1cnZX9_L70,9014
@@ -38,7 +38,7 @@ tinybird/tb/modules/job.py,sha256=n4dSSBgnA8NqD7srGahf2xRj6wxkmX9Vl0J-QJ_a2w0,29
 tinybird/tb/modules/llm.py,sha256=KfsCYmKeW1VQz0iDZhGKCRkQv_Y3kTHh6JuxvofOguE,1076
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
 tinybird/tb/modules/local.py,sha256=SUaGWH9TLDFFF9uCw4y7UW4NsKgnXG8uxTcxz1dbkCM,14230
-tinybird/tb/modules/local_common.py,sha256=OoBUFrMjXvQ3y8hHscpyVH9wMI7h0m79F7G6EiMSku0,4783
+tinybird/tb/modules/local_common.py,sha256=LutxdEG6AR1k5PrUPsDWGVPrr3mA7AbWKhj4Hlewlic,4837
 tinybird/tb/modules/login.py,sha256=fmXPSdvJnKPv03chptGuu3_Fm6LhP6kUsUKhrmT8rJc,8269
 tinybird/tb/modules/logout.py,sha256=ULooy1cDBD02-r7voZmhV7udA0ML5tVuflJyShrh56Y,1022
 tinybird/tb/modules/materialization.py,sha256=QJX5kCPhhm6IXBO1JsalVfbQdypCe_eOUDZ_WHJZWS8,5478
@@ -54,7 +54,7 @@ tinybird/tb/modules/telemetry.py,sha256=X0p5AVkM8BNsK_Rhdcg4p2eIf6OHimHO_VLldBqH
 tinybird/tb/modules/test.py,sha256=Yopg89cRwOQpgRzsb9nvu2Z-UR2as2vBjVa5PF3uiK0,13420
 tinybird/tb/modules/token.py,sha256=2fmKwu10_M0pqs6YmJVeILR9ZQB0ejRAET86agASbKM,13488
 tinybird/tb/modules/watch.py,sha256=H1FieLTVGRqmZ0hR0vELbQJ9l0CThrFCgGCta-MPuAY,8883
-tinybird/tb/modules/workspace.py,sha256=h4Re1_2DahZ9C-yF7vgquMrYiG7Gj62jNfbzkEJR2BY,8066
+tinybird/tb/modules/workspace.py,sha256=v9o4ImrEOxLhCYhVGtJ-slek8DGOW0cJGn7F91DMmj4,10390
 tinybird/tb/modules/workspace_members.py,sha256=RYLpyPM1ECCasHRg3uvpckzXplX0_KgNFsSPZn_i6qk,8744
 tinybird/tb/modules/datafile/build.py,sha256=d_h3pRFDPFrDKGhpFx2iejY25GuB2k8yfNouj6s8caw,50973
 tinybird/tb/modules/datafile/build_common.py,sha256=LU24kAQmxDJIyoIapDaYG-SU3P4FrMG9UBf8m9PgVSI,4565
@@ -80,8 +80,8 @@ tinybird/tb_cli_modules/config.py,sha256=IsgdtFRnUrkY8-Zo32lmk6O7u3bHie1QCxLwgp4
 tinybird/tb_cli_modules/exceptions.py,sha256=pmucP4kTF4irIt7dXiG-FcnI-o3mvDusPmch1L8RCWk,3367
 tinybird/tb_cli_modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird-0.0.1.dev161.dist-info/METADATA,sha256=zcXNbkwS4Ut4VaoEViYVtbi9qS0bQ82ra-c_nzjJ2pA,1607
-tinybird-0.0.1.dev161.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev161.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev161.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev161.dist-info/RECORD,,
+tinybird-0.0.1.dev163.dist-info/METADATA,sha256=2RhhimAjREyciOOUnO8pJ2Yw-HxEsDfGOZlYfgtIYNM,1607
+tinybird-0.0.1.dev163.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev163.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev163.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev163.dist-info/RECORD,,