tinybird 0.0.1.dev69__py3-none-any.whl → 0.0.1.dev71__py3-none-any.whl

tinybird/client.py

@@ -93,6 +93,7 @@ class TinyB:
         send_telemetry: bool = False,
         semver: Optional[str] = None,
         env: Optional[str] = "production",
+        preview: bool = False,
     ):
         ctx = ssl.create_default_context()
         ctx.check_hostname = False
@@ -105,6 +106,7 @@ class TinyB:
         self.send_telemetry = send_telemetry
         self.semver = semver
         self.env = env
+        self.preview = preview
 
     async def _req_raw(
         self,
@@ -125,6 +127,8 @@ class TinyB:
             url += ("&" if "?" in url else "?") + "cli_version=" + quote(self.version)
         if self.semver:
             url += ("&" if "?" in url else "?") + "__tb__semver=" + self.semver
+        if self.preview:
+            url += ("&" if "?" in url else "?") + "__tb__deployment=preview"
 
         verify_ssl = not self.disable_ssl_checks
         try:
@@ -197,7 +201,7 @@ class TinyB:
             if not token_to_use:
                 raise AuthNoTokenException(f"Forbidden: {error}")
             raise AuthException(f"Forbidden: {error}")
-        if response.status_code == 204 or response.status_code == 205:
+        if response.status_code in (204, 205):
             return None
         if response.status_code == 404:
             error = parse_error_response(response)

tinybird/sql_template.py

@@ -385,9 +385,12 @@ def array_type(types):
                     x = default
                 else:
                     if _type and _type in types:
-                        x = ",".join(map(str, [types[_type](x) for _ in range(2)]))
+                        if _type == "String":
+                            x = ""
+                        else:
+                            x = ",".join(map(str, [types[_type](x) for _ in range(2)]))
                     else:
-                        x = ",".join([f"__no_value__{i}" for i in range(2)])
+                        x = ""
             elif x is None:
                 x = default
                 if x is None:
@@ -405,7 +408,7 @@ def array_type(types):
                         values.append(expression_wrapper(types[_type](t), str(t)))
                     else:
                         raise SQLTemplateException(
-                            f"Error validating {x}[{i}]({t}) to type {_type}",
+                            f"Error validating [{x}][{i}] ({'empty string' if t == '' else t}) to type {_type}",
                             documentation="/cli/advanced-templates.html",
                         )
                 else:
@@ -2056,7 +2059,11 @@ def render_sql_template(
     >>> render_sql_template("SELECT {{ Array(embedding, 'Float32') }}", {'token':'testing', 'embedding': '1,2,3,4, null'})
     Traceback (most recent call last):
     ...
-    tinybird.sql_template.SQLTemplateException: Template Syntax Error: Error validating 1,2,3,4, null[4]( null) to type Float32
+    tinybird.sql_template.SQLTemplateException: Template Syntax Error: Error validating [1,2,3,4, null][4] ( null) to type Float32
+    >>> render_sql_template("SELECT {{ Array(embedding, 'Int32', '') }}", {'token':'testing', 'embedding': ''})
+    Traceback (most recent call last):
+    ...
+    tinybird.sql_template.SQLTemplateException: Template Syntax Error: Error validating [][0] (empty string) to type Int32
     >>> render_sql_template('{% if test %}SELECT 1{% else %} select 2 {% end %}')
     (' select 2 ', {}, [])
     >>> render_sql_template('{% if Int32(test, 1) %}SELECT 1{% else %} select 2 {% end %}')
@@ -2096,7 +2103,7 @@ def render_sql_template(
     >>> render_sql_template("SELECT {{Array(foo, 'Int32')}}", test_mode=True)
     ('SELECT [0,0]', {}, [])
     >>> render_sql_template("SELECT {{Array(foo)}}", test_mode=True)
-    ("SELECT ['__no_value__0','__no_value__1']", {}, [])
+    ("SELECT ['']", {}, [])
     >>> render_sql_template("{{max_threads(2)}} SELECT 1")
     ('-- max_threads 2\\n SELECT 1', {'max_threads': 2}, [])
     >>> render_sql_template("SELECT {{String(foo)}}", test_mode=True)
@@ -2183,6 +2190,14 @@ def render_sql_template(
     ("% SELECT ['pre','pro']", {}, [])
     >>> render_sql_template("% SELECT {{Array(click_selector, 'String', 'pre,pro')}}", {'click_selector': 'hi,hello'})
     ("% SELECT ['hi','hello']", {}, [])
+    >>> render_sql_template("% SELECT {{Array(click_selector, 'String', '')}}")
+    ("% SELECT ['']", {}, [])
+    >>> render_sql_template("SELECT {{ Array(embedding, 'Int32', '') }}")
+    Traceback (most recent call last):
+    ...
+    tinybird.sql_template.SQLTemplateException: Template Syntax Error: Error validating [][0] (empty string) to type Int32
+    >>> render_sql_template("% SELECT {{Array(click_selector, 'String', '')}}", test_mode=True)
+    ("% SELECT ['']", {}, [])
     >>> render_sql_template("% SELECT now() > {{DateTime64(variable, '2020-09-09 10:10:10.000')}}", {})
     ("% SELECT now() > '2020-09-09 10:10:10.000'", {}, [])
     >>> render_sql_template("% SELECT {% if defined(x) %} x, 1", {})
@@ -2241,6 +2256,16 @@ def render_sql_template(
     Traceback (most recent call last):
     ...
     tinybird.sql_template.SQLTemplateException: Template Syntax Error: Required parameter is not defined. Check the parameters test. Please provide a value or set a default value in the pipe code.
+    >>> render_sql_template("SELECT * FROM test WHERE {% for item in JSON(filters, '[{}]') %} {{item.get('operand')}} {% end %}")
+    Traceback (most recent call last):
+        ...
+    tinybird.sql_template.SQLTemplateException: Template Syntax Error: expression "item.get('operand')" evaluated to null
+    >>> render_sql_template("SELECT * FROM test WHERE {% for item in JSON(filters, '[{\\\"operand\\\":\\\"test\\\"}]') %} {{item.get('operand')}} {% end %}")
+    ("SELECT * FROM test WHERE  'test' ", {}, [])
+    >>> render_sql_template("SELECT * FROM test WHERE {% for item in JSON(filters, '[\\\"test\\\"]') %} {{item.get('operator')}} {% end %}")
+    Traceback (most recent call last):
+        ...
+    tinybird.sql_template.SQLTemplateException: Template Syntax Error: 'str' object has no attribute 'get'. Make sure you're using an object/dictionary where trying to use .get()
     """
     escape_split_to_array = ff_split_to_array_escape.get(False)
 
@@ -2308,6 +2333,14 @@ def render_sql_template(
     except SQLTemplateException as e:
         format_SQLTemplateException_message(e, vars_and_types=template_variables_with_types)
         raise
+    except AttributeError as e:
+        # This happens when trying to use `get` on a string or when the object is None
+        if "'str' object has no attribute 'get'" in str(e):
+            raise SQLTemplateException(
+                "'str' object has no attribute 'get'. Make sure you're using an object/dictionary where trying to use .get()",
+                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/cli/introduction.html'
 __author__ = 'Tinybird'
 __author_email__ = 'support@tinybird.co'
-__version__ = '0.0.1.dev69'
-__revision__ = 'f90d907'
+__version__ = '0.0.1.dev71'
+__revision__ = '4feda71'

tinybird/tb/modules/build.py

@@ -15,7 +15,7 @@ from tinybird.client import TinyB
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import push_data
 from tinybird.tb.modules.datafile.build import folder_build
-from tinybird.tb.modules.datafile.fixture import get_fixture_dir
+from tinybird.tb.modules.datafile.fixture import get_fixture_dir, persist_fixture
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.project import Project
 from tinybird.tb.modules.shell import Shell, print_table_formatted
@@ -210,6 +210,8 @@ def process(
     build_failed = False
     if file_changed and file_changed.endswith(".ndjson"):
         rebuild_fixture(project, tb_client, file_changed)
+    elif file_changed and file_changed.endswith(".sql"):
+        rebuild_fixture_sql(project, tb_client, file_changed)
     else:
         try:
             build_project(project, tb_client, file_changed)
@@ -247,3 +249,12 @@ def run_watch(
     )
     watcher_thread.start()
     shell.run()
+
+
+def rebuild_fixture_sql(project: Project, tb_client: TinyB, sql_file: str) -> None:
+    sql_path = Path(sql_file)
+    datasource_name = sql_path.stem
+    sql = sql_path.read_text()
+    result = asyncio.run(tb_client.query(f"{sql} FORMAT JSON"))
+    data = result.get("data", [])
+    persist_fixture(datasource_name, data, project.folder)

tinybird/tb/modules/cli.py

@@ -57,6 +57,7 @@ VERSION = f"{__cli__.__version__} (rev {__cli__.__revision__})"
 @click.option("--show-tokens", is_flag=True, default=False, help="Enable the output of tokens")
 @click.option("--cloud/--local", is_flag=True, default=False, help="Run against cloud or local")
 @click.option("--build", is_flag=True, default=False, help="Run against build mode")
+@click.option("--preview", is_flag=True, default=False, help="Run against new deployment")
 @click.version_option(version=VERSION)
 @click.pass_context
 @coro
@@ -68,6 +69,7 @@ async def cli(
     show_tokens: bool,
     cloud: bool,
     build: bool,
+    preview: bool,
 ) -> None:
     """
     Use `OBFUSCATE_REGEX_PATTERN` and `OBFUSCATE_PATTERN_SEPARATOR` environment variables to define a regex pattern and a separator (in case of a single string with multiple regex) to obfuscate secrets in the CLI output.
@@ -121,12 +123,13 @@ async def cli(
 
     logging.debug("debug enabled")
 
-    client = await create_ctx_client(ctx, config, cloud, build)
+    client = await create_ctx_client(ctx, config, cloud, build, preview)
 
     if client:
         ctx.ensure_object(dict)["client"] = client
 
     ctx.ensure_object(dict)["project"] = project
+    ctx.ensure_object(dict)["env"] = get_target_env(cloud, build)
 
 
 @cli.command(hidden=True)
@@ -385,7 +388,7 @@ def __unpatch_click_output():
     click.secho = __old_click_secho
 
 
-async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, build: bool):
+async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, build: bool, preview: bool):
     commands_without_ctx_client = ["auth", "check", "login", "local", "upgrade"]
     command = ctx.invoked_subcommand
     if command in commands_without_ctx_client:
@@ -393,7 +396,7 @@ async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, b
 
     commands_always_cloud = ["pull"]
     commands_always_build = ["build", "test", "dev"]
-    commands_always_local = ["create", "mock"]
+    commands_always_local = ["create"]
     if (
         (cloud or command in commands_always_cloud)
         and command not in commands_always_build
@@ -402,8 +405,16 @@ async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, b
         click.echo(
             FeedbackManager.gray(message=f"Running against Tinybird Cloud: Workspace {config.get('name', 'default')}")
         )
-        return _get_tb_client(config.get("token", None), config["host"])
+        return _get_tb_client(config.get("token", None), config["host"], preview=preview)
     build = command in commands_always_build or build
     if not build and command not in commands_always_local and command not in commands_always_build:
         click.echo(FeedbackManager.gray(message="Running against Tinybird Local\n"))
-    return await get_tinybird_local_client(config, build=build)
+    return await get_tinybird_local_client(config, build=build, preview=preview)
+
+
+def get_target_env(cloud: bool, build: bool) -> str:
+    if cloud:
+        return "cloud"
+    if build:
+        return "build"
+    return "local"

tinybird/tb/modules/common.py

@@ -375,9 +375,9 @@ def getenv_bool(key: str, default: bool) -> bool:
     return v.lower() == "true" or v == "1"
 
 
-def _get_tb_client(token: str, host: str) -> TinyB:
+def _get_tb_client(token: str, host: str, preview: bool = False) -> TinyB:
     disable_ssl: bool = getenv_bool("TB_DISABLE_SSL_CHECKS", False)
-    return TinyB(token, host, version=VERSION, disable_ssl_checks=disable_ssl, send_telemetry=True)
+    return TinyB(token, host, version=VERSION, disable_ssl_checks=disable_ssl, send_telemetry=True, preview=preview)
 
 
 def create_tb_client(ctx: Context) -> TinyB:
@@ -554,7 +554,7 @@ def format_host(host: str, subdomain: Optional[str] = None) -> str:
     if subdomain and not is_localhost:
         url_info = urlparse(host)
         current_subdomain = url_info.netloc.split(".")[0]
-        if current_subdomain == "api" or current_subdomain == "ui":
+        if current_subdomain in ("api", "ui"):
             host = host.replace(current_subdomain, subdomain)
     if "localhost" in host or is_localhost:
         host = f"http://{host}" if "http" not in host else host

tinybird/tb/modules/config.py

@@ -219,7 +219,7 @@ class CLIConfig:
             return CLIConfig.DEFAULTS["host"]
         return None
 
-    def get_client(self, token: Optional[str] = None, host: Optional[str] = None) -> tbc.TinyB:
+    def get_client(self, token: Optional[str] = None, host: Optional[str] = None, preview: bool = False) -> tbc.TinyB:
         """Returns a new TinyB client configured with:
 
         - token:
@@ -242,6 +242,7 @@ class CLIConfig:
             version=CURRENT_VERSION,
             disable_ssl_checks=FeatureFlags.ignore_ssl_errors(),
             send_telemetry=FeatureFlags.send_telemetry(),
+            preview=preview,
         )
 
     def get_user_client(self, host: Optional[str] = None) -> tbc.TinyB:

tinybird/tb/modules/datafile/fixture.py

@@ -8,6 +8,13 @@ def get_fixture_dir(folder: str) -> Path:
     return Path(folder) / "fixtures"
 
 
+def persist_fixture_sql(fixture_name: str, sql: str, folder: str) -> Path:
+    fixture_dir = get_fixture_dir(folder)
+    fixture_file = fixture_dir / f"{fixture_name}.sql"
+    fixture_file.write_text(sql)
+    return fixture_file
+
+
 def persist_fixture(fixture_name: str, data: Union[List[Dict[str, Any]], str], folder: str, format="ndjson") -> Path:
     fixture_dir = get_fixture_dir(folder)
     fixture_file = fixture_dir / f"{fixture_name}.{format}"

tinybird/tb/modules/datasource.py

@@ -5,6 +5,7 @@
 
 import asyncio
 import json
+import os
 import re
 from typing import Optional
 
@@ -23,8 +24,10 @@ from tinybird.tb.modules.common import (
     push_data,
 )
 from tinybird.tb.modules.datafile.common import get_name_version
+from tinybird.tb.modules.datafile.fixture import persist_fixture
 from tinybird.tb.modules.exceptions import CLIDatasourceException
 from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.project import Project
 
 
 @cli.group()
@@ -391,3 +394,63 @@ async def datasource_data(ctx: Context, datasource: str, limit: int):
         echo_safe_humanfriendly_tables_format_smart_table(
             data=[d.values() for d in res["data"]], column_names=res["data"][0].keys()
         )
+
+
+@datasource.command(name="export")
+@click.argument("datasource")
+@click.option(
+    "--format",
+    "format_",
+    type=click.Choice(["csv", "ndjson"], case_sensitive=False),
+    default="ndjson",
+    help="Output format (csv or ndjson)",
+)
+@click.option("--rows", type=int, default=100, help="Number of rows to export (default: 100)")
+@click.option("--where", type=str, default=None, help="Condition to filter data")
+@click.option("--target", type=str, help="Target file path (default: datasource_name.{format})")
+@click.pass_context
+@coro
+async def datasource_export(
+    ctx: Context,
+    datasource: str,
+    format_: str,
+    rows: int,
+    where: Optional[str],
+    target: Optional[str],
+):
+    """Export data from a datasource to a file in CSV or NDJSON format
+
+    Example usage:
+    - Export all rows as CSV: tb datasource export my_datasource
+    - Export 1000 rows as NDJSON: tb datasource export my_datasource --format ndjson --rows 1000
+    - Export to specific file: tb datasource export my_datasource --output ./data/export.csv
+    """
+    client: TinyB = ctx.ensure_object(dict)["client"]
+    project: Project = ctx.ensure_object(dict)["project"]
+
+    # Determine output filename if not provided
+    if not target:
+        target = f"{datasource}.{format_}"
+
+    # Build query with optional row limit
+    query = f"SELECT * FROM {datasource} WHERE {where or 1} LIMIT {rows}"
+
+    click.echo(FeedbackManager.highlight(message=f"\n» Exporting {datasource} to {target}"))
+
+    try:
+        if format_ == "csv":
+            query += " FORMAT CSVWithNames"
+        else:
+            query += " FORMAT JSONEachRow"
+
+        res = await client.query(query)
+
+        fixture_path = persist_fixture(datasource, res, project.folder)
+        file_size = os.path.getsize(fixture_path)
+
+        click.echo(
+            FeedbackManager.success(message=f"✓ Exported data to {target} ({humanfriendly.format_size(file_size)})")
+        )
+
+    except Exception as e:
+        raise CLIDatasourceException(FeedbackManager.error(message=str(e)))

tinybird/tb/modules/deployment.py

@@ -266,13 +266,13 @@ def deployment_rollback(ctx: click.Context, wait: bool) -> None:
 @click.option(
     "--wait/--no-wait",
     is_flag=True,
-    default=False,
+    default=True,
     help="Wait for deploy to finish. Disabled by default.",
 )
 @click.option(
     "--auto/--no-auto",
     is_flag=True,
-    default=False,
+    default=True,
     help="Auto-promote the deployment. Only works if --wait is enabled. Disabled by default.",
 )
 @click.option(
@@ -311,6 +311,7 @@ def create_deployment(
     }
     project: Project = ctx.ensure_object(dict)["project"]
     client = ctx.ensure_object(dict)["client"]
+    config = ctx.ensure_object(dict)["config"]
     TINYBIRD_API_URL = f"{client.host}/v1/deploy"
     TINYBIRD_API_KEY = client.token
 
@@ -357,8 +358,21 @@ def create_deployment(
         deploy_result = result.get("result")
         if deploy_result == "success":
             print_changes(result, project)
-            click.echo(FeedbackManager.success(message="Deployment submitted successfully"))
-            deployment = result.get("deployment")
+            deployment = result.get("deployment", {})
+            # We show the url in the case of region is public
+            if client.host == "https://api.europe-west2.gcp.tinybird.co":
+                click.echo(
+                    FeedbackManager.gray(message="Deployment URL: ")
+                    + FeedbackManager.info(
+                        message=f"https://cloud.tinybird.co/gcp/europe-west2/{config.get('name')}/deployments/{deployment.get('id')}"
+                    )
+                )
+
+            if wait:
+                click.echo(FeedbackManager.info(message="\n✓ Deployment submitted successfully"))
+            else:
+                click.echo(FeedbackManager.success(message="\n✓ Deployment submitted successfully"))
+
             feedback = deployment.get("feedback", [])
             for f in feedback:
                 click.echo(

tinybird/tb/modules/local_common.py

@@ -15,10 +15,10 @@ TB_LOCAL_PORT = int(os.getenv("TB_LOCAL_PORT", 80))
 TB_LOCAL_HOST = f"http://localhost:{TB_LOCAL_PORT}"
 
 
-async def get_tinybird_local_client(config_obj: Dict[str, Any], build: bool = False) -> TinyB:
+async def get_tinybird_local_client(config_obj: Dict[str, Any], build: bool = False, preview: bool = False) -> TinyB:
     """Get a Tinybird client connected to the local environment."""
     config = await get_tinybird_local_config(config_obj, build=build)
-    return config.get_client(host=TB_LOCAL_HOST)
+    return config.get_client(host=TB_LOCAL_HOST, preview=preview)
 
 
 async def get_tinybird_local_config(config_obj: Dict[str, Any], build: bool = False) -> CLIConfig:

tinybird/tb/modules/mock.py

@@ -1,6 +1,4 @@
 import glob
-import logging
-import os
 from pathlib import Path
 
 import click
@@ -8,9 +6,9 @@ import click
 from tinybird.client import TinyB
 from tinybird.prompts import mock_prompt
 from tinybird.tb.modules.cli import cli
-from tinybird.tb.modules.common import CLIException, check_user_token_with_client, coro
+from tinybird.tb.modules.common import CLIException, check_user_token_with_client, coro, push_data
 from tinybird.tb.modules.config import CLIConfig
-from tinybird.tb.modules.datafile.fixture import persist_fixture
+from tinybird.tb.modules.datafile.fixture import persist_fixture, persist_fixture_sql
 from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.llm_utils import extract_xml
@@ -41,11 +39,11 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str) -> N
     try:
         tb_client: TinyB = ctx.ensure_object(dict)["client"]
         project: Project = ctx.ensure_object(dict)["project"]
+        env = ctx.ensure_object(dict)["env"]
         datasource_path = Path(datasource)
         datasource_name = datasource
         folder = project.folder
         click.echo(FeedbackManager.highlight(message=f"\n» Creating fixture for {datasource_name}..."))
-
         if datasource_path.suffix == ".datasource":
             datasource_name = datasource_path.stem
         else:
@@ -56,14 +54,6 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str) -> N
         if not datasource_path.exists():
             raise CLIException(f"Datasource '{datasource_path.stem}' not found")
 
-        prompt_path = Path(folder) / "fixtures" / f"{datasource_name}.prompt"
-        if not prompt or prompt == "Use the datasource schema to generate sample data":
-            # load the prompt from the fixture.prompt file if it exists
-            if prompt_path.exists():
-                prompt = prompt_path.read_text()
-        else:
-            prompt_path.write_text(prompt)
-
         datasource_content = datasource_path.read_text()
         config = CLIConfig.get_project_config()
         user_client = config.get_client()
@@ -76,21 +66,39 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str) -> N
         except Exception:
             click.echo(FeedbackManager.error(message="This action requires authentication. Run 'tb login' first."))
             return
+
         llm = LLM(user_token=user_token, host=user_client.host)
         prompt = f"<datasource_schema>{datasource_content}</datasource_schema>\n<user_input>{prompt}</user_input>"
         sql = ""
-
         response = llm.ask(system_prompt=mock_prompt(rows), prompt=prompt)
         sql = extract_xml(response, "sql")
         result = await tb_client.query(f"{sql} FORMAT JSON")
         data = result.get("data", [])[:rows]
-        persist_fixture(datasource_name, data, folder)
-        click.echo(FeedbackManager.info(message=f"✓ /fixtures/{datasource_name}.ndjson created"))
+        if env == "build":
+            persist_fixture_sql(datasource_name, sql, folder)
+
+        fixture_path = persist_fixture(datasource_name, data, folder)
 
-        if os.environ.get("TB_DEBUG", "") != "":
-            logging.debug(sql)
+        click.echo(FeedbackManager.info(message=f"✓ /fixtures/{datasource_name}.ndjson created"))
+        if env == "cloud":
+            await append_fixture(tb_client, datasource_name, str(fixture_path))
 
         click.echo(FeedbackManager.success(message=f"✓ Sample data for {datasource_name} created with {rows} rows"))
 
     except Exception as e:
         click.echo(FeedbackManager.error_exception(error=f"Error: {e}"))
+
+
+async def append_fixture(
+    tb_client: TinyB,
+    datasource_name: str,
+    url: str,
+):
+    await push_data(
+        tb_client,
+        datasource_name,
+        url,
+        mode="append",
+        concurrency=1,
+        silent=True,
+    )

tinybird/tb/modules/telemetry.py

@@ -206,7 +206,7 @@ class TelemetryHelper:
 
             self.log(f"Received status {r.status_code}: {r.text}")
 
-            if r.status_code == 200 or r.status_code == 202:
+            if r.status_code in (200, 202):
                 self.log(f"Successfully sent {len(events)} events to {self.tb_host}")
                 self.events.clear()
                 return

tinybird/tb/modules/test.py

@@ -5,6 +5,7 @@
 
 import difflib
 import glob
+import sys
 import urllib.parse
 from pathlib import Path
 from typing import Any, Dict, List, Optional, Tuple
@@ -265,7 +266,7 @@ async def run_tests(ctx: click.Context, name: Tuple[str, ...]) -> None:
 
     if failed_tests_count:
         click.echo(FeedbackManager.error(message=f"\n✗ {test_count - failed_tests_count}/{test_count} passed"))
-        exit(1)
+        sys.exit(1)
     else:
         click.echo(FeedbackManager.success(message=f"\n✓ {test_count}/{test_count} passed"))
 

tinybird/tb/modules/tinyunit/tinyunit.py

@@ -190,8 +190,6 @@ def generate_file(file: str, overwrite: bool = False) -> None:
     else:
         raise CLIException(FeedbackManager.error_file_already_exists(file=p))
 
-    return
-
 
 async def run_test_file(tb_client: TinyB, file: str) -> List[TestResult]:
     results: List[TestResult] = []

tinybird/tb/modules/watch.py

@@ -29,7 +29,7 @@ class WatchProjectHandler(FileSystemEventHandler):
         if event.is_directory:
             return None
 
-        valid_extensions = [".datasource", ".pipe", ".ndjson"]
+        valid_extensions = [".datasource", ".pipe", ".ndjson", ".sql"]
 
         if not any(event.src_path.endswith(ext) for ext in valid_extensions):
             return None

tinybird/tb_cli_modules/common.py

@@ -600,7 +600,7 @@ def format_host(host: str, subdomain: Optional[str] = None) -> str:
     if subdomain and not is_localhost:
         url_info = urlparse(host)
         current_subdomain = url_info.netloc.split(".")[0]
-        if current_subdomain == "api" or current_subdomain == "ui":
+        if current_subdomain in ("api", "ui"):
             host = host.replace(current_subdomain, subdomain)
     if "localhost" in host or is_localhost:
         host = f"http://{host}" if "http" not in host else host

tinybird/tb_cli_modules/telemetry.py

@@ -206,7 +206,7 @@ class TelemetryHelper:
 
             self.log(f"Received status {r.status_code}: {r.text}")
 
-            if r.status_code == 200 or r.status_code == 202:
+            if r.status_code in (200, 202):
                 self.log(f"Successfully sent {len(events)} events to {self.tb_host}")
                 self.events.clear()
                 return

tinybird/tornado_template.py

@@ -1144,7 +1144,7 @@ def check_valid_expr(expr):
             return
         if isinstance(expr.value, str):
             return
-        if isinstance(expr.value, type(None)):
+        if expr.value is None:
             return
         raise SecurityException(f"Invalid Constant: {ast.dump(expr)}")
     elif isinstance(expr, ast.Name):

{tinybird-0.0.1.dev69.dist-info → tinybird-0.0.1.dev71.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.2
 Name: tinybird
-Version: 0.0.1.dev69
+Version: 0.0.1.dev71
 Summary: Tinybird Command Line Tool
 Home-page: https://www.tinybird.co/docs/cli/introduction.html
 Author: Tinybird

{tinybird-0.0.1.dev69.dist-info → tinybird-0.0.1.dev71.dist-info}/RECORD

@@ -1,5 +1,5 @@
 tinybird/__cli__.py,sha256=esPl5QDTzuQgHe5FuxWLm-fURFigGGwjnYLh9GuWUw4,232
-tinybird/client.py,sha256=W5Xttnz0bzwqqVGNJJs-4Ca2AbtCMh3URuQYPJ5APsE,52013
+tinybird/client.py,sha256=ewNqHPN4mM54-Z_QHHRIYf7J80XRAg92CA7mutHM4sI,52153
 tinybird/config.py,sha256=cd_RH7ZjqGjpWwu0efPkhS8VxD9K6Jix4QY2W3w1Pvs,5811
 tinybird/connectors.py,sha256=7Gjms7b5MAaBFGi3xytsJurCylprONpFcYrzp4Fw2Rc,15241
 tinybird/context.py,sha256=VaMhyHruH-uyMypPDfxtuo4scS18b7rxCCdeUVm6ysg,1301
@@ -8,25 +8,25 @@ tinybird/feedback_manager.py,sha256=ADuy1yC3fCFM_ii_Haphg8Gdzapa5pX9dlRTVbuUqxs,
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
 tinybird/prompts.py,sha256=LQfE0ruY9A7z1iqtt6xTkW-1NlHbXl7LbKqzfdOsPxA,30612
 tinybird/sql.py,sha256=LBi74GxhNAYTb6m2-KNGpAkguSKh7rcvBbERbE7nalA,46195
-tinybird/sql_template.py,sha256=zUK-fsrvKISNsTquEFL40iD6-fMoL0RF3vxHn4PZYvw,93500
+tinybird/sql_template.py,sha256=hHz-EbHWlWVMS4e6hTpoieJ1qRAzzZLDq9-fnjT60dw,95732
 tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
 tinybird/sql_toolset.py,sha256=32SNvxRFKQYWTvYPMJ_u3ukcd1hKZyEqx8T2cv2412w,14697
 tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
-tinybird/tornado_template.py,sha256=fUKIFrZLe0WyMFmV-mkjIdCSeFcZMBL3Uet4eHcids0,41960
+tinybird/tornado_template.py,sha256=KmW_VD7y-NVqrc8YZKwIaxoJB0XpCcB2izdmxmtmApM,41944
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=AUAww-KjGOZg9h0IBlKA3FeacJYB4rOtqcTGJhFM-g8,40392
-tinybird/tb/__cli__.py,sha256=oDKDPPvR2NMgaawv4tiYkQ17clB9vn1VDY4J4rqpHxI,251
+tinybird/tb/__cli__.py,sha256=VczMDcdJLOacMcmU2_4nzvVgYOpRHt6Z-v07TP3wI38,251
 tinybird/tb/cli.py,sha256=FD1pfbzu9YHJHEG6Vtn_EwPLTYhwqw-I6AxXeTaRHU8,926
 tinybird/tb/modules/auth.py,sha256=vBA-KsrjAp77kFunGSM-4o7AFdfO7ac4dnrHKrx0alI,9020
-tinybird/tb/modules/build.py,sha256=UN1d7EZ93VOlPCrtsay-KLgZnzxn2NCBDY3wvrUSP1Q,9198
+tinybird/tb/modules/build.py,sha256=GIXpIAHxulB-PgJ2jeZnwIXPbL7anp3rPi5a6DxfWuQ,9674
 tinybird/tb/modules/cicd.py,sha256=xxXwy-QekJcG14kkJeGNl7LkHduhZXfvBZE8WrU6-t4,5351
-tinybird/tb/modules/cli.py,sha256=AvqH7x3wiLShaXWxIsMRidPCmJs03ZYwuFzQZz-nMyU,15919
-tinybird/tb/modules/common.py,sha256=XZY0nWkK68vgh2Vml5JzyQuEWIyf2igq7IrAu0n94Lo,73203
-tinybird/tb/modules/config.py,sha256=mie3oMVTf5YOUFEiLs88P16U4LkJafJjSpjwyAkFHog,10979
+tinybird/tb/modules/cli.py,sha256=r0_HNx2hMlbIOmmJjhflF_lAUHNhmBxuVR4rHx2onOY,16293
+tinybird/tb/modules/common.py,sha256=CFERBZukoUEKzRON9hvLqCyTpPucKIyaVgHsKR_vQQM,73222
+tinybird/tb/modules/config.py,sha256=J305DWh7zKpQZFjvI6x-YvK7F4RPRxyTIiZ8quc1h-0,11031
 tinybird/tb/modules/copy.py,sha256=Aq6wh_wjRiyLQtEOKF9pKLPgJhSvbGTFWIw_LJB0t0U,5801
 tinybird/tb/modules/create.py,sha256=I01JDENOyGKK0Umd2_1Om_nFGP8Uk9vxaOw7PygK02o,12302
-tinybird/tb/modules/datasource.py,sha256=9NxmicauGwLwfcKAe3Vyiu5JHmYDl6iYmBJweu4Hl2s,14735
-tinybird/tb/modules/deployment.py,sha256=Xtaty4vPoSFk189cuupei3h3ze1FpPvi_dA7koPvjoI,16222
+tinybird/tb/modules/datasource.py,sha256=aOvjTVH9EdGNAw_lqhtZdJQhYnMOTGIqD3Pe4GtakFw,16905
+tinybird/tb/modules/deployment.py,sha256=qGkFwPR1TGtcbqUcKkhsfzK--sNzXo6JV8Ny8HrlSfo,16873
 tinybird/tb/modules/endpoint.py,sha256=zQJgJXTzMB3hPO-3Xnppi0Pv8byZunBNhbbmMgKtijE,11915
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=mrw5tdYycfvg6WLXlM0KIjfJardm_aNpnJkUg2vH0cA,68463
@@ -35,20 +35,20 @@ tinybird/tb/modules/job.py,sha256=956Pj8BEEsiD2GZsV9RKKVM3I_CveOLgS82lykO5ukk,29
 tinybird/tb/modules/llm.py,sha256=AC0VSphTOM2t-v1_3NLvNN_FIbgMo4dTyMqIv5nniPo,835
 tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCfxvg,3469
 tinybird/tb/modules/local.py,sha256=_PIa-1M-72bv9rhLwqaNthJM1ZhvcjWXFChZAfEPXRs,5658
-tinybird/tb/modules/local_common.py,sha256=W1fEnB1vBQ4YC5U1PdA0w0g3cTV78bQ5R-lRxdDj5-Y,2868
+tinybird/tb/modules/local_common.py,sha256=9ON3do1oiAmulWXjAmn9ep0coMNbtJbcSRhZ6XQk6ZQ,2908
 tinybird/tb/modules/login.py,sha256=EGxwVRmMX1Y7ZeCRyA8fqaCWpYYk7NvnZ3x_1g0NlYA,6063
 tinybird/tb/modules/materialization.py,sha256=dESybok66Fn7XLzQQr-fBCDNf5xQL8wkRiD2QBr5lOg,5748
-tinybird/tb/modules/mock.py,sha256=DVZjPv9p8k5fntHcLHnIsWS7KBa1jF02BYA_Ey4XWf4,3824
+tinybird/tb/modules/mock.py,sha256=7alBqLfQLb7JSZJth9JDXGUOvABWXr8Gx23edgyrW8w,3869
 tinybird/tb/modules/pipe.py,sha256=PbCM31q9lVB9xPWvPBhlpW53MGFGvzVv81trxC06kDY,2428
 tinybird/tb/modules/project.py,sha256=QdOG65Hcc6r_eQ938CfZeIXyp38RkiTYNPupgqy2gP0,2948
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb/modules/shell.py,sha256=qWpZ9TwDSDq3YsBxueNtezsslp3N7pwUPRF8HIlTjOo,13828
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/tag.py,sha256=anPmMUBc-TbFovlpFi8GPkKA18y7Y0GczMsMms5TZsU,3502
-tinybird/tb/modules/telemetry.py,sha256=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
-tinybird/tb/modules/test.py,sha256=tY7_FJRSL6EdztypWlvDJi5QpXDLS5dpO0i_dpGuzkI,11528
+tinybird/tb/modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
+tinybird/tb/modules/test.py,sha256=4nB11SXv3HC33HaORdg1Erf5GXV2Teyedgs9u750qfM,11543
 tinybird/tb/modules/token.py,sha256=WEB2xqRgOierJ_S7TrZLEpLpQ9mkXcYLH2-xipjoGDc,12814
-tinybird/tb/modules/watch.py,sha256=90FUrSD6cYfOi7GsEHfq_oTI6kap7uc8CbG0t3j_Tus,4953
+tinybird/tb/modules/watch.py,sha256=IIahyfmVNNu_EY2dVKZ_zjdSEFocwsmpvvgki4EmU-w,4961
 tinybird/tb/modules/workspace.py,sha256=7gOgt10j9emleqQ43BA6jU0CZjJxw7L5gDZhCxGafI4,6400
 tinybird/tb/modules/workspace_members.py,sha256=Vb5XEaKmkfONyfg2MS5EcpwolMvv7GLwFS5m2EuobT8,8726
 tinybird/tb/modules/datafile/build.py,sha256=seGFSvmgyRrAM1-icsKBkuog3WccfGUYFTPT-xoA5W8,50940
@@ -58,7 +58,7 @@ tinybird/tb/modules/datafile/build_pipe.py,sha256=Jgv3YKIvMfjPiSIdw1k2mpaoDdAWMi
 tinybird/tb/modules/datafile/common.py,sha256=1sKkQnyKOaFARQzORncpL6cRVpV9GDOD4oC36daX-Hk,79343
 tinybird/tb/modules/datafile/diff.py,sha256=-0J7PsBO64T7LOZSkZ4ZFHHCPvT7cKItnJkbz2PkndU,6754
 tinybird/tb/modules/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
-tinybird/tb/modules/datafile/fixture.py,sha256=XQpbppUQ4cow7dMzHs7OZBYBXA2PI5AOCmNdMDClbKo,833
+tinybird/tb/modules/datafile/fixture.py,sha256=si-9LB-LdKQSWDtVW82xDrHtFfko5bgBG1cvjqqrcPU,1064
 tinybird/tb/modules/datafile/format_common.py,sha256=WaNV4tXrQU5gjV6MJP-5TGqg_Bre6ilNS8emvFl-X3c,1967
 tinybird/tb/modules/datafile/format_datasource.py,sha256=gpRsGnDEMxEo0pIlEHXKvyuwKIpqJJUCN9JRSiDYs_4,6156
 tinybird/tb/modules/datafile/format_pipe.py,sha256=58iSTrJ5lg-IsbpX8TQumQTuZ6UIotMsCIkNJd1M-pM,7418
@@ -66,16 +66,16 @@ tinybird/tb/modules/datafile/parse_datasource.py,sha256=kk35PzesoJOd0LKjYp4kOyCw
 tinybird/tb/modules/datafile/parse_pipe.py,sha256=snoy8Ac_Sat7LIXLAKzxjJSl2-TKg9FaZTooxrx6muE,3420
 tinybird/tb/modules/datafile/pipe_checker.py,sha256=LnDLGIHLJ3N7qHb2ptEbPr8CoczNfGwpjOY8EMdxfHQ,24649
 tinybird/tb/modules/datafile/pull.py,sha256=vcjMUbjnZ9XQMGmL33J3ElpbXBTat8Yzp-haeDggZd4,5967
-tinybird/tb/modules/tinyunit/tinyunit.py,sha256=LZGrsvIAUy5O2bZtGbi9O80QGIfXe_Du8c0PDNxeQcc,11727
+tinybird/tb/modules/tinyunit/tinyunit.py,sha256=GlDgEXc6TDO3ODxgfATAL2fvbKy-b_CzqoeDrApRm0g,11715
 tinybird/tb/modules/tinyunit/tinyunit_lib.py,sha256=hGh1ZaXC1af7rKnX7222urkj0QJMhMWclqMy59dOqwE,1922
 tinybird/tb_cli_modules/cicd.py,sha256=0lMkb6CVOFZl5HOwgY8mK4T4mgI7O8335UngLXtCc-c,13851
-tinybird/tb_cli_modules/common.py,sha256=NpR56fwwj6u--gLdTIZodvMe4LsWAtnphX-kw4hoKGs,78726
+tinybird/tb_cli_modules/common.py,sha256=VZA_NQA4ct20TQ9DxOt1065Mt4YBt10xPRVVYkoBytY,78705
 tinybird/tb_cli_modules/config.py,sha256=6u6B5QCdiQLbJkCkwtnKGs9H3nP-KXXhC75mF7B-1DQ,11464
 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=iEGnMuCuNhvF6ln__j6X9MSTwL_0Hm-GgFHHHvhfknk,10466
-tinybird-0.0.1.dev69.dist-info/METADATA,sha256=Qy1Ob_W4u71B64I99GYQSlFunPcsP6kz7ie9WBl1I5I,2585
-tinybird-0.0.1.dev69.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev69.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev69.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev69.dist-info/RECORD,,
+tinybird/tb_cli_modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
+tinybird-0.0.1.dev71.dist-info/METADATA,sha256=ACRsG8C4j9Jdl_zf65waD4t88jgCMrIUcktNq7mhvNM,2585
+tinybird-0.0.1.dev71.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev71.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev71.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev71.dist-info/RECORD,,