tinybird 0.0.1.dev61__tar.gz → 0.0.1.dev62__tar.gz

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

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

@@ -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