tinybird 0.0.1.dev61__py3-none-any.whl → 0.0.1.dev62__py3-none-any.whl

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.dev62'
+__revision__ = '472a5f2'

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

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

{tinybird-0.0.1.dev61.dist-info → tinybird-0.0.1.dev62.dist-info}/RECORD

@@ -15,7 +15,7 @@ tinybird/syncasync.py,sha256=IPnOx6lMbf9SNddN1eBtssg8vCLHMt76SuZ6YNYm-Yk,27761
 tinybird/tornado_template.py,sha256=FL85SMPq2dH4JqKovmSbaolGdEzwOO91NqOzqXo2Qr0,41863
 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=J74jDM5nA1KeHEC9VTPkVN7HFbtSPRviL37Zz3u1ywM,251
+tinybird/tb/__cli__.py,sha256=Y3-22BYsp8UsmnVrY3OWWytK2xzHpGUrQRjrKGXUBhM,251
 tinybird/tb/cli.py,sha256=FD1pfbzu9YHJHEG6Vtn_EwPLTYhwqw-I6AxXeTaRHU8,926
 tinybird/tb/modules/auth.py,sha256=EzRWFmwRkXNhUmRaruEVFLdkbUg8xMSix0cAWl5D4Jg,9029
 tinybird/tb/modules/build.py,sha256=zo7qexonQ5vDNHZ1u7AK_jdo62hMfG5vWnoQBe_2oyw,9042
@@ -27,7 +27,7 @@ tinybird/tb/modules/copy.py,sha256=Aq6wh_wjRiyLQtEOKF9pKLPgJhSvbGTFWIw_LJB0t0U,5
 tinybird/tb/modules/create.py,sha256=Q7ZY4oUmEtAYpUbBLjjyiUZ6IKlkU81qJJrmdgjduOk,13914
 tinybird/tb/modules/datasource.py,sha256=TQ4wSag3CCw34d54FEXPJFGLQNYyNqv2nQbU6QT9uAE,14725
 tinybird/tb/modules/deployment.py,sha256=E3YdQgFmX7rPlgPUw9yAB0jyl7m1MQcqiQF7uHs8wK4,14272
-tinybird/tb/modules/endpoint.py,sha256=9arqN1JQCMb0Nd3-EJ7lukOYkGHHCpQmiiZpp5FqPhc,9432
+tinybird/tb/modules/endpoint.py,sha256=M2ZerljZHLJHvDfH30r4Yma3-JGExg2PlRunb2l4r-A,11931
 tinybird/tb/modules/exceptions.py,sha256=4A2sSjCEqKUMqpP3WI00zouCWW4uLaghXXLZBSw04mY,3363
 tinybird/tb/modules/feedback_manager.py,sha256=mrw5tdYycfvg6WLXlM0KIjfJardm_aNpnJkUg2vH0cA,68463
 tinybird/tb/modules/fmt.py,sha256=poh6_cwVGSf-sBu6LKWuO2TANL_J8Sgm25sPpwxa3Aw,3558
@@ -74,8 +74,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.dev61.dist-info/METADATA,sha256=2DnZbQC6aEzXMKMLff8KgszlSLd9r9HhEI3MsNcLczM,2482
-tinybird-0.0.1.dev61.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
-tinybird-0.0.1.dev61.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
-tinybird-0.0.1.dev61.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
-tinybird-0.0.1.dev61.dist-info/RECORD,,
+tinybird-0.0.1.dev62.dist-info/METADATA,sha256=TSfflXA3BB3Bmp-J9OHaPfaXsq4ft7QlDsFiRbpBmVI,2482
+tinybird-0.0.1.dev62.dist-info/WHEEL,sha256=G16H4A3IeoQmnOrYV4ueZGKSjhipXx8zc8nu9FGlvMA,92
+tinybird-0.0.1.dev62.dist-info/entry_points.txt,sha256=LwdHU6TfKx4Qs7BqqtaczEZbImgU7Abe9Lp920zb_fo,43
+tinybird-0.0.1.dev62.dist-info/top_level.txt,sha256=VqqqEmkAy7UNaD8-V51FCoMMWXjLUlR0IstvK7tJYVY,54
+tinybird-0.0.1.dev62.dist-info/RECORD,,