tinybird 0.0.1.dev39__py3-none-any.whl → 0.0.1.dev41__py3-none-any.whl

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/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/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/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/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.dist-info → tinybird-0.0.1.dev41.dist-info}/METADATA

@@ -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.dist-info → tinybird-0.0.1.dev41.dist-info}/RECORD

@@ -6,7 +6,7 @@ tinybird/context.py,sha256=A3GBApac9xO6hrAMJ1s9dMrI_ou9aKF84CdEjtPddMk,1417
 tinybird/datatypes.py,sha256=XNypumfqNjsvLJ5iNXnbVHRvAJe0aQwI3lS6Cxox-e0,10979
 tinybird/feedback_manager.py,sha256=ON5Zu-G3-QDVfH2i_P-V4EtyhlNtAzyp1YDZsnce0_U,67826
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
-tinybird/prompts.py,sha256=EDQXikdLUFK0TvDsT7ROKcbAKNOyUC2JhX_Ff5GN_Sg,24515
+tinybird/prompts.py,sha256=2UdLOCjIWbLINI1TuUSDXOW2T0YaPD0RHZAa26j6VlE,25247
 tinybird/sql.py,sha256=eulpRe05ZFrKFrxYawgxDxxrktFE8uL6hSL1gHIWKyg,46166
 tinybird/sql_template.py,sha256=GmMLAI10MTqjQo9qztuQHLRWs67teozsWDxUBdvkAn4,93668
 tinybird/sql_template_fmt.py,sha256=1z-PuqSZXtzso8Z_mPqUc-NxIxUrNUcVIPezNieZk-M,10196
@@ -15,17 +15,17 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=oflXyoL2LSCegvl6bAzqw2JIqRaN5WPjhYYDtQcfuOE,41869
 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=s5p1QjpWex5BFp26bvSzNxyuKrmji3nz6m_DT_LLOFQ,251
+tinybird/tb/__cli__.py,sha256=Py43Okm6MRi1REBHdZBS-rEI97DI49FulIJwEZXDzhs,251
 tinybird/tb/cli.py,sha256=_kYDnDS3a45MMKJFZnYZx1gLuVqs4N_Rt8GO4sueAeg,957
 tinybird/tb/modules/auth.py,sha256=EzRWFmwRkXNhUmRaruEVFLdkbUg8xMSix0cAWl5D4Jg,9029
-tinybird/tb/modules/build.py,sha256=95Ll9UPIxMqJU3CP7csJMuZdNmUiDlotFyP5-o5Qxgo,7310
+tinybird/tb/modules/build.py,sha256=N1I_MWPChBdfrFMcHdIbsTbMrGSNDBPBmhv2xSLsoWw,7500
 tinybird/tb/modules/build_client.py,sha256=na4MH0D4_yXMkNEW2a9bslW6t1fBRnr8JTN68MfRvvw,7229
 tinybird/tb/modules/cicd.py,sha256=SjCyvvy0WUnsjFs2biwwXvcf0Ddpmghhd8-SnMyfsRM,5355
 tinybird/tb/modules/cli.py,sha256=hD_mUAXtUofU20UC9j8Pzc4QI1gV0Y9Zefwe24T-qZA,19427
 tinybird/tb/modules/common.py,sha256=e4U7AT0dUBG6O-7Iq2CVN1UHPd6-ZCFucyW0L5gBi4g,70592
 tinybird/tb/modules/config.py,sha256=mie3oMVTf5YOUFEiLs88P16U4LkJafJjSpjwyAkFHog,10979
 tinybird/tb/modules/copy.py,sha256=qrtU1PFXmu-xhq1LR2uo7-1MoCOOQeVb_j10vcasXmI,2415
-tinybird/tb/modules/create.py,sha256=x-lXLlr9d0kvkdqHbioA0m-hgutRXWeJS_pjJjoqrFM,10854
+tinybird/tb/modules/create.py,sha256=iUYt5XG-GPwE3LnHrWqlOHmke0Bv8VMPxjAffxRFYoQ,11438
 tinybird/tb/modules/datasource.py,sha256=-VG2qKlu0fmkhsIB5bPiTp3XuktB_r-ZkIoohEBEXtI,13713
 tinybird/tb/modules/deployment.py,sha256=fCWHlfMrAbHLst_-pveH6oPHPnDsXqv1bAkzoFtxvEM,9597
 tinybird/tb/modules/endpoint.py,sha256=iYSWzi3_VJzHcq1_j_Hv4cfG1GFKXKxqEY4jLjKhxag,6488
@@ -38,7 +38,7 @@ tinybird/tb/modules/llm_utils.py,sha256=zUwcF8hgBsxWVWgCzSRh--O7lWVZ_wzj7bxN8Yl5
 tinybird/tb/modules/local.py,sha256=x4xuCGVkoa8KLYGZEJnFUP8HUkKX05Frp_djRVjVjTs,5669
 tinybird/tb/modules/local_common.py,sha256=afPW6bSc-YI7Q4ngvBV53sM7QxnPGBMe5N91iETM2yE,2783
 tinybird/tb/modules/login.py,sha256=0cS-f3MsQFHc6xjw8FRWJm4EJBH9C7Ri68EcO_tiwes,6508
-tinybird/tb/modules/mock.py,sha256=TdqAQwF73b95_EVBFVHH0gOghtm3fuMsMCsddCEsZAY,3699
+tinybird/tb/modules/mock.py,sha256=XoCaFFroJf2jWVxEztPelwXYNle__KilON1e81Mxkd4,3764
 tinybird/tb/modules/pipe.py,sha256=Kay7AZVf_M5biIvX5hi-Vaz4l9C7AV-s0C2Nle_gkJo,17528
 tinybird/tb/modules/project.py,sha256=EyrX5-5i3vhWk6v3KzbrZg8im4ey23hpzM7U651u6OE,1432
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
@@ -75,8 +75,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.dev39.dist-info/METADATA,sha256=BCc5Sxqt1NIgGWXpzjatFsNovK4Z_jHKv8syR7nlQ6M,2482
-tinybird-0.0.1.dev39.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev39.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev39.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
-tinybird-0.0.1.dev39.dist-info/RECORD,,
+tinybird-0.0.1.dev41.dist-info/METADATA,sha256=VrfRatTvL5BvgKELZsK-vEW6vu_4_5cbWnWf7Penawo,2482
+tinybird-0.0.1.dev41.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev41.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev41.dist-info/top_level.txt,sha256=pgw6AzERHBcW3YTi2PW4arjxLkulk2msOz_SomfOEuc,45
+tinybird-0.0.1.dev41.dist-info/RECORD,,