stdb-cli 0.1.0__py3-none-any.whl

stdb_cli/client.py

@@ -0,0 +1,16 @@
+from __future__ import annotations
+import os
+import httpx
+
+STDB_ENDPOINT = os.getenv("STDB_ENDPOINT", "https://stdb-mcp.azurecontainerapps.io")
+STDB_API_KEY = os.getenv("STDB_API_KEY", "")
+
+
+def get_client() -> httpx.Client:
+    if not STDB_API_KEY:
+        raise SystemExit("STDB_API_KEY environment variable is not set.")
+    return httpx.Client(
+        base_url=f"{STDB_ENDPOINT}/v1",
+        headers={"Authorization": f"Bearer {STDB_API_KEY}"},
+        timeout=60.0,
+    )

stdb_cli/commands/create_key.py

@@ -0,0 +1,12 @@
+import typer
+
+app = typer.Typer()
+
+
+@app.command()
+def create_key(
+    name: str = typer.Option(..., help="Key name"),
+    email: str = typer.Option(..., help="Email address"),
+):
+    """Create a new API key (admin operation — requires direct DB access via server)."""
+    typer.echo("create-key is an admin operation. Use the STDB portal or run scripts/create_key.py on the server.")

stdb_cli/commands/retrieve.py

@@ -0,0 +1,39 @@
+from __future__ import annotations
+import json
+import typer
+from stdb_cli.client import get_client
+
+app = typer.Typer()
+
+
+@app.command()
+def retrieve(
+    cf: str = typer.Option(..., help="Dataset cf name"),
+    cq: str = typer.Option(..., help="Dataset cq name"),
+    h3_resolution: int = typer.Option(..., "--res", help="H3 resolution (7-10)"),
+    timestamps: list[str] = typer.Option(..., "--ts", help="Timestamps, e.g. 2023-12-31"),
+    location_ids: list[int] = typer.Option([], "--location-id", help="Location IDs from search-location"),
+    bbox: list[float] = typer.Option([], "--bbox", help="Bounding box: west south east north"),
+    format: str = typer.Option("parquet", help="Output format: parquet or csv"),
+    ttl_seconds: int = typer.Option(3600, help="Artifact TTL in seconds"),
+    json_output: bool = typer.Option(False, "--json", help="Output raw JSON"),
+):
+    """Retrieve H3 spatial data as a downloadable artifact URL."""
+    payload = {
+        "cf": cf, "cq": cq, "h3_resolution": h3_resolution,
+        "timestamps": timestamps, "location_ids": location_ids,
+        "bbox": bbox if bbox else None,
+        "format": format, "ttl_seconds": ttl_seconds,
+    }
+    with get_client() as client:
+        resp = client.post("/retrieve", json=payload)
+        resp.raise_for_status()
+        data = resp.json()
+
+    if json_output:
+        typer.echo(json.dumps(data, ensure_ascii=False, indent=2))
+        return
+
+    typer.echo(f"Artifact URL: {data['artifact_url']}")
+    typer.echo(f"Expires in:  {data.get('expires_in_seconds', '?')}s")
+    typer.echo(f"Summary:     {data.get('summary', '')}")

stdb_cli/commands/search_dataset.py

@@ -0,0 +1,30 @@
+from __future__ import annotations
+import json
+import typer
+from stdb_cli.client import get_client
+
+app = typer.Typer()
+
+
+@app.command()
+def search_dataset(
+    query: str = typer.Argument(..., help="Natural language query, e.g. '台北租屋坪數'"),
+    top_k: int = typer.Option(10, help="Number of candidates before rerank"),
+    json_output: bool = typer.Option(False, "--json", help="Output raw JSON"),
+):
+    """Search for available datasets matching a query."""
+    with get_client() as client:
+        resp = client.post("/search-dataset", json={"query": query, "top_k": top_k})
+        resp.raise_for_status()
+        data = resp.json()
+
+    if json_output:
+        typer.echo(json.dumps(data, ensure_ascii=False, indent=2))
+        return
+
+    results = data.get("results", [])
+    if not results:
+        typer.echo("No results found.")
+        return
+    for r in results:
+        typer.echo(f"  {r['cf']}/{r['cq']}  {r.get('readable_name_zh', '')}  [{', '.join(r.get('available_timestamps', []))}]")

stdb_cli/commands/search_location.py

