tinybird 0.0.1.dev85__py3-none-any.whl → 0.0.1.dev87__py3-none-any.whl

tinybird/sql_template.py

@@ -1452,6 +1452,13 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
         template_execution_results["activate"] = feature
         return Expression(f"-- activate {feature}\n")
 
+    def set_disable_feature(feature):
+        valid_features = "analyzer"
+        if feature not in valid_features:
+            raise SQLTemplateException(f"'{feature}' is not a valid 'disable' argument")
+        template_execution_results["disable"] = feature
+        return Expression(f"-- disable {feature}\n")
+
     namespace.update(_namespace)
     namespace.update(kwargs)
     namespace.update(
@@ -1466,6 +1473,7 @@ def generate(self, **kwargs) -> Tuple[str, TemplateExecutionResults]:
             "backend_hint": set_backend_hint,
             "cache_ttl": set_cache_ttl,
             "activate": set_activate,
+            "disable": set_disable_feature,
         }
     )
 

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.dev85'
-__revision__ = 'd751c6c'
+__version__ = '0.0.1.dev87'
+__revision__ = '029c262'

tinybird/tb/cli.py

@@ -17,6 +17,7 @@ import tinybird.tb.modules.fmt
 import tinybird.tb.modules.job
 import tinybird.tb.modules.local
 import tinybird.tb.modules.login
+import tinybird.tb.modules.logout
 import tinybird.tb.modules.materialization
 import tinybird.tb.modules.mock
 import tinybird.tb.modules.pipe

tinybird/tb/modules/build.py

