tinybird 0.0.1.dev63__py3-none-any.whl → 0.0.1.dev65__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.dev63'
-__revision__ = '4d0f19d'
+__version__ = '0.0.1.dev65'
+__revision__ = 'bb7c13c'

tinybird/tb/modules/build.py

@@ -4,8 +4,9 @@ import logging
 import sys
 import threading
 import time
+from functools import partial
 from pathlib import Path
-from typing import Optional
+from typing import Callable, Optional
 
 import click
 import requests
@@ -31,9 +32,13 @@ def build(ctx: click.Context, watch: bool) -> None:
     project: Project = ctx.ensure_object(dict)["project"]
     tb_client: TinyB = ctx.ensure_object(dict)["client"]
     click.echo(FeedbackManager.highlight_building_project())
-    process(project=project, tb_client=tb_client, watch=watch)
+    process(project=project, tb_client=tb_client, watch=False)
     if watch:
-        run_watch(project=project, tb_client=tb_client)
+        run_watch(
+            project=project,
+            tb_client=tb_client,
+            process=partial(process, project=project, tb_client=tb_client, watch=True),
+        )
 
 
 @cli.command("dev", help="Build the project server side and watch for changes")
@@ -43,7 +48,7 @@ def dev(ctx: click.Context) -> None:
     tb_client: TinyB = ctx.ensure_object(dict)["client"]
     click.echo(FeedbackManager.highlight_building_project())
     process(project=project, tb_client=tb_client, watch=True)
-    run_watch(project=project, tb_client=tb_client)
+    run_watch(project=project, tb_client=tb_client, process=partial(process, project=project, tb_client=tb_client))
 
 
 def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str] = None) -> None:
@@ -55,7 +60,7 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
     TINYBIRD_API_URL = tb_client.host + "/v1/build"
     logging.debug(TINYBIRD_API_URL)
     TINYBIRD_API_KEY = tb_client.token
