catworld-sdk

Cliente oficial Python para a API Catworld.

Instalação

pip install catworld-sdk

Início rápido

from catworld import CatworldClient

with CatworldClient("https://seu-catworld.exemplo.com", "cw_live_...") as client:
    result = client.query("SELECT * FROM banco_horas", dataset_id="<dataset-id>")
    df = result.dataframe
    print(df.head())

Autenticação

Passe a URL base do seu Catworld e um token de API gerado na interface:

client = CatworldClient(
    base_url="https://seu-catworld.exemplo.com",
    token="cw_live_...",
    timeout=30,  # timeout padrão das requisições HTTP (segundos)
)

Use como context manager (with) para fechar a conexão automaticamente, ou chame client.close() manualmente.

Métodos

query(sql, dataset_id=None, project_id=None, timeout=30, limit=10000)

Executa uma query SQL somente leitura.

Contexto de schema — passe dataset_id ou project_id para que o servidor resolva automaticamente tabelas não qualificadas para o schema correto:

# Com dataset_id: resolve tabelas do schema desse dataset
result = client.query("SELECT * FROM banco_horas", dataset_id="<dataset-id>")

# Com project_id: resolve tabelas de qualquer dataset do projeto
result = client.query("SELECT * FROM banco_horas", project_id="<project-id>")

# Sem contexto: você precisa qualificar manualmente com schema.tabela
result = client.query("SELECT * FROM meu_schema.banco_horas")

Se a mesma tabela existir em mais de um schema do contexto informado, o servidor retorna erro pedindo que você qualifique com schema.tabela.

Retorno:

query retorna um QueryResult: ele funciona como dict para compatibilidade, mas tambem expoe .rows, .columns e .dataframe.

result = client.query("SELECT * FROM banco_horas", dataset_id="<dataset-id>")

rows = result.rows
columns = result.columns
df = result.dataframe

Quando limit=None, o SDK usa streaming NDJSON para buscar todos os dados em uma unica requisicao. Nesse modo, o SDK nao envia timeout no payload; o servidor aplica o timeout fixo de 300 segundos para streams.

Para usar .dataframe, instale o extra com pandas:

pip install "catworld-sdk[dataframe]"

Parâmetros:

upload(path, dataset_id, mode="replace", key_column=None, table_id=None)

Envia um arquivo para um dataset.

result = client.upload(
    "dados.xlsx",
    dataset_id="<dataset-id>",
    mode="replace",       # "replace" sobrescreve, "append" adiciona, "upsert" atualiza/insere
    key_column="id",      # obrigatório para mode="upsert"
)
print(result["id"])

O método envia o arquivo, marca o upload como recebido e retorna os metadados do upload. A importação continua em background no Catworld. O arquivo pode ser .csv, .xlsx ou .xls.

Parâmetros:

projects()

Lista todos os projetos acessíveis pelo token.

projects = client.projects()
for p in projects:
    print(p["id"], p["name"])

datasets()

Lista todos os datasets acessíveis pelo token.

datasets = client.datasets()
for d in datasets:
    print(d["id"], d["name"], d["schemaName"])

tables(dataset_id)

Lista as tabelas de um dataset com colunas e tipos.

tables = client.tables("<dataset-id>")
for t in tables:
    origem = t.get("source")
    print(t["name"], origem["mode"] if origem else "catworld")

sources(dataset_id)

Lista as fontes conectadas de um dataset.

sources = client.sources("<dataset-id>")
for s in sources:
    print(s["id"], s["name"], s["mode"], s["connection"]["name"])

rows(table_id, limit=100)

Retorna as primeiras linhas de uma tabela pelo ID. Funciona tanto para tabelas materializadas no Catworld quanto para tabelas live; se a tabela for live, o servidor consulta o Postgres da fonte.

rows = client.rows("<table-id>", limit=50)
print(rows)

Tabelas live em query

Quando voce passa dataset_id, o SDK trata tabelas live como tabelas do dataset. Se a SQL referencia uma tabela live conhecida, query chama a fonte Postgres correta por baixo.

result = client.query("SELECT * FROM clientes", dataset_id="<dataset-id>")
df = result.dataframe
print(df.head())

Queries que misturam uma tabela live com tabela interna/extract nao sao roteadas automaticamente, porque elas precisariam cruzar bancos diferentes. Nesses casos, materialize a fonte como extract ou consulte a fonte live separadamente.

live_query(source_id, sql=None, timeout=30, limit=10000)

Executa uma consulta diretamente na fonte live Postgres. Normalmente prefira query(..., dataset_id=...); este metodo fica disponivel para casos em que voce ja tem o ID da fonte. Se sql for omitido, o servidor usa a consulta/tabela configurada na fonte.

result = client.live_query("<source-id>", "SELECT * FROM clientes LIMIT 100")
print(result.dataframe.head())

refresh_source(source_id)

Enfileira uma atualização de uma fonte do tipo cópia no Catworld.

job = client.refresh_source("<source-id>")
print(job["id"])

Exceções

from catworld import CatworldClient
from catworld.exceptions import PermissionDeniedError, QueryTimeoutError

try:
    result = client.query("SELECT * FROM banco_horas", dataset_id="<id>")
except PermissionDeniedError:
    print("Token sem acesso a este dataset")
except QueryTimeoutError:
    print("Query demorou demais, tente limitar com TOP ou WHERE")