@@ -57,6 +57,7 @@ def build_project(project: Project, tb_client: TinyB, file_changed: Optional[str
     DATAFILE_TYPE_TO_CONTENT_TYPE = {
         ".datasource": "text/plain",
         ".pipe": "text/plain",
+        ".connection": "text/plain",
     }
     TINYBIRD_API_URL = tb_client.host + "/v1/build"
     logging.debug(TINYBIRD_API_URL)

tinybird/tb/modules/cli.py

@@ -7,7 +7,6 @@ import json
 import logging
 import os
 import sys
-from os import getcwd
 from typing import Any, Callable, Dict, List, Optional, Tuple, Union
 
 import click
@@ -146,9 +145,9 @@ async def pull(ctx: Context, force: bool, fmt: bool) -> None:
     """Retrieve latest versions for project files from Tinybird."""
 
     client = ctx.ensure_object(dict)["client"]
-    folder = getcwd()
+    project = ctx.ensure_object(dict)["project"]
 
-    return await folder_pull(client, folder, force, fmt=fmt)
+    return await folder_pull(client, project.path, force, fmt=fmt)
 
 
 @cli.command(hidden=True)
@@ -394,7 +393,7 @@ def __unpatch_click_output():
 
 
 async def create_ctx_client(ctx: Context, config: Dict[str, Any], cloud: bool, build: bool, staging: bool):
-    commands_without_ctx_client = ["auth", "check", "login", "local", "update", "upgrade"]
+    commands_without_ctx_client = ["auth", "check", "login", "local", "update", "upgrade", "logout"]
     command = ctx.invoked_subcommand
     if command in commands_without_ctx_client:
         return None

tinybird/tb/modules/config.py

@@ -79,7 +79,6 @@ class CLIConfig:
     _projects: Dict[str, "CLIConfig"] = {}
 
     def __init__(self, path: Optional[str] = None) -> None:
-        self.cwd = os.getcwd()
         self._path = path or self.get_tinyb_file() or os.path.join(os.getcwd(), ".tinyb")
         self._values: Dict[str, ConfigValue] = {}
         self._values["version"] = ConfigValue("version", CURRENT_VERSION, ConfigValueOrigin.DEFAULT)

tinybird/tb/modules/create.py

@@ -51,6 +51,9 @@ async def create(
         config.persist_to_file()
         project.folder = folder
 
+    if cwd := config.get("cwd"):
+        click.echo(FeedbackManager.gray(message=f"Using '{cwd.replace(os.getcwd(), '')}' as target folder"))
+
     root_folder = os.getcwd()
     if config._path:
         root_folder = os.path.dirname(config._path)

tinybird/tb/modules/datafile/common.py

@@ -168,14 +168,18 @@ class UnkownExtensionerror(Exception): ...
 class DatafileKind(Enum):
     pipe = "pipe"
     datasource = "datasource"
+    connection = "connection"
 
     @classmethod
     def from_extension(cls, extension: str) -> DatafileKind:
-        if extension == ".pipe":
-            return cls.pipe
-        elif extension == ".datasource":
-            return cls.datasource
-        raise UnkownExtensionerror(f"Unknown extension {extension} for data file")
+        extension_map = {
+            ".pipe": cls.pipe,
+            ".datasource": cls.datasource,
+            ".connection": cls.connection,
+        }
+        if extension not in extension_map:
+            raise UnkownExtensionerror(f"Unknown extension {extension} for data file")
+        return extension_map[extension]
 
 
 class Datafile:
@@ -1099,9 +1103,7 @@ def parse(
 ) -> Datafile:
     """
     Parses `s` string into a document
-    >>> d = parse("MAINTAINER 'rambo' #this is me\\nNODE \\"test_01\\"\\n    DESCRIPTION this is a node that does whatever\\nSQL >\\n\\n        SELECT * from test_00\\n\\n\\nNODE \\"test_02\\"\\n    DESCRIPTION this is a node that does whatever\\nSQL >\\n\\n    SELECT * from test_01\\n    WHERE a > 1\\n    GROUP by a\\n")
-    >>> d.maintainer
-    'rambo'
+    >>> d = parse("NODE \\"test_01\\"\\n    DESCRIPTION this is a node that does whatever\\nSQL >\\n\\n        SELECT * from test_00\\n\\n\\nNODE \\"test_02\\"\\n    DESCRIPTION this is a node that does whatever\\nSQL >\\n\\n    SELECT * from test_01\\n    WHERE a > 1\\n    GROUP by a\\n")
     >>> len(d.nodes)
     2
     >>> d.nodes[0]
@@ -1377,7 +1379,6 @@ def parse(
 
     cmds = {
         "source": sources,
-        "maintainer": assign("maintainer"),
         "schema": schema,
         "indexes": indexes,
         "engine": set_engine,
@@ -1417,6 +1418,7 @@ def parse(
         "kafka_store_binary_headers": assign_var("kafka_store_binary_headers"),
         "kafka_key_avro_deserialization": assign_var("kafka_key_avro_deserialization"),
         "kafka_ssl_ca_pem": assign_var("kafka_ssl_ca_pem"),
+        "kafka_security_protocol": assign_var("kafka_security_protocol"),
         "kafka_sasl_mechanism": assign_var("kafka_sasl_mechanism"),
         "import_service": assign_var("import_service"),
         "import_connection_name": assign_var("import_connection_name"),

tinybird/tb/modules/deployment.py

@@ -324,6 +324,7 @@ def create_deployment(
     DATAFILE_TYPE_TO_CONTENT_TYPE = {
         ".datasource": "text/plain",
         ".pipe": "text/plain",
+        ".connection": "text/plain",
     }
     project: Project = ctx.ensure_object(dict)["project"]
     client = ctx.ensure_object(dict)["client"]

tinybird/tb/modules/logout.py

@@ -0,0 +1,30 @@
+import click
+
+from tinybird.tb.modules.cli import cli
+from tinybird.tb.modules.config import CLIConfig
+from tinybird.tb.modules.feedback_manager import FeedbackManager
+
+
+@cli.command(name="logout", hidden=True)
+def logout() -> None:
+    """
+    Remove authentication from Tinybird.
+
+    This command will remove the authentication credentials from the CLI configuration.
+    """
+    conf = CLIConfig.get_project_config()
+    if workspace_name := conf.get("name", ""):
+        click.echo(FeedbackManager.highlight(message=f"» Logging out from {workspace_name}..."))
+    else:
+        click.echo(FeedbackManager.highlight(message="» Logging out..."))
+    conf.set_user_token("")
+    conf.set_token("")
+    conf.set_workspace_token(conf.get("id", ""), "")
+    conf.set_token_for_host("", conf.get("host", ""))
+    conf["tokens"] = {}
+    conf["name"] = ""
+    conf["id"] = ""
+    conf["user_id"] = ""
+    conf["user_email"] = ""
+    conf.persist_to_file()
+    click.echo(FeedbackManager.success(message="✓ Logged out!"))

tinybird/tb/modules/mock.py

@@ -74,7 +74,7 @@ async def mock(ctx: click.Context, datasource: str, rows: int, prompt: str) -> N
         sql = extract_xml(response, "sql")
         result = await tb_client.query(f"{sql} FORMAT JSON")
         data = result.get("data", [])[:rows]
-        if env == "build":
+        if env in ("build", "local"):
             persist_fixture_sql(datasource_name, sql, folder)
 
         fixture_path = persist_fixture(datasource_name, data, folder)

tinybird/tb/modules/project.py

@@ -9,7 +9,7 @@ from tinybird.tb.modules.datafile.parse_pipe import parse_pipe
 
 
 class Project:
-    extensions = ("datasource", "pipe")
+    extensions = ("datasource", "pipe", "connection")
 
     def __init__(self, folder: str):
         self.folder = folder

tinybird/tornado_template.py

@@ -1042,6 +1042,7 @@ VALID_FUNCTION_NAMES = {
     "custom_error",
     "backend_hint",
     "activate",
+    "disable",
     "sql_unescape",
     "max_threads",
     "tb_secret",

{tinybird-0.0.1.dev85.dist-info → tinybird-0.0.1.dev87.dist-info}/METADATA

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

{tinybird-0.0.1.dev85.dist-info → tinybird-0.0.1.dev87.dist-info}/RECORD

@@ -8,25 +8,25 @@ tinybird/feedback_manager.py,sha256=YSjtFDJvc8y66j2J0iIkb3SVzDdYAJbzFL-JPQ26pak,
 tinybird/git_settings.py,sha256=Sw_8rGmribEFJ4Z_6idrVytxpFYk7ez8ei0qHULzs3E,3934
 tinybird/prompts.py,sha256=rDIrkK5S1NJ0bIFjULsCYnGApv9Vc0jmR4tdRuJhM8c,32953
 tinybird/sql.py,sha256=LBi74GxhNAYTb6m2-KNGpAkguSKh7rcvBbERbE7nalA,46195
-tinybird/sql_template.py,sha256=hHz-EbHWlWVMS4e6hTpoieJ1qRAzzZLDq9-fnjT60dw,95732
+tinybird/sql_template.py,sha256=1qO6FaEe9rdLAKIORZsNk53XklLeP6oLnMhlw0PBUcI,96091
 tinybird/sql_template_fmt.py,sha256=KUHdj5rYCYm_rKKdXYSJAE9vIyXUQLB0YSZnUXHeBlY,10196
 tinybird/sql_toolset.py,sha256=32SNvxRFKQYWTvYPMJ_u3ukcd1hKZyEqx8T2cv2412w,14697
 tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
-tinybird/tornado_template.py,sha256=KmW_VD7y-NVqrc8YZKwIaxoJB0XpCcB2izdmxmtmApM,41944
+tinybird/tornado_template.py,sha256=jjNVDMnkYFWXflmT8KU_Ssbo5vR8KQq3EJMk5vYgXRw,41959
 tinybird/ch_utils/constants.py,sha256=aYvg2C_WxYWsnqPdZB1ZFoIr8ZY-XjUXYyHKE9Ansj0,3890
 tinybird/ch_utils/engine.py,sha256=AUAww-KjGOZg9h0IBlKA3FeacJYB4rOtqcTGJhFM-g8,40392
-tinybird/tb/__cli__.py,sha256=fmMvEpRU9NZ5mPO3UcdGwK4LO7Z1fNenvNHcaiN2NWI,251
-tinybird/tb/cli.py,sha256=KpJ_-V6xVEzcdPRPnHhEdh2EgRPCyhZnfJVqeqMsftI,964
+tinybird/tb/__cli__.py,sha256=LqvlRqPpw46Nvopi4aHBuiUk4LaKNklSwNZJhp6LJWs,251
+tinybird/tb/cli.py,sha256=qon0Lim-v5hHWWRW7TXaOoGo3p2kc5Xwgg3feI4mbMQ,998
 tinybird/tb/modules/auth.py,sha256=L1IatO2arRSzys3t8px8xVt8uPWUL5EVD0sFzAV_uVU,9022
-tinybird/tb/modules/build.py,sha256=46ZTxubughaq8QDV6yq9thatSVVY4yNqyzbfxRBcgDA,10446
+tinybird/tb/modules/build.py,sha256=-lRGBxKtuipmyl3pmiGcfp67fH1Ed-COfHAZKdgLIWo,10483
 tinybird/tb/modules/cicd.py,sha256=T0lb9u_bDdTUVe8TwNNb1qQ5KnSPHMVjqPfKF4BBNBw,5347
-tinybird/tb/modules/cli.py,sha256=Vwppm05Y9fKLY_Yo5pleMkcd8q7e-bqgXWGtNQmLan0,16447
+tinybird/tb/modules/cli.py,sha256=rE-0PtttuXvigKP1BkdgolP833pEhe-daK6l86bnaEw,16468
 tinybird/tb/modules/common.py,sha256=f5yKjb1dgUzR-SrC4UHnpHXYIu6INcTHwNO96jW8A9w,73201
-tinybird/tb/modules/config.py,sha256=wIsoIf2yMCT2JQ5f6nZr-WzB7Uuah9EqXf6Glxe_p-Q,11353
+tinybird/tb/modules/config.py,sha256=RWDrhH1OM9OWMixOyQfM6RVLBLb7Ul2Ar1tR8K46tOs,11322
 tinybird/tb/modules/copy.py,sha256=MAVqKip8_QhOYq99U_XuqSO6hCLJEh5sFtbhcXtI3SI,5802
-tinybird/tb/modules/create.py,sha256=EAPI-ZsY4Nr_FBehUMpM4y1HZRoWLgdtHQtb71WnauE,13137
+tinybird/tb/modules/create.py,sha256=KjotVfIQLfcPyQBykTHnPLn4ikrm6qqeMcbRE1d-6Jo,13280
 tinybird/tb/modules/datasource.py,sha256=dNCK9iCR2xPLfwqqwg2ixyE6NuoVEiJU2mBZBmOYrVY,16906
-tinybird/tb/modules/deployment.py,sha256=LzxOWjI9hSIXbNxR0BmQfXWBUHhVX2LoZUeChkcmg6c,17528
+tinybird/tb/modules/deployment.py,sha256=XAVt6a_vtH8sJ99Z9Ht1q5_ni7AFPRSeidfgfuJtqLw,17565
 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=Lv5MBGWNbIFTIy4nNSi0c61bmiQRv5J0nHRiiXUqDuc,68478
@@ -37,11 +37,12 @@ tinybird/tb/modules/llm_utils.py,sha256=nS9r4FAElJw8yXtmdYrx-rtI2zXR8qXfi1QqUDCf
 tinybird/tb/modules/local.py,sha256=revEEYP-Oq5nqURgmZmc0kt1_tEjM31dyUtcH5YbOuc,5842
 tinybird/tb/modules/local_common.py,sha256=jWNyoMBxamjEWLqyIhF50t3vz75CuZThekhDmP4cgQ4,3055
 tinybird/tb/modules/login.py,sha256=NB-evr7b00ChKPulX7c8YLN3EX6cr0CwALN0wroAq3o,6147
+tinybird/tb/modules/logout.py,sha256=ULooy1cDBD02-r7voZmhV7udA0ML5tVuflJyShrh56Y,1022
 tinybird/tb/modules/materialization.py,sha256=r8Q9HXcYEmfrEzP4WpiasCKDJdSkTPaAKJtZMoJKhi8,5749
-tinybird/tb/modules/mock.py,sha256=ASJTm4p11vkevGXCvU0h57aw7Inx03CTdxd5o2uzd8k,3870
+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/project.py,sha256=vxNEsc5gIQJM7ybXklnT7wna2M-GfbUybj4nKckrZM8,2949
+tinybird/tb/modules/project.py,sha256=ei0TIAuRksdV2g2FJqByuV4DPyivQGrZ42z_eQDNBgI,2963
 tinybird/tb/modules/regions.py,sha256=QjsL5H6Kg-qr0aYVLrvb1STeJ5Sx_sjvbOYO0LrEGMk,166
 tinybird/tb/modules/shell.py,sha256=a98W4L4gfrmxEyybtu6S4ENXrBYtgNASB5e_evuXQvI,13936
 tinybird/tb/modules/table.py,sha256=4XrtjM-N0zfNtxVkbvLDQQazno1EPXnxTyo7llivfXk,11035
@@ -56,7 +57,7 @@ tinybird/tb/modules/datafile/build.py,sha256=seGFSvmgyRrAM1-icsKBkuog3WccfGUYFTP
 tinybird/tb/modules/datafile/build_common.py,sha256=rT7VJ5mnQ68R_8US91DAtkusfvjWuG_NObOzNgtN_ko,4562
 tinybird/tb/modules/datafile/build_datasource.py,sha256=VjxaKKLZhPYt3XHOyMmfoqEAWAPI5D78T-8FOaN77MY,17355
 tinybird/tb/modules/datafile/build_pipe.py,sha256=Tf49kZmXub45qGcePFfqGO7p-FH5eYM46DtVI3AQJEc,11358
-tinybird/tb/modules/datafile/common.py,sha256=QhlQkYW5xLNEz_yZmMz1Dl8cnWG274Wu7QHVOIPcooM,79993
+tinybird/tb/modules/datafile/common.py,sha256=J9hXDdi27Quc0AkzoS2ChlEYwDBDePcW2sZ1qruZKdY,80091
 tinybird/tb/modules/datafile/diff.py,sha256=-0J7PsBO64T7LOZSkZ4ZFHHCPvT7cKItnJkbz2PkndU,6754
 tinybird/tb/modules/datafile/exceptions.py,sha256=8rw2umdZjtby85QbuRKFO5ETz_eRHwUY5l7eHsy1wnI,556
 tinybird/tb/modules/datafile/fixture.py,sha256=si-9LB-LdKQSWDtVW82xDrHtFfko5bgBG1cvjqqrcPU,1064
@@ -76,8 +77,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.dev85.dist-info/METADATA,sha256=Pi6jcS_5h9NY_EEyOI82gkHnVqqkSMWbR1bVMV23Ew0,2585
-tinybird-0.0.1.dev85.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
-tinybird-0.0.1.dev85.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev85.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev85.dist-info/RECORD,,
+tinybird-0.0.1.dev87.dist-info/METADATA,sha256=B58e0OeSZ-M9Z70rKj58zzouvl9WpGwl7pQy9tvjzyg,2585
+tinybird-0.0.1.dev87.dist-info/WHEEL,sha256=In9FTNxeP60KnTkGw7wk6mJPYd_dQSjEZmXdBdMCI-8,91
+tinybird-0.0.1.dev87.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev87.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev87.dist-info/RECORD,,