tinybird 0.0.1.dev18__py3-none-any.whl → 0.0.1.dev20__py3-none-any.whl

tinybird/tb/modules/{build_shell.py → shell.py}

@@ -1,7 +1,6 @@
 import asyncio
 import concurrent.futures
 import os
-import random
 import subprocess
 import sys
 from typing import List
@@ -16,16 +15,19 @@ from prompt_toolkit.shortcuts import CompleteStyle
 from prompt_toolkit.styles import Style
 
 from tinybird.client import TinyB
-from tinybird.feedback_manager import FeedbackManager, bcolors
 from tinybird.tb.modules.exceptions import CLIException
+from tinybird.tb.modules.feedback_manager import FeedbackManager, bcolors
 from tinybird.tb.modules.table import format_table
 
 
 class DynamicCompleter(Completer):
-    def __init__(self, datasources: List[str], pipes: List[str]):
+    def __init__(self, datasources: List[str], shared_datasources: List[str], endpoints: List[str], pipes: List[str]):
         self.datasources = datasources
+        self.shared_datasources = shared_datasources
+        self.endpoints = endpoints
         self.pipes = pipes
         self.static_commands = ["create", "mock", "test", "select"]
+        self.test_commands = ["create", "run", "update"]
         self.mock_flags = ["--prompt", "--rows"]
         self.common_rows = ["10", "50", "100", "500", "1000"]
         self.sql_keywords = ["select", "from", "where", "group by", "order by", "limit"]
@@ -47,6 +49,8 @@ class DynamicCompleter(Completer):
 
         if command == "mock":
             yield from self._handle_mock_completions(words)
+        elif command == "test":
+            yield from self._handle_test_completions(words)
         elif command == "select" or self._is_sql_query(text.lower()):
             yield from self._handle_sql_completions(text)
         else:
@@ -71,9 +75,9 @@ class DynamicCompleter(Completer):
         if words[-1] == "from" or (
             "from" in words and len(words) > words.index("from") + 1 and text_lower.endswith(" ")
         ):
-            for x in self.datasources:
+            for x in self.datasources + self.shared_datasources:
                 yield Completion(x, start_position=0, display=x, style="class:completion.datasource")
-            for x in self.pipes:
+            for x in self.endpoints + self.pipes:
                 yield Completion(x, start_position=0, display=x, style="class:completion.pipe")
             return
 
@@ -89,22 +93,32 @@ class DynamicCompleter(Completer):
         if len(words) == 1:
             # After 'mock', show datasources
             for ds in self.datasources:
-                yield Completion(ds, start_position=0, display=ds, style="class:completion.cmd")
+                yield Completion(ds, start_position=0, display=ds, style="class:completion.datasource")
             return
 
         if len(words) == 2 or len(words) == 4:
             # After datasource or after a flag value, show available flags
             available_flags = [f for f in self.mock_flags if f not in words]
             for flag in available_flags:
-                yield Completion(flag, start_position=0, display=flag)
+                yield Completion(flag, start_position=0, display=flag, style="class:completion.cmd")
             return
 
         last_word = words[-1]
         if last_word == "--prompt":
-            yield Completion('""', start_position=0, display='"Enter your prompt..."')
+            yield Completion('""', start_position=0, display='"Enter your prompt..."', style="class:completion.cmd")
         elif last_word == "--rows":
             for rows in self.common_rows:
-                yield Completion(rows, start_position=0, display=rows)
+                yield Completion(rows, start_position=0, display=rows, style="class:completion.cmd")
+
+    def _handle_test_completions(self, words: List[str]):
+        if len(words) == 1:
+            for cmd in self.test_commands:
+                yield Completion(cmd, start_position=0, display=cmd, style="class:completion.cmd")
+            return
+        elif len(words) == 2:
+            for cmd in self.endpoints:
+                yield Completion(cmd, start_position=0, display=cmd, style="class:completion.pipe")
+            return
 
     def _yield_static_commands(self, current_word: str):
         for cmd in self.static_commands:
