tinybird 0.0.1.dev39__tar.gz → 0.0.1.dev41__tar.gz

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/PKG-INFO

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

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/tinybird/prompts.py

@@ -646,6 +646,7 @@ datasource_instructions = """
     - No indentation is allowed for property names: DESCRIPTION, SCHEMA, ENGINE, ENGINE_PARTITION_KEY, ENGINE_SORTING_KEY, etc.
     - Use MergeTree engine by default.
     - Use AggregatingMergeTree engine when the datasource is the target of a materialized pipe.
+    - Use always json paths to define the schema. Example: `user_id` String `json:$.user_id`,
 </datasource_file_instructions>
 """
 
@@ -779,3 +780,24 @@ Use the following format to generate the response and do not wrap it in any othe
         copy_pipe_instructions=copy_pipe_instructions,
         materialized_pipe_instructions=materialized_pipe_instructions,
     )
+
+
+cursorrules_prompt = """
+You are an expert in SQL and Tinybird. Follow these instructions when working with .datasource and .pipe files:
+
+{datasource_instructions}
+{pipe_instructions}
+{sql_instructions}
+{datasource_example}
+{pipe_example}
+{copy_pipe_instructions}
+{materialized_pipe_instructions}
+""".format(
+    datasource_instructions=datasource_instructions,
+    pipe_instructions=pipe_instructions,
+    sql_instructions=sql_instructions,
+    datasource_example=datasource_example,
+    pipe_example=pipe_example,
+    copy_pipe_instructions=copy_pipe_instructions,
+    materialized_pipe_instructions=materialized_pipe_instructions,
+)

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/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.dev39'
-__revision__ = 'b1233f1'
+__version__ = '0.0.1.dev41'
+__revision__ = '86b5c65'

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/tinybird/tb/modules/build.py

@@ -36,7 +36,7 @@ def build(ctx: click.Context, watch: bool) -> None:
 
     def process(file_changed: Optional[str] = None) -> None:
         if file_changed and file_changed.endswith(".ndjson"):
-            rebuild_fixture(project, tb_client, file_changed)
+            rebuild_fixture(project, file_changed)
         else:
             build_project(project, tb_client)
             new_tb_client = asyncio.run(get_tinybird_local_client(str(project.path)))
@@ -154,7 +154,6 @@ def append_fixture(
     datasource_name: str,
     url: str,
 ):
-    asyncio.run(tb_client.datasource_truncate(datasource_name))
     asyncio.run(
         push_data(
             tb_client,
@@ -167,21 +166,26 @@ def append_fixture(
     )
 
 
-def rebuild_fixture(project: Project, tb_client: TinyB, fixture: str) -> None:
-    fixture_path = Path(fixture)
-    datasources_path = Path(project.folder) / "datasources"
-    ds_name = fixture_path.stem
+def rebuild_fixture(project: Project, fixture: str) -> None:
+    try:
+        tb_client = asyncio.run(get_tinybird_local_client(str(project.path)))
+        fixture_path = Path(fixture)
+        datasources_path = Path(project.folder) / "datasources"
+        ds_name = fixture_path.stem
 
-    if ds_name not in project.datasources:
-        try:
-            ds_name = "_".join(fixture_path.stem.split("_")[:-1])
-        except Exception:
-            pass
+        if ds_name not in project.datasources:
+            try:
+                ds_name = "_".join(fixture_path.stem.split("_")[:-1])
+            except Exception:
+                pass
 
-    ds_path = datasources_path / f"{ds_name}.datasource"
+        ds_path = datasources_path / f"{ds_name}.datasource"
 
-    if ds_path.exists():
-        append_fixture(tb_client, ds_name, str(fixture_path))
+        if ds_path.exists():
+            asyncio.run(tb_client.datasource_truncate(ds_name))
+            append_fixture(tb_client, ds_name, str(fixture_path))
+    except Exception as e:
+        click.echo(FeedbackManager.error_exception(error=e))
 
 
 def build_and_print_resource(tb_client: TinyB, filename: str):

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/tinybird/tb/modules/create.py

@@ -7,7 +7,7 @@ from typing import Optional
 import click
 
 from tinybird.client import TinyB
-from tinybird.prompts import create_prompt, mock_prompt
+from tinybird.prompts import create_prompt, cursorrules_prompt, mock_prompt
 from tinybird.tb.modules.cicd import init_cicd
 from tinybird.tb.modules.cli import cli
 from tinybird.tb.modules.common import _generate_datafile, check_user_token_with_client, coro, generate_datafile
@@ -40,12 +40,14 @@ from tinybird.tb.modules.local_common import get_tinybird_local_client
     help="Folder where datafiles will be placed",
 )
 @click.option("--rows", type=int, default=10, help="Number of events to send")
+@click.option("--cursor", default=False, is_flag=True, help="Create .cursorrules file with Tinybird rules")
 @coro
 async def create(
     data: Optional[str],
     prompt: Optional[str],
     folder: Optional[str],
     rows: int,
+    cursor: bool,
 ) -> None:
     """Initialize a new project."""
     folder = folder or getcwd()
@@ -121,7 +123,11 @@ async def create(
                             persist_fixture(fixture_name, data, folder)
                             click.echo(FeedbackManager.info(message=f"✓ /fixtures/{datasource_name}"))
 
+        if cursor and not already_has_cursorrules(folder):
+            click.echo(FeedbackManager.highlight(message="\n» Creating .cursorrules..."))
+            create_cursorrules(folder)
             click.echo(FeedbackManager.success(message="✓ Done!\n"))
+
     except Exception as e:
         click.echo(FeedbackManager.error(message=f"Error: {str(e)}"))
 
@@ -286,3 +292,12 @@ def generate_pipe_file(name: str, content: str, folder: str):
     with open(f"{f}", "w") as file:
         file.write(content)
     click.echo(FeedbackManager.info_file_created(file=f.relative_to(folder)))
+
+
+def already_has_cursorrules(folder: str) -> bool:
+    return (Path(folder) / ".cursorrules").exists()
+
+
+def create_cursorrules(folder: str):
+    cursorrules_file = Path(folder) / ".cursorrules"
+    cursorrules_file.write_text(cursorrules_prompt)

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/tinybird/tb/modules/mock.py

@@ -47,7 +47,7 @@ async def mock(datasource: str, rows: int, prompt: str, folder: str) -> None:
         datasource_path = Path(folder) / datasource_path
 
         prompt_path = Path(folder) / "fixtures" / f"{datasource_name}.prompt"
-        if not 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()

{tinybird-0.0.1.dev39 → tinybird-0.0.1.dev41}/tinybird.egg-info/PKG-INFO

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