-    error = False
+    error: Optional[str] = None
     try:
         files = [
             ("context://", ("cli-version", "1.0.0", "text/plain")),
@@ -129,22 +134,19 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
 
         elif build_result == "failed":
             build_errors = result.get("errors")
-            error = True
             for build_error in build_errors:
                 filename_bit = f"{build_error.get('filename', '')}"
                 error_msg = ((filename_bit + "\n\n") if filename_bit else "") + build_error.get("error")
-                click.echo(FeedbackManager.error(message=error_msg))
+                error = error_msg
         else:
-            error = True
-            click.echo(FeedbackManager.error(message=f"Unknown build result. Error: {result.get('error')}"))
-
+            error = f"Unknown build result. Error: {result.get('error')}"
     except Exception as e:
-        click.echo(FeedbackManager.error_exception(error="Error: " + str(e)))
+        error = str(e)
     finally:
         for fd in fds:
             fd.close()
         if error:
-            raise click.ClickException(FeedbackManager.error(message="Error: Build failed"))
+            raise click.ClickException(error)
 
 
 def append_fixture(
@@ -212,10 +214,10 @@ def process(
         try:
             build_project(project, tb_client, file_changed)
         except click.ClickException as e:
-            click.echo(e)
+            click.echo(FeedbackManager.error(message=e))
             build_failed = True
         try:
-            if file_changed:
+            if file_changed and not build_failed:
                 asyncio.run(folder_build(project, tb_client, filenames=[file_changed]))
                 show_data(tb_client, file_changed, diff)
         except Exception:
@@ -224,7 +226,7 @@ def process(
     time_end = time.time()
     elapsed_time = time_end - time_start
 
-    rebuild_str = "Rebuild" if watch else "Build"
+    rebuild_str = "Rebuild" if watch and file_changed else "Build"
     if build_failed:
         click.echo(FeedbackManager.error(message=f"\n✗ {rebuild_str} failed"))
         if not watch:
@@ -233,7 +235,9 @@ def process(
         click.echo(FeedbackManager.success(message=f"\n✓ {rebuild_str} completed in {elapsed_time:.1f}s"))
 
 
-def run_watch(project: Project, tb_client: TinyB) -> None:
+def run_watch(
+    project: Project, tb_client: TinyB, process: Callable[[bool, Optional[str], Optional[str]], None]
+) -> None:
     shell = Shell(project=project, tb_client=tb_client)
     click.echo(FeedbackManager.gray(message="\nWatching for changes..."))
     watcher_thread = threading.Thread(

tinybird/tb/modules/endpoint.py

@@ -269,7 +269,7 @@ async def endpoint_stats(ctx: click.Context, pipes: Tuple[str, ...], format_: st
             echo_safe_humanfriendly_tables_format_smart_table(table_human_readable, column_names=columns)
 
 
-def build_endpoint_snippet(tb_client: TinyB, pipe_name: str, token: Optional[str], language: str) -> Optional[str]:
+def build_endpoint_snippet(tb_client: TinyB, pipe_name: str, token: str, language: str) -> Optional[str]:
     endpoint_url = build_endpoint_url(tb_client, pipe_name, token)
     if language == "http":
         return endpoint_url
@@ -282,9 +282,9 @@ def build_endpoint_snippet(tb_client: TinyB, pipe_name: str, token: Optional[str
     elif language == "javascript":
         snippet = build_javascript_snippet(endpoint_url, token)
     elif language == "rust":
-        snippet = build_rust_snippet(endpoint_url, token)
+        snippet = build_rust_snippet(endpoint_url)
     elif language == "go":
-        snippet = build_go_snippet(endpoint_url, token)
+        snippet = build_go_snippet(endpoint_url)
 
     if not snippet:
         raise CLIPipeException(FeedbackManager.error(message=f"Language {language} not supported"))

tinybird/tb/modules/mock.py

@@ -51,7 +51,7 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str, skip
         if datasource_path.suffix == ".datasource":
             datasource_name = datasource_path.stem
         else:
-            datasource_from_glob = glob.glob(f"**/*/{datasource}.datasource")
+            datasource_from_glob = glob.glob(f"{folder}/**/{datasource}.datasource")
             if datasource_from_glob:
                 datasource_path = Path(datasource_from_glob[0])
 

tinybird/tb/modules/shell.py

@@ -249,11 +249,11 @@ class Shell:
         if cmd in ["exit", "quit"]:
             sys.exit(0)
         elif cmd == "build":
-            self.handle_build(arg)
+            self.handle_build()
         elif cmd == "auth":
-            self.handle_auth(arg)
+            self.handle_auth()
         elif cmd == "workspace":
-            self.handle_workspace(arg)
+            self.handle_workspace()
         elif cmd == "mock":
             self.handle_mock(arg)
         elif cmd == "tb":
@@ -262,27 +262,30 @@ class Shell:
             # Check if it looks like a SQL query or run as a tb command
             self.default(line)
 
-    def handle_build(self, arg):
-        click.echo(FeedbackManager.error(message=f"'tb {arg}' command is not available in watch mode"))
+    def handle_build(self):
+        click.echo(FeedbackManager.error(message="'tb build' command is not available in watch mode"))
 
-    def handle_auth(self, arg):
-        click.echo(FeedbackManager.error(message=f"'tb {arg}' command is not available in watch mode"))
+    def handle_auth(self):
+        click.echo(FeedbackManager.error(message="'tb auth' command is not available in watch mode"))
 
-    def handle_workspace(self, arg):
-        click.echo(FeedbackManager.error(message=f"'tb {arg}' command is not available in watch mode"))
+    def handle_workspace(self):
+        click.echo(FeedbackManager.error(message="'tb workspace' command is not available in watch mode"))
 
     def handle_mock(self, arg):
+        arg = arg.strip().lower()
+        if "mock" in arg:
+            arg = arg.replace("mock", "")
         subprocess.run(f"tb --build mock {arg} --skip", shell=True, text=True)
 
     def handle_tb(self, arg):
         click.echo("")
         arg = arg.strip().lower()
         if arg.startswith("build"):
-            self.handle_build(arg)
+            self.handle_build()
         elif arg.startswith("auth"):
-            self.handle_auth(arg)
+            self.handle_auth()
         elif arg.startswith("workspace"):
-            self.handle_workspace(arg)
+            self.handle_workspace()
         elif arg.startswith("mock"):
             self.handle_mock(arg)
         else:

tinybird/tb/modules/watch.py

@@ -18,12 +18,7 @@ from tinybird.tb.modules.shell import Shell
 
 
 class WatchProjectHandler(PatternMatchingEventHandler):
-    def __init__(
-        self,
-        shell: Shell,
-        project: Project,
-        process: Callable[[bool, Optional[str], Optional[str]], None],
-    ):
+    def __init__(self, shell: Shell, project: Project, process: Callable):
         self.shell = shell
         self.project = project
         self.process = process
@@ -51,7 +46,7 @@ class WatchProjectHandler(PatternMatchingEventHandler):
 
     def _process(self, path: Optional[str] = None) -> None:
         click.echo(FeedbackManager.highlight(message="» Rebuilding project..."))
-        self.process(True, path, self.diff(path))
+        self.process(watch=True, file_changed=path, diff=self.diff(path))
         self.shell.reprint_prompt()
 
     def diff(self, path: Optional[str] = None) -> Optional[str]:

{tinybird-0.0.1.dev63.dist-info → tinybird-0.0.1.dev65.dist-info}/METADATA

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

{tinybird-0.0.1.dev63.dist-info → tinybird-0.0.1.dev65.dist-info}/RECORD

@@ -15,10 +15,10 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=FL85SMPq2dH4JqKovmSbaolGdEzwOO91NqOzqXo2Qr0,41863
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=OXkBhlzGjZotjD0vaT-rFIbSGV4tpiHxE8qO_ip0SyQ,40454
-tinybird/tb/__cli__.py,sha256=QktikNnLUbi0rxQxIzziq0Aaqu14iBSAtYvvVy79A8M,251
+tinybird/tb/__cli__.py,sha256=fD49yxghhJjtdVUmPoEWjIta0RRP1hCf3gR6vBT8PB4,251
 tinybird/tb/cli.py,sha256=FD1pfbzu9YHJHEG6Vtn_EwPLTYhwqw-I6AxXeTaRHU8,926
 tinybird/tb/modules/auth.py,sha256=EzRWFmwRkXNhUmRaruEVFLdkbUg8xMSix0cAWl5D4Jg,9029
-tinybird/tb/modules/build.py,sha256=zo7qexonQ5vDNHZ1u7AK_jdo62hMfG5vWnoQBe_2oyw,9042
+tinybird/tb/modules/build.py,sha256=UN1d7EZ93VOlPCrtsay-KLgZnzxn2NCBDY3wvrUSP1Q,9198
 tinybird/tb/modules/cicd.py,sha256=xxXwy-QekJcG14kkJeGNl7LkHduhZXfvBZE8WrU6-t4,5351
 tinybird/tb/modules/cli.py,sha256=oE1hMCQUxfp1m2r9k6y5LlNgvJCNJhwxfilJbUHp7dY,15914
 tinybird/tb/modules/common.py,sha256=TWcGJUgzJCQvzI1oMKbNdx-KTRmMGvB25BawHpsaV8Q,70610
@@ -27,7 +27,7 @@ tinybird/tb/modules/copy.py,sha256=Aq6wh_wjRiyLQtEOKF9pKLPgJhSvbGTFWIw_LJB0t0U,5
 tinybird/tb/modules/create.py,sha256=Q7ZY4oUmEtAYpUbBLjjyiUZ6IKlkU81qJJrmdgjduOk,13914
 tinybird/tb/modules/datasource.py,sha256=TQ4wSag3CCw34d54FEXPJFGLQNYyNqv2nQbU6QT9uAE,14725
 tinybird/tb/modules/deployment.py,sha256=E3YdQgFmX7rPlgPUw9yAB0jyl7m1MQcqiQF7uHs8wK4,14272
-tinybird/tb/modules/endpoint.py,sha256=M2ZerljZHLJHvDfH30r4Yma3-JGExg2PlRunb2l4r-A,11931
+tinybird/tb/modules/endpoint.py,sha256=7pOF4eLZ41vA5H6bKMCFZnOyKQKfdyytgwTO2Fcxtso,11907
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=mrw5tdYycfvg6WLXlM0KIjfJardm_aNpnJkUg2vH0cA,68463
 tinybird/tb/modules/fmt.py,sha256=poh6_cwVGSf-sBu6LKWuO2TANL_J8Sgm25sPpwxa3Aw,3558
@@ -38,17 +38,17 @@ tinybird/tb/modules/local.py,sha256=_PIa-1M-72bv9rhLwqaNthJM1ZhvcjWXFChZAfEPXRs,
 tinybird/tb/modules/local_common.py,sha256=W1fEnB1vBQ4YC5U1PdA0w0g3cTV78bQ5R-lRxdDj5-Y,2868
 tinybird/tb/modules/login.py,sha256=EGxwVRmMX1Y7ZeCRyA8fqaCWpYYk7NvnZ3x_1g0NlYA,6063
 tinybird/tb/modules/materialization.py,sha256=HQKRTH6lkcYiDQJihbFqF_in58ezXG4ggZ_7Ywp_nUM,5738
-tinybird/tb/modules/mock.py,sha256=7v7ccTrjSRcwgVLs_NfOpAxqdY2L16cUJ3mgm4lf11w,5085
+tinybird/tb/modules/mock.py,sha256=6_hY9a7jdFDwrHnOmOpn1mGchDl9MBIqcNTn8Gm9-Eg,5092
 tinybird/tb/modules/pipe.py,sha256=pH2KwgH6Xbvl3kT8vMelpKvT6bcyB4EKFDvGfOsxXbg,2418
 tinybird/tb/modules/project.py,sha256=RkejMzY6OLWlJMckziZiZFJLnjFxSeQjagklXAYCzoQ,2945
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
-tinybird/tb/modules/shell.py,sha256=LgH5K9mXO0YQ8m0WLw4r9DV7FrnSuFqT0JiAJYjaUsI,14511
+tinybird/tb/modules/shell.py,sha256=B9PLdsSIMN9nsSPecQrlBKNwE5ZUTWoyj_U6KOIxfts,14580
 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=xqu-oPscTVDmH4Q9F6ShQOdOP-aNT0HU-WAQCPzclRc,12871
 tinybird/tb/modules/token.py,sha256=A2I5wTUUmo0KfTI1BH6M6pqCQOd5dE4w2-Xaa1yM5PE,12810
-tinybird/tb/modules/watch.py,sha256=uvLQYejOzcUzupGlRrs4jIgqnmZ1sq_DJxQ0EAG9_58,4591
+tinybird/tb/modules/watch.py,sha256=SfqFfgVML6D73172y0WRPfTpGdUzeJNqrZvVTh8RoM8,4532
 tinybird/tb/modules/workspace.py,sha256=sfT9QkoeFlN7ndUXxyImp4a7EFEHjY9MlGlldOViz0Y,6404
 tinybird/tb/modules/workspace_members.py,sha256=Ai6iCOzXX1zQ8q9iXIFSFHsBJlT-8Q28DaG5Ie-UweY,8726
 tinybird/tb/modules/datafile/build.py,sha256=seGFSvmgyRrAM1-icsKBkuog3WccfGUYFTPT-xoA5W8,50940
@@ -74,8 +74,8 @@ tinybird/tb_cli_modules/config.py,sha256=6u6B5QCdiQLbJkCkwtnKGs9H3nP-KXXhC75mF7B
 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.dev63.dist-info/METADATA,sha256=geQOxg7QCmumWpO-8_OTWSWhE4XICYuUorRLujiLBQQ,2482
-tinybird-0.0.1.dev63.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev63.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev63.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev63.dist-info/RECORD,,
+tinybird-0.0.1.dev65.dist-info/METADATA,sha256=XOsZaJhbk6z2jxLEATAMBLOo4EtGmXnij7ZjGPYabNY,2482
+tinybird-0.0.1.dev65.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev65.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev65.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev65.dist-info/RECORD,,