@@ -116,6 +130,24 @@ class DynamicCompleter(Completer):
                     style="class:completion.cmd",
                 )
 
+        for cmd in self.datasources + self.shared_datasources:
+            if cmd.startswith(current_word):
+                yield Completion(
+                    cmd,
+                    start_position=-len(current_word) if current_word else 0,
+                    display=cmd,
+                    style="class:completion.datasource",
+                )
+
+        for cmd in self.endpoints + self.pipes:
+            if cmd.startswith(current_word):
+                yield Completion(
+                    cmd,
+                    start_position=-len(current_word) if current_word else 0,
+                    display=cmd,
+                    style="class:completion.pipe",
+                )
+
 
 style = Style.from_dict(
     {
@@ -143,18 +175,27 @@ def _(event):
         b.start_completion(select_first=False)
 
 
-class BuildShell:
-    def __init__(self, folder: str, client: TinyB, datasources: List[str], pipes: List[str]):
+class Shell:
+    def __init__(
+        self,
+        folder: str,
+        client: TinyB,
+        datasources: List[str],
+        shared_datasources: List[str],
+        pipes: List[str],
+        endpoints: List[str],
+    ):
         self.history = self.get_history()
         self.folder = folder
         self.client = client
         self.datasources = datasources
+        self.shared_datasources = shared_datasources
         self.pipes = pipes
+        self.endpoints = endpoints
         self.prompt_message = "\ntb > "
         self.commands = ["create", "mock", "test", "tb", "select"]
-
-        self.session = PromptSession(
-            completer=DynamicCompleter(self.datasources, self.pipes),
+        self.session: PromptSession = PromptSession(
+            completer=DynamicCompleter(self.datasources, self.shared_datasources, self.endpoints, self.pipes),
             complete_style=CompleteStyle.COLUMN,
             complete_while_typing=True,
             history=self.history,
@@ -167,7 +208,7 @@ class BuildShell:
         except Exception:
             return None
 
-    def run_shell(self):
+    def run(self):
         while True:
             try:
                 user_input = self.session.prompt(
@@ -241,10 +282,9 @@ class BuildShell:
         if not arg:
             return
         if arg.startswith("with") or arg.startswith("select"):
-            try:
-                self.run_sql(argline)
-            except Exception as e:
-                click.echo(FeedbackManager.error(message=str(e)))
+            self.run_sql(argline)
+        elif len(arg.split()) == 1 and arg in self.endpoints + self.pipes + self.datasources + self.shared_datasources:
+            self.run_sql(f"select * from {arg}")
         else:
             subprocess.run(f"tb --local {arg}", shell=True, text=True)
 
@@ -271,10 +311,10 @@ class BuildShell:
                 res = executor.submit(run_query_in_thread).result()
 
         except Exception as e:
-            raise CLIException(FeedbackManager.error_exception(error=str(e)))
+            click.echo(FeedbackManager.error_exception(error=str(e)))
 
         if isinstance(res, dict) and "error" in res:
-            raise CLIException(FeedbackManager.error_exception(error=res["error"]))
+            click.echo(FeedbackManager.error_exception(error=res["error"]))
 
         if isinstance(res, dict) and "data" in res and res["data"]:
             print_table_formatted(res, "QUERY")
@@ -286,9 +326,6 @@ class BuildShell:
 
 
 def print_table_formatted(res: dict, name: str):
-    rebuild_colors = [bcolors.FAIL, bcolors.OKBLUE, bcolors.WARNING, bcolors.OKGREEN, bcolors.HEADER]
-    rebuild_index = random.randint(0, len(rebuild_colors) - 1)
-    rebuild_color = rebuild_colors[rebuild_index % len(rebuild_colors)]
     data = []
     limit = 5
     for d in res["data"][:5]:
@@ -299,24 +336,17 @@ def print_table_formatted(res: dict, name: str):
     elapsed = stats.get("elapsed", 0)
     cols = len(meta)
     try:
-
-        def print_message(message: str, color=bcolors.CGREY):
-            return f"{color}{message}{bcolors.ENDC}"
-
         table = format_table(data, meta)
-        colored_char = print_message("│", rebuild_color)
-        table_with_marker = "\n".join(f"{colored_char} {line}" for line in table.split("\n"))
-        click.echo(f"\n{colored_char} {print_message('⚡', rebuild_color)} Running {name}")
-        click.echo(colored_char)
-        click.echo(table_with_marker)
-        click.echo(colored_char)
+        click.echo(FeedbackManager.highlight(message=f"\n» Running {name}\n"))
+        click.echo(table)
+        click.echo("")
         rows_read = humanfriendly.format_number(stats.get("rows_read", 0))
         bytes_read = humanfriendly.format_size(stats.get("bytes_read", 0))
         elapsed = humanfriendly.format_timespan(elapsed) if elapsed >= 1 else f"{elapsed * 1000:.2f}ms"
         stats_message = f"» {bytes_read} ({rows_read} rows x {cols} cols) in {elapsed}"
         rows_message = f"» Showing {limit} first rows" if row_count > limit else "» Showing all rows"
-        click.echo(f"{colored_char} {print_message(stats_message, bcolors.OKGREEN)}")
-        click.echo(f"{colored_char} {print_message(rows_message, bcolors.CGREY)}")
+        click.echo(FeedbackManager.success(message=stats_message))
+        click.echo(FeedbackManager.gray(message=rows_message))
     except ValueError as exc:
         if str(exc) == "max() arg is an empty sequence":
             click.echo("------------")

tinybird/tb/modules/table.py

@@ -1,10 +1,10 @@
 # Standard library modules.
 import collections
 import re
+from typing import Union
 
 # Modules included in our package.
 from humanfriendly.compat import coerce_string
-from humanfriendly.tables import format_robust_table
 from humanfriendly.terminal import (
     ansi_strip,
     ansi_width,
@@ -13,7 +13,7 @@ from humanfriendly.terminal import (
     terminal_supports_colors,
 )
 
-from tinybird.feedback_manager import bcolors
+from tinybird.tb.modules.feedback_manager import bcolors
 
 NUMERIC_DATA_PATTERN = re.compile(r"^\d+(\.\d+)?$")
 
@@ -133,7 +133,7 @@ def format_pretty_table(data, column_names=None, column_types=None, horizontal_b
                 column_types = [highlight_column_type(t) for t in column_types]
                 data.insert(1, column_types)
     # Calculate the maximum width of each column.
-    widths = collections.defaultdict(int)
+    widths: dict[int, int] = collections.defaultdict(int)
     numeric_data = collections.defaultdict(list)
     for row_index, row in enumerate(data):
         for column_index, column in enumerate(row):
@@ -177,9 +177,92 @@ def highlight_column_name(name):
     return ansi_wrap(name)
 
 
+def highlight_column_name_robust(name):
+    return f"{bcolors.CGREY}{name}{bcolors.ENDC}"
+
+
 def highlight_column_type(type):
     return f"{bcolors.CGREY}\033[3m{type}\033[23m{bcolors.ENDC}"
 
 
-def highlight_horizontal_bar(horizontal_bar):
-    return f"{bcolors.CGREY}\033[3m{horizontal_bar}\033[23m{bcolors.ENDC}"
+def highlight_horizontal_bar(horizontal_bar: str) -> str:
+    return f"\033[38;5;247m\033[3m{horizontal_bar}\033[23m{bcolors.ENDC}"
+
+
+def format_robust_table(data, column_names):
+    """
+    Render tabular data with one column per line (allowing columns with line breaks).
+
+    :param data: An iterable (e.g. a :func:`tuple` or :class:`list`)
+                 containing the rows of the table, where each row is an
+                 iterable containing the columns of the table (strings).
+    :param column_names: An iterable of column names (strings).
+    :returns: The rendered table (a string).
+
+    Here's an example:
+
+    >>> from humanfriendly.tables import format_robust_table
+    >>> column_names = ['Version', 'Uploaded on', 'Downloads']
+    >>> humanfriendly_releases = [
+    ... ['1.23', '2015-05-25', '218'],
+    ... ['1.23.1', '2015-05-26', '1354'],
+    ... ['1.24', '2015-05-26', '223'],
+    ... ['1.25', '2015-05-26', '4319'],
+    ... ['1.25.1', '2015-06-02', '197'],
+    ... ]
+    >>> print(format_robust_table(humanfriendly_releases, column_names))
+    -----------------------
+    Version: 1.23
+    Uploaded on: 2015-05-25
+    Downloads: 218
+    -----------------------
+    Version: 1.23.1
+    Uploaded on: 2015-05-26
+    Downloads: 1354
+    -----------------------
+    Version: 1.24
+    Uploaded on: 2015-05-26
+    Downloads: 223
+    -----------------------
+    Version: 1.25
+    Uploaded on: 2015-05-26
+    Downloads: 4319
+    -----------------------
+    Version: 1.25.1
+    Uploaded on: 2015-06-02
+    Downloads: 197
+    -----------------------
+
+    The column names are highlighted in bold font and color so they stand out a
+    bit more (see :data:`.HIGHLIGHT_COLOR`).
+    """
+    blocks: list[Union[str, list[str]]] = []
+    column_names = ["%s:" % n for n in normalize_columns(column_names)]
+    if terminal_supports_colors():
+        column_names = [highlight_column_name_robust(n) for n in column_names]
+    # Convert each row into one or more `name: value' lines (one per column)
+    # and group each `row of lines' into a block (i.e. rows become blocks).
+    for row in data:
+        lines = []
+        for column_index, column_text in enumerate(normalize_columns(row)):
+            stripped_column = column_text.strip()
+            if "\n" not in stripped_column:
+                # Columns without line breaks are formatted inline.
+                lines.append("%s %s" % (column_names[column_index], stripped_column))
+            else:
+                # Columns with line breaks could very well contain indented
+                # lines, so we'll put the column name on a separate line. This
+                # way any indentation remains intact, and it's easier to
+                # copy/paste the text.
+                lines.append(column_names[column_index])
+                lines.extend(column_text.rstrip().splitlines())
+        blocks.append(lines)
+    # Calculate the width of the row delimiter.
+    num_rows, num_columns = find_terminal_size()
+    longest_line = max(max(map(ansi_width, lines)) for lines in blocks)
+    delimiter = highlight_horizontal_bar("\n%s\n" % ("─" * min(longest_line, num_columns)))
+    # Force a delimiter at the start and end of the table.
+    blocks.insert(0, "")
+    blocks.append("")
+    # Embed the row delimiter between every two blocks.
+    return delimiter.join("\n".join(b) for b in blocks).strip()

tinybird/tb/modules/tag.py

@@ -3,12 +3,12 @@ from typing import Optional
 import click
 from click import Context
 
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import coro, echo_safe_humanfriendly_tables_format_smart_table
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-@cli.group()
+@cli.group(hidden=True)
 @click.pass_context
 def tag(ctx: Context) -> None:
     """Tag commands"""

tinybird/tb/modules/test.py

@@ -7,20 +7,20 @@ import difflib
 import glob
 import os
 from pathlib import Path
-from typing import Iterable, List, Optional, Tuple
+from typing import Any, Dict, Iterable, List, Optional, Tuple
 
 import click
 import yaml
 
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import coro
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import CLIException
-from tinybird.tb.modules.llm import LLM, TestExpectation
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.llm import LLM
 from tinybird.tb.modules.local_common import get_tinybird_local_client
 
-yaml.SafeDumper.org_represent_str = yaml.SafeDumper.represent_str
+yaml.SafeDumper.org_represent_str = yaml.SafeDumper.represent_str  # type: ignore[attr-defined]
 
 
 def repr_str(dumper, data):
@@ -32,7 +32,7 @@ def repr_str(dumper, data):
 yaml.add_representer(str, repr_str, Dumper=yaml.SafeDumper)
 
 
-def generate_test_file(pipe_name: str, tests: List[TestExpectation], folder: Optional[str], mode: str = "w"):
+def generate_test_file(pipe_name: str, tests: List[Dict[str, Any]], folder: Optional[str], mode: str = "w"):
     base = Path("tests")
     if folder:
         base = Path(folder) / base
@@ -65,9 +65,8 @@ def test(ctx: click.Context) -> None:
     help="Folder where datafiles will be placed",
 )
 @click.option("--prompt", type=str, default=None, help="Prompt to be used to create the test")
-@click.pass_context
 @coro
-async def test_create(ctx: click.Context, pipe: str, prompt: Optional[str], folder: Optional[str]) -> None:
+async def test_create(pipe: str, prompt: Optional[str], folder: str) -> None:
     """
     Create a test for an existing endpoint
     """
@@ -102,13 +101,21 @@ async def test_create(ctx: click.Context, pipe: str, prompt: Optional[str], fold
             test_params = (
                 valid_test["parameters"] if valid_test["parameters"].startswith("?") else f"?{valid_test['parameters']}"
             )
-            response = ""
+
+            response = None
             try:
-                response = await client._req(f"/v0/pipes/{pipe_name}.ndjson{test_params}")
+                response = await client._req_raw(f"/v0/pipes/{pipe_name}.ndjson{test_params}")
             except Exception:
-                valid_test["expected_http_status"] = 500
+                continue
+
+            if response.status_code >= 400:
+                valid_test["expected_http_status"] = response.status_code
+                valid_test["expected_result"] = response.json()["error"]
+            else:
+                if "expected_http_status" in valid_test:
+                    del valid_test["expected_http_status"]
+                valid_test["expected_result"] = response.text or ""
 
-            valid_test["expected_result"] = response
             valid_test_expectations.append(valid_test)
         if valid_test_expectations:
             generate_test_file(pipe_name, valid_test_expectations, folder, mode="a")
@@ -129,11 +136,9 @@ async def test_create(ctx: click.Context, pipe: str, prompt: Optional[str], fold
     type=click.Path(exists=True, file_okay=False),
     help="Folder where datafiles will be placed",
 )
-@click.pass_context
 @coro
-async def test_update(ctx: click.Context, pipe: str, folder: Optional[str]) -> None:
+async def test_update(pipe: str, folder: str) -> None:
     client = await get_tinybird_local_client(os.path.abspath(folder))
-
     pipe_tests_path = Path(pipe)
     pipe_name = pipe
     if pipe_tests_path.suffix == ".yaml":
@@ -146,16 +151,20 @@ async def test_update(ctx: click.Context, pipe: str, folder: Optional[str]) -> N
     pipe_tests_content = yaml.safe_load(pipe_tests_path.read_text())
     for test in pipe_tests_content:
         test_params = test["parameters"] if test["parameters"].startswith("?") else f"?{test['parameters']}"
-        response = ""
+        response = None
         try:
-            response = await client._req(f"/v0/pipes/{pipe_name}.ndjson{test_params}")
+            response = await client._req_raw(f"/v0/pipes/{pipe_name}.ndjson{test_params}")
         except Exception:
-            test["expected_http_status"] = 500
+            continue
+
+        if response.status_code >= 400:
+            test["expected_http_status"] = response.status_code
+            test["expected_result"] = response.json()["error"]
         else:
             if "expected_http_status" in test:
                 del test["expected_http_status"]
 
-        test["expected_result"] = response
+            test["expected_result"] = response.text or ""
 
     generate_test_file(pipe_name, pipe_tests_content, folder)
     click.echo(FeedbackManager.info(message=f"✓ /tests/{pipe_name}.yaml"))
@@ -166,18 +175,19 @@ async def test_update(ctx: click.Context, pipe: str, folder: Optional[str]) -> N
     name="run",
     help="Run the test suite, a file, or a test.",
 )
-@click.argument("file", nargs=-1)
+@click.argument("name", nargs=-1)
 @click.option(
     "--folder",
     default=".",
     type=click.Path(exists=True, file_okay=False),
     help="Folder where tests will be placed",
 )
-@click.pass_context
 @coro
-async def test_run(ctx: click.Context, file: Tuple[str, ...], folder: Optional[str]) -> None:
+async def test_run(name: Tuple[str, ...], folder: str) -> None:
     client = await get_tinybird_local_client(os.path.abspath(folder))
-    file_list: Iterable[str] = file if len(file) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
+    paths = [Path(n) for n in name]
+    endpoints = [f"./tests/{p.stem}.yaml" for p in paths]
+    file_list: Iterable[str] = endpoints if len(endpoints) > 0 else glob.glob("./tests/**/*.y*ml", recursive=True)
 
     async def run_test(test_file):
         test_file_path = Path(test_file)
@@ -185,22 +195,28 @@ async def test_run(ctx: click.Context, file: Tuple[str, ...], folder: Optional[s
         for test in test_file_content:
             try:
                 test_params = test["parameters"] if test["parameters"].startswith("?") else f"?{test['parameters']}"
-                response = ""
+
+                response = None
                 try:
-                    response = await client._req(f"/v0/pipes/{test_file_path.stem}.ndjson{test_params}")
+                    response = await client._req_raw(f"/v0/pipes/{test_file_path.stem}.ndjson{test_params}")
                 except Exception:
+                    raise Exception("Expected to not fail but got an error")
+
+                expected_result = response.text
+                if response.status_code >= 400:
+                    expected_result = response.json()["error"]
                     if "expected_http_status" not in test:
                         raise Exception("Expected to not fail but got an error")
-                    if test["expected_http_status"] != 500:
-                        raise Exception(f"Expected {test['expected_http_status']} but got another status")
+                    if test["expected_http_status"] != response.status_code:
+                        raise Exception(f"Expected {test['expected_http_status']} but got {response.status_code}")
 
-                if test["expected_result"] != response:
+                if test["expected_result"] != expected_result:
                     diff = difflib.ndiff(
-                        test["expected_result"].splitlines(keepends=True), response.splitlines(keepends=True)
+                        test["expected_result"].splitlines(keepends=True), expected_result.splitlines(keepends=True)
                     )
                     printable_diff = "".join(diff)
                     raise Exception(
-                        f"\nExpected: \n{test['expected_result']}\nGot: \n{response}\nDiff: \n{printable_diff}"
+                        f"\nExpected: \n{test['expected_result']}\nGot: \n{expected_result}\nDiff: \n{printable_diff}"
                     )
                 click.echo(FeedbackManager.success(message=f"✓ {test_file_path.name} - {test['name']}"))
             except Exception as e:

tinybird/tb/modules/tinyunit/tinyunit.py

@@ -8,8 +8,8 @@ from humanfriendly.tables import format_smart_table
 from typing_extensions import override
 
 from tinybird.client import TinyB
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.common import CLIException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
 @dataclass

tinybird/tb/modules/token.py

@@ -7,7 +7,6 @@ from click import Context
 from humanfriendly import parse_timespan
 
 from tinybird.client import AuthNoTokenException, TinyB
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import (
     DoesNotExistException,
@@ -15,9 +14,10 @@ from tinybird.tb.modules.common import (
     echo_safe_humanfriendly_tables_format_smart_table,
 )
 from tinybird.tb.modules.exceptions import CLITokenException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
-@cli.group()
+@cli.group(hidden=True)
 @click.pass_context
 def token(ctx: Context) -> None:
     """Token commands."""

tinybird/tb/modules/watch.py

@@ -0,0 +1,72 @@
+import asyncio
+import time
+from typing import Any, Callable, List
+
+import click
+from watchdog.events import FileSystemEventHandler
+from watchdog.observers import Observer
+
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+from tinybird.tb.modules.shell import Shell
+
+
+class FileChangeHandler(FileSystemEventHandler):
+    def __init__(self, filenames: List[str], process: Callable[[List[str]], None], build_ok: bool):
+        self.filenames = filenames
+        self.process = process
+        self.build_ok = build_ok
+
+    def on_modified(self, event: Any) -> None:
+        is_not_vendor = "vendor/" not in event.src_path
+        if (
+            is_not_vendor
+            and not event.is_directory
+            and any(event.src_path.endswith(ext) for ext in [".datasource", ".pipe", ".ndjson"])
+        ):
+            filename = event.src_path.split("/")[-1]
+            click.echo(FeedbackManager.highlight(message=f"\n\n⟲ Changes detected in {filename}\n"))
+            try:
+                to_process = [event.src_path] if self.build_ok else self.filenames
+                self.process(to_process)
+                self.build_ok = True
+            except Exception as e:
+                click.echo(FeedbackManager.error_exception(error=e))
+
+
+def watch_files(
+    filenames: List[str],
+    process: Callable,
+    shell: Shell,
+    folder: str,
+    build_ok: bool,
+) -> None:
+    # Handle both sync and async process functions
+    async def process_wrapper(files: List[str]) -> None:
+        click.echo("⚡ Rebuilding...")
+        time_start = time.time()
+        if asyncio.iscoroutinefunction(process):
+            await process(files, watch=True)
+        else:
+            process(files, watch=True)
+        time_end = time.time()
+        elapsed_time = time_end - time_start
+        click.echo(
+            FeedbackManager.success(message="\n✓ ")
+            + FeedbackManager.gray(message=f"Rebuild completed in {elapsed_time:.1f}s")
+        )
+        shell.reprint_prompt()
+
+    event_handler = FileChangeHandler(filenames, lambda f: asyncio.run(process_wrapper(f)), build_ok)
+    observer = Observer()
+
+    observer.schedule(event_handler, path=folder, recursive=True)
+
+    observer.start()
+
+    try:
+        while True:
+            time.sleep(1)
+    except KeyboardInterrupt:
+        observer.stop()
+
+    observer.join()

tinybird/tb/modules/workspace.py

@@ -10,7 +10,6 @@ from click import Context
 
 from tinybird.client import CanNotBeDeletedException, DoesNotExistException, TinyB
 from tinybird.config import get_display_host
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import (
     _get_workspace_plan_name,
@@ -28,6 +27,7 @@ from tinybird.tb.modules.common import (
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.datafile.common import PipeTypes
 from tinybird.tb.modules.exceptions import CLIWorkspaceException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 
 
 @cli.group()

tinybird/tb/modules/workspace_members.py

@@ -12,7 +12,6 @@ from click import Context
 
 from tinybird.client import TinyB
 from tinybird.config import get_display_host
-from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb.modules.common import (
     ask_for_user_token,
     check_user_token,
@@ -22,6 +21,7 @@ from tinybird.tb.modules.common import (
 )
 from tinybird.tb.modules.config import CLIConfig
 from tinybird.tb.modules.exceptions import CLIWorkspaceMembersException
+from tinybird.tb.modules.feedback_manager import FeedbackManager
 from tinybird.tb.modules.workspace import workspace
 
 ROLES = ["viewer", "guest", "admin"]

{tinybird-0.0.1.dev18.dist-info → tinybird-0.0.1.dev20.dist-info}/METADATA

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