tinybird 0.0.1.dev304__py3-none-any.whl → 0.0.1.dev305__py3-none-any.whl

tinybird/sql_template.py

@@ -2385,6 +2385,14 @@ def render_sql_template(
                 documentation="/cli/advanced-templates.html",
             )
         raise SQLTemplateException(str(e), documentation="/cli/advanced-templates.html")
+    except IndexError as e:
+        # This happens when trying to access string indices on empty strings
+        if "string index out of range" in str(e):
+            raise SQLTemplateException(
+                "String index out of range. Check that string parameters have values before accessing specific characters (e.g., param[0]). Provide default values or add length checks in your template.",
+                documentation="/cli/advanced-templates.html",
+            )
+        raise SQLTemplateException(str(e), documentation="/cli/advanced-templates.html")
     except Exception as e:
         # errors might vary here, we need to support as much as possible
         # https://gitlab.com/tinybird/analytics/-/issues/943

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.dev304'
-__revision__ = '9a414a1'
+__version__ = '0.0.1.dev305'
+__revision__ = 'cc453b3'

tinybird/tb/modules/local.py

@@ -1,11 +1,15 @@
+import json
+import uuid
 from pathlib import Path
+from typing import Any
 
 import click
+import jwt
 import requests
 from docker.client import DockerClient
 
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import update_cli
+from tinybird.tb.modules.common import echo_json, update_cli
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.local_common import (
     TB_CONTAINER_NAME,
@@ -125,7 +129,21 @@ def remove() -> None:
     is_flag=True,
     help="Skip pulling the latest Tinybird Local image. Use directly your current local image.",
 )
-def start(use_aws_creds: bool, volumes_path: str, skip_new_version: bool) -> None:
+@click.option(
+    "--user-token",
+    default=None,
+    envvar="TB_LOCAL_USER_TOKEN",
+    help="User token to use for the Tinybird Local container.",
+)
+@click.option(
+    "--workspace-token",
+    default=None,
+    envvar="TB_LOCAL_WORKSPACE_TOKEN",
+    help="Workspace token to use for the Tinybird Local container.",
+)
+def start(
+    use_aws_creds: bool, volumes_path: str, skip_new_version: bool, user_token: str, workspace_token: str
+) -> None:
     """Start Tinybird Local"""
     if volumes_path is not None:
         absolute_path = Path(volumes_path).absolute()
@@ -134,7 +152,7 @@ def start(use_aws_creds: bool, volumes_path: str, skip_new_version: bool) -> Non
 
     click.echo(FeedbackManager.highlight(message="» Starting Tinybird Local..."))
     docker_client = get_docker_client()
-    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version)
+    start_tinybird_local(docker_client, use_aws_creds, volumes_path, skip_new_version, user_token, workspace_token)
     click.echo(FeedbackManager.success(message="✓ Tinybird Local is ready!"))
 
 
@@ -183,3 +201,35 @@ def version() -> None:
     """Show Tinybird Local version"""
     response = requests.get(f"{TB_LOCAL_ADDRESS}/version")
     click.echo(FeedbackManager.success(message=f"✓ Tinybird Local version: {response.text}"))
+
+
+@local.command()
+@click.pass_context
+def generate_tokens(ctx: click.Context) -> None:
+    """Generate static tokens for initializing Tinybird Local"""
+    output = ctx.ensure_object(dict).get("output")
+    user_id = str(uuid.uuid4())
+    workspace_id = str(uuid.uuid4())
+    user_token_id = str(uuid.uuid4())
+    workspace_token_id = str(uuid.uuid4())
+    payload = {"u": user_id, "id": user_token_id, "host": None}
+    user_token = generate_token(payload)
+    payload = {"u": workspace_id, "id": workspace_token_id, "host": None}
+    workspace_token = generate_token(payload)
+
+    if output == "json":
+        echo_json({"workspace_token": workspace_token, "user_token": user_token})
+    else:
+        click.echo(FeedbackManager.gray(message="Workspace token: ") + FeedbackManager.info(message=workspace_token))
+        click.echo(FeedbackManager.gray(message="User token: ") + FeedbackManager.info(message=user_token))
+        click.echo(FeedbackManager.success(message="✓ Tinybird Local tokens generated!"))
+
+
+def generate_token(payload: dict[str, Any]) -> str:
+    algo = jwt.algorithms.get_default_algorithms()["HS256"]
+    msg = json.dumps(payload)
+    msg_base64 = jwt.utils.base64url_encode(msg.encode())
+    sign_key = algo.prepare_key("abcd")
+    signature = algo.sign(msg_base64, sign_key)
+    token = msg_base64 + b"." + jwt.utils.base64url_encode(signature)
+    return "p." + token.decode()

tinybird/tb/modules/local_common.py

@@ -234,7 +234,9 @@ def start_tinybird_local(
     docker_client: DockerClient,
     use_aws_creds: bool,
     volumes_path: Optional[str] = None,
-    skip_new_version: bool = False,
+    skip_new_version: bool = True,
+    user_token: Optional[str] = None,
+    workspace_token: Optional[str] = None,
 ) -> None:
     """Start the Tinybird container."""
     pull_show_prompt = False
