tinybird 0.0.1.dev155__py3-none-any.whl → 0.0.1.dev156__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.dev155'
-__revision__ = '4d0ff0d'
+__version__ = '0.0.1.dev156'
+__revision__ = '888fa61'

tinybird/tb/modules/workspace.py

@@ -3,7 +3,7 @@
 # - If it makes sense and only when strictly necessary, you can create utility functions in this file.
 # - But please, **do not** interleave utility functions and command definitions.
 
-from typing import Optional
+from typing import Any, Dict, List, Optional
 
 import click
 from click import Context
@@ -21,8 +21,8 @@ from tinybird.tb.modules.common import (
     get_organizations_by_user,
     get_user_token,
     is_valid_starterkit,
-    print_current_workspace,
     switch_workspace,
+    try_update_config_with_remote,
 )
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import CLIWorkspaceException
@@ -70,21 +70,65 @@ async def workspace_ls(ctx: Context) -> None:
 
 @workspace.command(name="use")
 @click.argument("workspace_name_or_id")
+@click.pass_context
 @coro
-async def workspace_use(workspace_name_or_id: str) -> None:
+async def workspace_use(ctx: Context, workspace_name_or_id: str) -> None:
     """Switch to another workspace. Use 'tb workspace ls' to list the workspaces you have access to."""
     config = CLIConfig.get_project_config()
+    is_cloud = ctx.ensure_object(dict)["env"] == "cloud"
+    if not is_cloud:
+        raise CLIWorkspaceException(
+            FeedbackManager.error(
+                message="`tb workspace use` is not available in local mode. Use --cloud to switch to a cloud workspace and it will be used in Tinybird Local."
+            )
+        )
 
     await switch_workspace(config, workspace_name_or_id)
 
 
 @workspace.command(name="current")
+@click.pass_context
 @coro
-async def workspace_current():
+async def workspace_current(ctx: Context):
     """Show the workspace you're currently authenticated to"""
     config = CLIConfig.get_project_config()
+    env = ctx.ensure_object(dict)["env"]
+    client: TinyB = ctx.ensure_object(dict)["client"]
+    if env == "cloud":
+        _ = await try_update_config_with_remote(config, only_if_needed=True)
+
+    user_workspaces = await client.user_workspaces(version="v1")
+    current_workspace = await client.workspace_info(version="v1")
+
+    def _get_current_workspace(user_workspaces: Dict[str, Any], current_workspace_id: str) -> Optional[Dict[str, Any]]:
+        def get_workspace_by_name(workspaces: List[Dict[str, Any]], name: str) -> Optional[Dict[str, Any]]:
+            return next((ws for ws in workspaces if ws["name"] == name), None)
+
+        workspaces: Optional[List[Dict[str, Any]]] = user_workspaces.get("workspaces")
+        if not workspaces:
+            return None
+
+        current: Optional[Dict[str, Any]] = get_workspace_by_name(workspaces, current_workspace_id)
+        return current
 
-    await print_current_workspace(config)
+    current_main_workspace = _get_current_workspace(user_workspaces, config.get("name", current_workspace["name"]))
+
+    assert isinstance(current_main_workspace, dict)
+
+    columns = ["name", "id", "role", "plan", "current"]
+
+    table = [
+        (
+            current_main_workspace["name"],
+            current_main_workspace["id"],
+            current_main_workspace["role"],
+            _get_workspace_plan_name(current_main_workspace["plan"]),
+            True,
+        )
+    ]
+
+    click.echo(FeedbackManager.info_current_workspace())
+    echo_safe_humanfriendly_tables_format_smart_table(table, column_names=columns)
 
 
 @workspace.command(name="create", short_help="Create a new Workspace for your Tinybird user")

{tinybird-0.0.1.dev155.dist-info → tinybird-0.0.1.dev156.dist-info}/METADATA

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

{tinybird-0.0.1.dev155.dist-info → tinybird-0.0.1.dev156.dist-info}/RECORD

@@ -12,7 +12,7 @@ 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=c0frnD9sSC9DoVv6tmOBeg9cbiU8bfwHuvqBYJ8hu0Y,247
+tinybird/tb/__cli__.py,sha256=9Uom43LFpjyePOXqbI6-l2C6x4WEikJFuS8e2W83184,247
 tinybird/tb/check_pypi.py,sha256=rW4QmDRbtgKdUUwJCnBkVjmTjZSZGN-XgZhx7vMkC0w,1009
 tinybird/tb/cli.py,sha256=uPV6pvi4aYVfaiGs0DQO-eoi1g9dHlrgvhutkXkdJko,1075
 tinybird/tb/client.py,sha256=aaPKq5C77e72kR7IMv9WrvnvNki8mKMOTi9EsCp0eUc,55962
@@ -54,7 +54,7 @@ tinybird/tb/modules/telemetry.py,sha256=IHHks8-jnrkyQIY_Y6ZmU8hdVHMGSS0pkSUOgCPc
 tinybird/tb/modules/test.py,sha256=Yopg89cRwOQpgRzsb9nvu2Z-UR2as2vBjVa5PF3uiK0,13420
 tinybird/tb/modules/token.py,sha256=2fmKwu10_M0pqs6YmJVeILR9ZQB0ejRAET86agASbKM,13488
 tinybird/tb/modules/watch.py,sha256=_92co0BjTikQuy7MbHr4TDu9A75QdHsrAO8v7DlYLg4,8898
-tinybird/tb/modules/workspace.py,sha256=SlK08psp0tu5t_URHyIczm696buW6KD6FPs9Lg1aNRE,6614
+tinybird/tb/modules/workspace.py,sha256=JIPfsh243gCiCkB3bw0Zx0VcivCeMq34xqlfuT3aURE,8518
 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.dev155.dist-info/METADATA,sha256=j8A-XBOYKUMU7mB9rvXYDikP6UqX0zLcOrHO861Bg7k,1607
-tinybird-0.0.1.dev155.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev155.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev155.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev155.dist-info/RECORD,,
+tinybird-0.0.1.dev156.dist-info/METADATA,sha256=aukxrlc7qfXhgKxSlMHSAOoWo15HAwKnXu2stIMf4bI,1607
+tinybird-0.0.1.dev156.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev156.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev156.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev156.dist-info/RECORD,,