tinybird 0.0.1.dev85__tar.gz → 0.0.1.dev87__tar.gz

{tinybird-0.0.1.dev85 → tinybird-0.0.1.dev87}/PKG-INFO

@@ -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 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev87/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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-0.0.1.dev85 → tinybird-0.0.1.dev87}/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 → tinybird-0.0.1.dev87}/tinybird.egg-info/PKG-INFO

@@ -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 → tinybird-0.0.1.dev87}/tinybird.egg-info/SOURCES.txt

@@ -46,6 +46,7 @@ tinybird/tb/modules/llm_utils.py
 tinybird/tb/modules/local.py
 tinybird/tb/modules/local_common.py
 tinybird/tb/modules/login.py
+tinybird/tb/modules/logout.py
 tinybird/tb/modules/materialization.py
 tinybird/tb/modules/mock.py
 tinybird/tb/modules/pipe.py