@@ -0,0 +1,30 @@
+from __future__ import annotations
+import json
+import typer
+from stdb_cli.client import get_client
+
+app = typer.Typer()
+
+
+@app.command()
+def search_location(
+    location_name: str = typer.Argument(..., help="One specific place name, e.g. '大安區'"),
+    top_k: int = typer.Option(3, help="Number of candidates to return"),
+    json_output: bool = typer.Option(False, "--json", help="Output raw JSON"),
+):
+    """Find location_id candidates for a place name."""
+    with get_client() as client:
+        resp = client.post("/search-location", json={"location_name": location_name, "top_k": top_k})
+        resp.raise_for_status()
+        data = resp.json()
+
+    if json_output:
+        typer.echo(json.dumps(data, ensure_ascii=False, indent=2))
+        return
+
+    candidates = data.get("candidates", [])
+    if not candidates:
+        typer.echo("No candidates found.")
+        return
+    for c in candidates:
+        typer.echo(f"  [{c['location_id']}] {c['name']}  confidence={c['confidence']}  source={c['source']}")

stdb_cli/main.py

@@ -0,0 +1,14 @@
+import typer
+from stdb_cli.commands.search_dataset import app as search_dataset_app
+from stdb_cli.commands.search_location import app as search_location_app
+from stdb_cli.commands.retrieve import app as retrieve_app
+from stdb_cli.commands.create_key import app as create_key_app
+
+app = typer.Typer(name="stdb", help="STDB CLI — spatio-temporal database client")
+app.add_typer(search_dataset_app, name="search-dataset")
+app.add_typer(search_location_app, name="search-location")
+app.add_typer(retrieve_app, name="retrieve")
+app.add_typer(create_key_app, name="create-key")
+
+if __name__ == "__main__":
+    app()

stdb_cli-0.1.0.dist-info/METADATA

@@ -0,0 +1,7 @@
+Metadata-Version: 2.4
+Name: stdb-cli
+Version: 0.1.0
+Requires-Python: >=3.12
+Requires-Dist: httpx>=0.28.0
+Requires-Dist: python-dotenv>=1.2.2
+Requires-Dist: typer>=0.15.0

stdb_cli-0.1.0.dist-info/RECORD

@@ -0,0 +1,12 @@
+stdb_cli/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+stdb_cli/client.py,sha256=LWHJis-JhfjnY6mmxB8XIRhYRR2hVOg0MXlrwpn4Fk8,478
+stdb_cli/main.py,sha256=7qyJK-dTl-6BvXCmKbbJ_5rp8FCICfJ640vLnZ4Xe5A,612
+stdb_cli/commands/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
+stdb_cli/commands/create_key.py,sha256=ObqWZwKR6c4qWPI57F9FMtnMpVEJeUhk6pwAv49t9Q0,387
+stdb_cli/commands/retrieve.py,sha256=UwVZYL-y3WUQfcPORuBSp687LQ9ox5Flp2PoieVYQHM,1621
+stdb_cli/commands/search_dataset.py,sha256=5_PM_29okvfFeCSXENWNgCs7sBbu-7OLW-P8eK3FoVA,1015
+stdb_cli/commands/search_location.py,sha256=N2Eg1FqAFdeFhC8YLT4G76egGZoWQGBJLud1XLG2xIE,1023
+stdb_cli-0.1.0.dist-info/METADATA,sha256=0uPh6xNyjBuNGeQeEEwnQkXbVvBp3-S7Hdv7HR0V6n4,170
+stdb_cli-0.1.0.dist-info/WHEEL,sha256=QccIxa26bgl1E6uMy58deGWi-0aeIkkangHcxk2kWfw,87
+stdb_cli-0.1.0.dist-info/entry_points.txt,sha256=wQDjD1a0pXOdOhXvawVIBHMfUJt6cdQQj0wXOeSxRLE,43
+stdb_cli-0.1.0.dist-info/RECORD,,

stdb_cli-0.1.0.dist-info/WHEEL

@@ -0,0 +1,4 @@
+Wheel-Version: 1.0
+Generator: hatchling 1.29.0
+Root-Is-Purelib: true
+Tag: py3-none-any

stdb_cli-0.1.0.dist-info/entry_points.txt

@@ -0,0 +1,2 @@
+[console_scripts]
+stdb = stdb_cli.main:app