tinybird 0.0.1.dev91__py3-none-any.whl → 0.0.1.dev93__py3-none-any.whl

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.dev91'
-__revision__ = 'b965ddf'
+__version__ = '0.0.1.dev93'
+__revision__ = '8ee6a21'

tinybird/tb/modules/deployment.py

@@ -4,7 +4,7 @@ import sys
 import time
 from datetime import datetime
 from pathlib import Path
-from typing import Optional, Union
+from typing import Any, Dict, Optional, Union
 
 import click
 import requests
@@ -330,7 +330,7 @@ def create_deployment(
     }
     project: Project = ctx.ensure_object(dict)["project"]
     client = ctx.ensure_object(dict)["client"]
-    config = ctx.ensure_object(dict)["config"]
+    config: Dict[str, Any] = ctx.ensure_object(dict)["config"]
     TINYBIRD_API_URL = f"{client.host}/v1/deploy"
     TINYBIRD_API_KEY = client.token
 
@@ -460,18 +460,27 @@ def print_changes(result: dict, project: Project) -> None:
     for p in deployment.get("new_pipe_names", []):
         resources.append(["new", p, project.get_resource_path(p, "pipe")])
 
+    for dc in deployment.get("new_data_connector_names", []):
+        resources.append(["new", dc, project.get_resource_path(dc, "data_connector")])
+
     for ds in deployment.get("changed_datasource_names", []):
         resources.append(["modified", ds, project.get_resource_path(ds, "datasource")])
 
     for p in deployment.get("changed_pipe_names", []):
         resources.append(["modified", p, project.get_resource_path(p, "pipe")])
 
+    for dc in deployment.get("changed_data_connector_names", []):
+        resources.append(["modified", dc, project.get_resource_path(dc, "data_connector")])
+
     for ds in deployment.get("deleted_datasource_names", []):
         resources.append(["deleted", ds, project.get_resource_path(ds, "datasource")])
 
     for p in deployment.get("deleted_pipe_names", []):
         resources.append(["deleted", p, project.get_resource_path(p, "pipe")])
 
+    for dc in deployment.get("deleted_data_connector_names", []):
+        resources.append(["deleted", dc, project.get_resource_path(dc, "data_connector")])
+
     if resources:
         click.echo(FeedbackManager.highlight(message="\n» Changes to be deployed...\n"))
         echo_safe_humanfriendly_tables_format_smart_table(resources, column_names=columns)

tinybird/tb/modules/playground.py