@@ -261,7 +263,13 @@ def start_tinybird_local(
         if pull_required:
             docker_client.images.pull(TB_IMAGE_NAME, platform="linux/amd64")
 
-    environment = get_use_aws_creds() if use_aws_creds else {}
+    environment = {}
+    if use_aws_creds:
+        environment.update(get_use_aws_creds())
+    if user_token:
+        environment["TB_LOCAL_USER_TOKEN"] = user_token
+    if workspace_token:
+        environment["TB_LOCAL_WORKSPACE_TOKEN"] = workspace_token
 
     container = get_existing_container_with_matching_env(docker_client, TB_CONTAINER_NAME, environment)
 

{tinybird-0.0.1.dev304.dist-info → tinybird-0.0.1.dev305.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev304
+Version: 0.0.1.dev305
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/forward/commands
 Author: Tinybird
@@ -39,6 +39,7 @@ Requires-Dist: packaging<24,>=23.1
 Requires-Dist: llm>=0.19
 Requires-Dist: thefuzz==0.22.1
 Requires-Dist: python-dotenv==1.1.0
+Requires-Dist: pyjwt[crypto]==2.9.0
 Dynamic: author
 Dynamic: author-email
 Dynamic: description

{tinybird-0.0.1.dev304.dist-info → tinybird-0.0.1.dev305.dist-info}/RECORD

@@ -6,7 +6,7 @@ tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
 tinybird/prompts.py,sha256=HoDv9TxPiP8v2XoGTWYxP133dK9CEbXVv4XE5IT339c,45483
 tinybird/service_datasources.py,sha256=y36l2QYLxFEYRBEBuYCKtjkPjGu0P2a8Fz_AjOTTNM4,48914
 tinybird/sql.py,sha256=7pkwQMUVy0CH5zFgLMZyCx1BeaJSQYC05EmOb9vO3Ls,48694
-tinybird/sql_template.py,sha256=vOpIAgHoLu3D3aFwywgYHJgRmYxtrxvro8sRwYHX-tk,102415
+tinybird/sql_template.py,sha256=qz1Q-d8jlGpU8w-jcwn_NM4ApFtJAhEL-3_3-JELrl4,102979
 tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
 tinybird/sql_toolset.py,sha256=Y2_fQrbk5GSNoRncAmRS_snpV61XQbiOy4uYkeF6pr4,19767
 tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
@@ -18,7 +18,7 @@ tinybird/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1w
 tinybird/datafile/parse_connection.py,sha256=tRyn2Rpr1TeWet5BXmMoQgaotbGdYep1qiTak_OqC5E,1825
 tinybird/datafile/parse_datasource.py,sha256=ssW8QeFSgglVFi3sDZj_HgkJiTJ2069v2JgqnH3CkDE,1825
 tinybird/datafile/parse_pipe.py,sha256=8e9LMecSQVWHC4AXf8cdxoQ1nxUR4fTObYxTctO_EXQ,3816
-tinybird/tb/__cli__.py,sha256=zKFXL0dY2JM1qGDKRazQWAKaCs3Ex5FQRqD_WLnFa5Q,247
+tinybird/tb/__cli__.py,sha256=KoDxX7t1llHSTiafcOR0AIeehU6WnVkQeNq7EuQ6nuw,247
 tinybird/tb/check_pypi.py,sha256=Gp0HkHHDFMSDL6nxKlOY51z7z1Uv-2LRexNTZSHHGmM,552
 tinybird/tb/cli.py,sha256=FdDFEIayjmsZEVsVSSvRiVYn_FHOVg_zWQzchnzfWho,1008
 tinybird/tb/client.py,sha256=Z27O6kXbQ1ZVi_K6qzdUJoS877nt7rqQDRckOycjNmI,53544
@@ -45,8 +45,8 @@ tinybird/tb/modules/infra.py,sha256=J9Noe9aZo5Y9ZKAhqh9jnv8azfivXLLHQ2a9TeMWN9s,
 tinybird/tb/modules/job.py,sha256=wBsnu8UPTOha2rkLvucgmw4xYv73ubmui3eeSIF68ZM,3107
 tinybird/tb/modules/llm.py,sha256=fPBBCmM3KlCksLlgJkg4joDn6y3H5QjDzE-Pm4YNf7E,1782
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
-tinybird/tb/modules/local.py,sha256=cez8ZV7Gt9F1siz54dpgtynEw-il3Y9mNcQgEtOVvmw,6512
-tinybird/tb/modules/local_common.py,sha256=KhALzU0L3AC56vqs3peBe-dqGzw5hWkOg_CfS0skohk,18528
+tinybird/tb/modules/local.py,sha256=aGqBtRyr9Ks0BWUFEveez2g_-4h6YjLO6Rt49b1i-yE,8384
+tinybird/tb/modules/local_common.py,sha256=UUDGS5M3C261NUtZO_LrmfEYfUH1sjn1kjjBMlY32xI,18801
 tinybird/tb/modules/login.py,sha256=q2fc0dy5brwofQSNQppipS1aDcyrxHvVjZZbzIA3b4g,1916
 tinybird/tb/modules/login_common.py,sha256=u1Eh1u6L7V3x8qmxVO-LnWDELHWAnXm3bvb7BNjVXF8,11990
 tinybird/tb/modules/logout.py,sha256=sniI4JNxpTrVeRCp0oGJuQ3yRerG4hH5uz6oBmjv724,1009
@@ -121,8 +121,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.dev304.dist-info/METADATA,sha256=lhsEuveQfvJdP9vvlunpZqBKmKtSyhi4V7cy3xfe8nw,1845
-tinybird-0.0.1.dev304.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev304.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev304.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev304.dist-info/RECORD,,
+tinybird-0.0.1.dev305.dist-info/METADATA,sha256=kf9UMEPqhBrJvZD_DpAjOJ-demRErf4gMk42ifnZ7cQ,1881
+tinybird-0.0.1.dev305.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev305.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev305.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev305.dist-info/RECORD,,