tinybird 0.0.1.dev61__tar.gz → 0.0.1.dev63__tar.gz

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/PKG-INFO

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

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/config.py

@@ -49,12 +49,6 @@ LEGACY_HOSTS = {
     "https://ui.us-west-2.aws.tinybird.co": "https://app.tinybird.co/aws/us-west-2",
     "https://ui.eu-central-1.aws.tinybird.co": "https://app.tinybird.co/aws/eu-central-1",
     "https://ui.ap-east.aws.tinybird.co": "https://app.tinybird.co/aws/ap-east",
-    "https://inditex-tech.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-tech",
-    "https://inditex-c-stg.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-c-stg",
-    "https://inditex-c-pro.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-c-pro",
-    "https://inditex-z-stg.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-z-stg",
-    "https://inditex-rt-pro.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-rt-pro",
-    "https://inditex-pro.tinybird.co": "https://app.inditex.tinybird.co/gcp/inditex-pro",
     "https://ui.split.tinybird.co": "https://app.tinybird.co/aws/split-us-east",
     "https://ui.split.us-west-2.aws.tinybird.co": "https://app.tinybird.co/aws/split-us-west-2",
     "https://api.split.tinybird.co": "https://app.tinybird.co/aws/split-us-east",
@@ -116,3 +110,7 @@ class FeatureFlags:
     @classmethod
     def is_localhost(cls) -> bool:
         return "SET_LOCALHOST" in environ
+
+    @classmethod
+    def enable_snowflake_connector_command(cls) -> bool:
+        return "ENABLE_SNOWFLAKE_CONNECTOR_COMMAND" in environ

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/feedback_manager.py

@@ -420,6 +420,9 @@ class FeedbackManager:
     error_updating_tag = error_message("Error updating tag: {error}")
     error_tag_generic = error_message("There was an issue updating tags. {error}")
     error_tag_not_found = error_message("Tag {tag_name} not found.")
+    error_snowflake_connector_not_enabled = error_message(
+        "🚨🚨🚨 [DEPRECATED] Snowflake connector is deprecated. Can't create the connector. **"
+    )
 
     info_incl_relative_path = info_message("** Relative path {path} does not exist, skipping.")
     info_ignoring_incl_file = info_message(

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/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.dev61'
-__revision__ = '0fed0e1'
+__version__ = '0.0.1.dev63'
+__revision__ = '4d0f19d'

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb/modules/config.py

@@ -321,10 +321,6 @@ class CLIConfig:
         """
         working_dir = working_dir or os.getcwd()
         path: str = os.path.join(working_dir, ".tinyb")
-
-        if not os.path.exists(path):
-            path = os.path.join(os.path.dirname(working_dir), ".tinyb")
-
         return CLIConfig(path, parent=CLIConfig.get_global_config())
 
     @staticmethod

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb/modules/datafile/parse_pipe.py

@@ -47,6 +47,8 @@ def parse_pipe(
             finally:
                 raise e
         for node in doc.nodes:
+            if "type" in node:
+                node["type"] = node["type"].lower()
             sql = node.get("sql", "")
             if sql.strip()[0] == "%":
                 # Note(eclbg): not sure what test_mode is for. I think it does something like using placeholder values

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb/modules/endpoint.py

@@ -10,6 +10,7 @@ from urllib.parse import urlencode
 
 import click
 import humanfriendly
+import pyperclip
 import requests
 from click import Context
 
@@ -54,7 +55,7 @@ async def endpoint_ls(ctx: Context, match: str, format_: str):
         tk = get_name_version(t["name"])
         if pattern and not pattern.search(tk["name"]):
             continue
-        token = get_endpoint_token(tokens, tk["name"])
+        token = get_endpoint_token(tokens, tk["name"]) or client.token
         endpoint_url = build_endpoint_url(client, tk["name"], token)
         table_human_readable.append((tk["name"], t["updated_at"][:-7], len(t["nodes"]), endpoint_url))
         table_machine_readable.append(
@@ -151,29 +152,23 @@ async def endpoint_data(ctx: Context, pipe: str, query: str, format_: str):
 
 @endpoint.command(name="url")
 @click.argument("pipe")
+@click.option(
+    "--language",
+    default="http",
+    help="Language used for sending the request. Options: http, python, curl, javascript, rust, go",
+)
 @click.pass_context
 @coro
-async def endpoint_url(ctx: Context, pipe: str):
+async def endpoint_url(ctx: Context, pipe: str, language: str):
     """Print the URL of an endpoint"""
+    if language != "http":
+        click.echo(FeedbackManager.highlight(message=f"\n» Generating snippet for {language} language"))
     client: TinyB = ctx.ensure_object(dict)["client"]
     tokens = await client.tokens()
-    token = get_endpoint_token(tokens, pipe)
-    click.echo(build_endpoint_url(client, pipe, token))
-
-
-def build_endpoint_url(tb_client: TinyB, pipe_name: str, token: Optional[str]) -> Optional[str]:
-    try:
-        token = token or tb_client.token
-        example_params = {
-            "format": "json",
-            "pipe": pipe_name,
-            "q": "",
-            "token": token,
-        }
-        response = requests.get(f"{tb_client.host}/examples/query.http?{urlencode(example_params)}")
-        return response.text.replace("http://localhost:8001", tb_client.host)
-    except Exception:
-        return None
+    token = get_endpoint_token(tokens, pipe) or client.token
+    click.echo(build_endpoint_snippet(client, pipe, token, language))
+    if language != "http":
+        click.echo(FeedbackManager.success(message="\n✓ Code snippet copied to clipboard!\n"))
 
 
 def get_endpoint_token(tokens: List[Dict[str, Any]], pipe_name: str) -> Optional[str]:
@@ -272,3 +267,104 @@ async def endpoint_stats(ctx: click.Context, pipes: Tuple[str, ...], format_: st
             click.echo(json.dumps({"pipes": table_machine_readable}, indent=2))
         else:
             echo_safe_humanfriendly_tables_format_smart_table(table_human_readable, column_names=columns)
+
+
+def build_endpoint_snippet(tb_client: TinyB, pipe_name: str, token: Optional[str], language: str) -> Optional[str]:
+    endpoint_url = build_endpoint_url(tb_client, pipe_name, token)
+    if language == "http":
+        return endpoint_url
+
+    snippet = None
+    if language == "python":
+        snippet = build_python_snippet(endpoint_url, token)
+    elif language == "curl":
+        snippet = build_curl_snippet(endpoint_url)
+    elif language == "javascript":
+        snippet = build_javascript_snippet(endpoint_url, token)
+    elif language == "rust":
+        snippet = build_rust_snippet(endpoint_url, token)
+    elif language == "go":
+        snippet = build_go_snippet(endpoint_url, token)
+
+    if not snippet:
+        raise CLIPipeException(FeedbackManager.error(message=f"Language {language} not supported"))
+
+    pyperclip.copy(snippet.strip())
+    return snippet
+
+
+def build_endpoint_url(tb_client: TinyB, pipe_name: str, token: str) -> str:
+    example_params = {
+        "format": "json",
+        "pipe": pipe_name,
+        "q": "",
+        "token": token,
+    }
+    response = requests.get(f"{tb_client.host}/examples/query.http?{urlencode(example_params)}")
+    return response.text.replace("http://localhost:8001", tb_client.host)
+
+
+def build_python_snippet(endpoint_url: str, token: str) -> str:
+    endpoint_url = endpoint_url.replace(f"token={token}", "token={{token}}")
+    return f"""
+import requests
+
+token = "{token}"
+url = "{endpoint_url}"
+response = requests.get(url)
+print(response.json())
+    """
+
+
+def build_curl_snippet(endpoint_url: str) -> str:
+    return f"""
+curl -X GET "{endpoint_url}"
+    """
+
+
+def build_javascript_snippet(endpoint_url: str, token: str) -> str:
+    endpoint_url = endpoint_url.replace(f"token={token}", "token=${token}")
+    return f"""
+const token = "{token}";
+fetch(`{endpoint_url}`)
+.then(response => response.json())
+.then(data => console.log(data));
+    """
+
+
+def build_rust_snippet(endpoint_url: str) -> str:
+    return f"""
+use reqwest::Client;
+
+let client = Client::new();
+let response = client.get("{endpoint_url}").send().await?;
+    """
+
+
+def build_go_snippet(endpoint_url: str) -> str:
+    return f"""
+package main
+
+import (
+    "fmt"
+    "io"
+    "log"
+    "net/http"
+)
+
+func main() {{
+    url := "{endpoint_url}"
+    resp, err := http.Get(url)
+    if err != nil {{
+        log.Fatal(err)
+    }}
+    defer resp.Body.Close()
+
+    body, err := io.ReadAll(resp.Body)
+    if err != nil {{
+        log.Fatal(err)
+    }}
+
+    fmt.Println(string(body))
+}}
+    """

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb/modules/local_common.py

@@ -5,8 +5,9 @@ from typing import Any, Dict
 
 import requests
 
-from tinybird.client import AuthException, AuthNoTokenException, TinyB
+from tinybird.client import AuthNoTokenException, TinyB
 from tinybird.tb.modules.config import CLIConfig
+from tinybird.tb.modules.exceptions import CLIException
 
 TB_IMAGE_NAME = "tinybirdco/tinybird-local:beta"
 TB_CONTAINER_NAME = "tinybird-local"
@@ -32,7 +33,7 @@ async def get_tinybird_local_config(config_obj: Dict[str, Any], build: bool = Fa
         # ruff: noqa: ASYNC210
         tokens = requests.get(f"{TB_LOCAL_HOST}/tokens").json()
     except Exception:
-        raise AuthException("Error: Tinybird local is not running. Please run `tb local start` first.")
+        raise CLIException("Tinybird local is not running. Please run `tb local start` first.")
 
     user_token = tokens["user_token"]
     admin_token = tokens["admin_token"]

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb_cli_modules/common.py

@@ -593,8 +593,6 @@ def format_host(host: str, subdomain: Optional[str] = None) -> str:
     'https://api.us-east.tinybird.co'
     >>> format_host('https://ui.us-east.tinybird.co', subdomain='api')
     'https://api.us-east.tinybird.co'
-    >>> format_host('https://inditex-rt-pro.tinybird.co', subdomain='ui')
-    'https://inditex-rt-pro.tinybird.co'
     >>> format_host('https://cluiente-tricky.tinybird.co', subdomain='api')
     'https://cluiente-tricky.tinybird.co'
     """

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird/tb_cli_modules/connection.py

@@ -13,6 +13,7 @@ import click
 from click import Context
 
 from tinybird.client import DoesNotExistException, TinyB
+from tinybird.config import FeatureFlags
 from tinybird.feedback_manager import FeedbackManager
 from tinybird.tb_cli_modules.cli import cli
 from tinybird.tb_cli_modules.common import (
@@ -193,7 +194,9 @@ async def connection_create_kafka(
     click.echo(FeedbackManager.success_connection_created(id=id))
 
 
-@connection_create.command(name="snowflake", short_help="Creates a Snowflake connection in the current workspace")
+@connection_create.command(
+    name="snowflake", short_help="Creates a Snowflake connection in the current workspace", hidden=True
+)
 @click.option("--account", help="The account identifier of your Snowflake account (e.g. myorg-account123)")
 @click.option("--username", help="The Snowflake user you want to use for the connection")
 @click.option("--password", help="The Snowflake password of the chosen user")
@@ -242,6 +245,11 @@ async def connection_create_snowflake(
     $ tb connection create snowflake
     """
 
+    snowflake_connector_enabled = FeatureFlags.enable_snowflake_connector_command()
+    if not snowflake_connector_enabled:
+        click.echo(FeedbackManager.error_snowflake_connector_not_enabled())
+        return
+
     obj: Dict[str, Any] = ctx.ensure_object(dict)
     client: TinyB = obj["client"]
 

{tinybird-0.0.1.dev61 → tinybird-0.0.1.dev63}/tinybird.egg-info/PKG-INFO

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