@@ -62,7 +62,7 @@ def playground(
     """Build the project in Tinybird Local."""
     project: Project = ctx.ensure_object(dict)["project"]
     tb_client: TinyB = ctx.ensure_object(dict)["client"]
-    config: CLIConfig = ctx.ensure_object(dict)["config"]
+    config = CLIConfig.get_project_config()
     context.disable_template_security_validation.set(True)
 
     async def process(filenames: List[str], watch: bool = False):

tinybird/tb/modules/test.py

@@ -5,6 +5,7 @@
 
 import difflib
 import glob
+import os
 import sys
 import urllib.parse
 from pathlib import Path
@@ -75,8 +76,8 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
     """
     project: Project = ctx.ensure_object(dict)["project"]
     client: TinyB = ctx.ensure_object(dict)["client"]
-    config: CLIConfig = ctx.ensure_object(dict)["config"]
-    root_path = Path(project.folder)
+    config = CLIConfig.get_project_config()
+    root_path = project.path
     folder = project.folder
     try:
         if ".pipe" in name_or_filename:
@@ -85,20 +86,17 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
         else:
             pipe_folders = ("endpoints", "copies", "materializations", "sinks", "pipes")
-            try:
-                pipe_path = next(
-                    root_path / folder / f"{name_or_filename}.pipe"
-                    for folder in pipe_folders
-                    if (root_path / folder / f"{name_or_filename}.pipe").exists()
-                )
-            except Exception:
+            pipe_path = next(
+                root_path / folder / f"{name_or_filename}.pipe"
+                for folder in pipe_folders
+                if os.path.exists(root_path / folder / f"{name_or_filename}.pipe")
+            )
+            if not pipe_path:
                 raise CLIException(FeedbackManager.error(message=f"Pipe {name_or_filename} not found"))
 
         pipe_name = pipe_path.stem
         click.echo(FeedbackManager.highlight(message=f"\n» Creating tests for {pipe_name} endpoint..."))
-        pipe_path = root_path / pipe_path
         pipe_content = pipe_path.read_text()
-
         pipe = await client._req(f"/v0/pipes/{pipe_name}")
         parameters = set([param["name"] for node in pipe["nodes"] for param in node["params"]])
 
@@ -112,7 +110,6 @@ async def test_create(ctx: click.Context, name_or_filename: str, prompt: str) ->
             raise CLIException(FeedbackManager.error(message="No user token found"))
 
         llm = LLM(user_token=user_token, host=config.get_client().host)
-        click.echo(FeedbackManager.highlight(message=f"\n» Creating test for {pipe_name} endpoint..."))
         response_llm = llm.ask(system_prompt=system_prompt, prompt=prompt)
         response_xml = extract_xml(response_llm, "response")
         tests_content = parse_xml(response_xml, "test")
@@ -216,9 +213,12 @@ async def test_update(ctx: click.Context, pipe: str) -> None:
 async def run_tests(ctx: click.Context, name: Tuple[str, ...]) -> None:
     click.echo(FeedbackManager.highlight(message="\n» Running tests"))
     client: TinyB = ctx.ensure_object(dict)["client"]
+    project: Project = ctx.ensure_object(dict)["project"]
     paths = [Path(n) for n in name]
-    endpoints = [f"./tests/{p.stem}.yaml" for p in paths]
-    test_files: List[str] = endpoints if len(endpoints) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+    endpoints = [f"{project.path}/tests/{p.stem}.yaml" for p in paths]
+    test_files: List[str] = (
+        endpoints if len(endpoints) > 0 else glob.glob(f"{project.path}/tests/**/*.y*ml", recursive=True)
+    )
 
     async def run_test(test_file):
         test_file_path = Path(test_file)

{tinybird-0.0.1.dev91.dist-info → tinybird-0.0.1.dev93.dist-info}/METADATA

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

{tinybird-0.0.1.dev91.dist-info → tinybird-0.0.1.dev93.dist-info}/RECORD

@@ -15,7 +15,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=T_ssIbV6pifxlCyF-t9ua_F2d_5mle0hocUCAZm8PNY,251
+tinybird/tb/__cli__.py,sha256=bNoBpW1G6tYs3EKrC-mlXTGvxaOMRSrxlyybi-I06Y0,251
 tinybird/tb/cli.py,sha256=SSXALqjtPShcQbgvT1u7up2YL0ls36wXABX0hZwebdQ,1070
 tinybird/tb/modules/auth.py,sha256=L1IatO2arRSzys3t8px8xVt8uPWUL5EVD0sFzAV_uVU,9022
 tinybird/tb/modules/build.py,sha256=-lRGBxKtuipmyl3pmiGcfp67fH1Ed-COfHAZKdgLIWo,10483
@@ -27,7 +27,7 @@ tinybird/tb/modules/connection.py,sha256=WKeDxbTpSsQ1PUmsT730g3S5RT2PtR5mPpVEanD
 tinybird/tb/modules/copy.py,sha256=MAVqKip8_QhOYq99U_XuqSO6hCLJEh5sFtbhcXtI3SI,5802
 tinybird/tb/modules/create.py,sha256=KjotVfIQLfcPyQBykTHnPLn4ikrm6qqeMcbRE1d-6Jo,13280
 tinybird/tb/modules/datasource.py,sha256=dNCK9iCR2xPLfwqqwg2ixyE6NuoVEiJU2mBZBmOYrVY,16906
-tinybird/tb/modules/deployment.py,sha256=oSRnQnz2Wkvp1y0yt_tVa9g0aCYyFxNAsUyxoZTIcpY,17615
+tinybird/tb/modules/deployment.py,sha256=ZX9fTxenVN0oCbJTa_5Ap1bmyqT8zt9PMBmBP949Ag4,18109
 tinybird/tb/modules/endpoint.py,sha256=EhVoGAXsFz-83Fiwj1gI-I73iRRvL49d0W81un7hvPE,12080
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=7nNiOx7OMebiheLED1r0d75SbuXCNxyBmF4e20rCBNc,69511
@@ -42,7 +42,7 @@ tinybird/tb/modules/logout.py,sha256=ULooy1cDBD02-r7voZmhV7udA0ML5tVuflJyShrh56Y
 tinybird/tb/modules/materialization.py,sha256=r8Q9HXcYEmfrEzP4WpiasCKDJdSkTPaAKJtZMoJKhi8,5749
 tinybird/tb/modules/mock.py,sha256=3q4i6CXKcS-zsgevbN_zpAP4AnB9_WIVxmVSJV3FNPQ,3881
 tinybird/tb/modules/pipe.py,sha256=gcLz0qHgwKDLsWFY3yFLO9a0ETAV1dFbI8YeLHi9460,2429
-tinybird/tb/modules/playground.py,sha256=CQaz2JqFDdReK2fJY1yZsSwiSY24_jeTb9PKw1WUigA,4848
+tinybird/tb/modules/playground.py,sha256=bN0whphoSO6p1_u3b6OAUoc3ieG5Cl3qNXwt2HcUOp8,4834
 tinybird/tb/modules/project.py,sha256=ei0TIAuRksdV2g2FJqByuV4DPyivQGrZ42z_eQDNBgI,2963
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb/modules/secret.py,sha256=xxzfKxfFN7GORib1WslCaFDHt_dgnjmfOewyptPU_VM,2820
@@ -50,7 +50,7 @@ tinybird/tb/modules/shell.py,sha256=a98W4L4gfrmxEyybtu6S4ENXrBYtgNASB5e_evuXQvI,
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
 tinybird/tb/modules/tag.py,sha256=anPmMUBc-TbFovlpFi8GPkKA18y7Y0GczMsMms5TZsU,3502
 tinybird/tb/modules/telemetry.py,sha256=Hh2Io8ZPROSunbOLuMvuIFU4TqwWPmQTqal4WS09K1A,10449
-tinybird/tb/modules/test.py,sha256=oa6_ApTCT526W8-hngpZwNyl2AbIpOztOsxf0iwlSwc,11547
+tinybird/tb/modules/test.py,sha256=FUU-drY8mdjNoKsw16O5ZqvYvZqzycrZBEpSwbhGDUE,11456
 tinybird/tb/modules/token.py,sha256=OhqLFpCHVfYeBCxJ0n7n2qoho9E9eGcUfHgL7R1MUVQ,13485
 tinybird/tb/modules/watch.py,sha256=qMQhewRSso1AFSEFLuyeyGFA8Lxf9ccYJxmVdPU1BgM,8808
 tinybird/tb/modules/workspace.py,sha256=SYkEULv_Gg8FhnAnZspengzyT5N4w0wjsvWWZ3vy3Ho,7753
@@ -79,8 +79,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.dev91.dist-info/METADATA,sha256=iN68_RO9ZH5bntib_gq07U3hcPtPv206bagyLYciECw,2585
-tinybird-0.0.1.dev91.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev91.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev91.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev91.dist-info/RECORD,,
+tinybird-0.0.1.dev93.dist-info/METADATA,sha256=IVb38UDMgY5n1eKji5lTeEKlcIytmZ6bi8GmcYVd3Y0,2585
+tinybird-0.0.1.dev93.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev93.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev93.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev93.dist-info/RECORD,,