Getting Started

import os
import time

import requests

BASE = os.environ["BRITIVE_TENANT_URL"].rstrip("/")
TOKEN = os.environ["BRITIVE_TOKEN"]
PATH = f"{BASE}/api/bizintel/analytics/query"
HEADERS = {"Authorization": f"TOKEN {TOKEN}", "Content-Type": "application/json"}

SQL = "SELECT * FROM users_latest LIMIT 10"
POLL_SECONDS = 10
TIMEOUT_SECONDS = 300


def run_query(sql: str) -> str:
  """Submit the query and return its execution id."""
  response = requests.post(PATH, json={"sql": sql}, headers=HEADERS, timeout=30)
  response.raise_for_status()
  return response.json()["query_execution_id"]


def wait_for_results(query_execution_id: str) -> str:
  """Poll until the query settles, then return a signed download URL."""
  deadline = time.monotonic() + TIMEOUT_SECONDS
  while time.monotonic() < deadline:
      response = requests.get(
          f"{PATH}/{query_execution_id}",
          headers={"Authorization": f"TOKEN {TOKEN}"},
          timeout=30,
      )
      response.raise_for_status()
      body = response.json()
      status = body["status"]

      if status == "SUCCEEDED":
          return body["download_url"]
      if status in ("FAILED", "CANCELLED"):
          raise RuntimeError(f"query {status}: {body.get('error', 'no reason given')}")

      time.sleep(POLL_SECONDS)

  raise TimeoutError(f"query {query_execution_id} did not finish in {TIMEOUT_SECONDS}s")


if __name__ == "__main__":
  qid = run_query(SQL)
  print(f"submitted {qid}")

  url = wait_for_results(qid)
  # No auth header here - the presigned URL carries its own authorization.
  csv = requests.get(url, timeout=60)
  csv.raise_for_status()

  with open("results.csv", "wb") as handle:
      handle.write(csv.content)
  print(f"wrote results.csv ({len(csv.content)} bytes)")

#!/usr/bin/env bash
set -euo pipefail

BASE="${BRITIVE_TENANT_URL%/}"
PATH_Q="$BASE/api/bizintel/analytics/query"
AUTH="Authorization: TOKEN $BRITIVE_TOKEN"
SQL='SELECT * FROM users_latest LIMIT 10'
POLL_SECONDS=10
TIMEOUT_SECONDS=300

qid=$(curl -sS -X POST "$PATH_Q" \
-H "$AUTH" -H "Content-Type: application/json" \
-d "$(jq -nc --arg sql "$SQL" '{sql: $sql}')" | jq -r '.query_execution_id')
echo "submitted $qid"

for _ in $(seq 1 $(( TIMEOUT_SECONDS / POLL_SECONDS ))); do
body=$(curl -sS "$PATH_Q/$qid" -H "$AUTH")
status=$(jq -r '.status' <<<"$body")

case "$status" in
  SUCCEEDED)
    # No auth header - the presigned URL carries its own authorization.
    curl -sS -o results.csv "$(jq -r '.download_url' <<<"$body")"
    echo "wrote results.csv"
    exit 0
    ;;
  FAILED|CANCELLED)
    echo "query $status: $(jq -r '.error // "no reason given"' <<<"$body")" >&2
    exit 1
    ;;
esac

sleep "$POLL_SECONDS"
done

echo "query $qid did not finish in time" >&2
exit 1