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

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, wait=False, poll_interval=2, timeout=None)

Envia um arquivo para um dataset.

result = client.upload(
    "dados.xlsx",
    dataset_id="<dataset-id>",
    table_id="<table-id>",  # recomendado sempre que o nome do arquivo pode variar
    mode="replace",         # "replace" sobrescreve, "append" adiciona, "upsert" atualiza/insere
    key_column="id",        # obrigatório para mode="upsert"
    wait=True,               # recomendado — veja abaixo
)
print(result["status"], result["rowCount"])

⚠️ Por padrão (wait=False) este método NÃO informa se a importação deu certo. Ele retorna assim que o arquivo é enfileirado — preview e import rodam em background, e um erro de schema, por exemplo, só aparece depois, olhando get_upload(upload_id) ou o painel. Recomendamos fortemente wait=True em qualquer script que precise saber o resultado: com isso, o método bloqueia até o import terminar (poll em GET /api/v1/uploads/{id}) e levanta UploadError com a mensagem de erro real do servidor se falhar (ex: "Schema incompatível. Esperado: ...") — em vez de falhar em silêncio.

O arquivo pode ser .csv, .xlsx ou .xls.

Parâmetros:

Erros comuns e como evitá-los:

get_upload(upload_id)

Consulta o estado atual de um upload — útil com wait=False, ou pra checar um upload antigo.

upload = client.get_upload(upload_id)
print(upload["status"], upload.get("errorMessage"))

check_append_compat(dataset_id, table_id, headers)

Confere, antes de enviar o arquivo, se os cabeçalhos batem com o schema físico da tabela — pega o erro mais comum de append/upsert sem gastar tempo/banda subindo o arquivo primeiro. Aceita cabeçalhos crus (com acento, espaço, etc.) e normaliza internamente do mesmo jeito que o servidor.

headers = ["Período Início", "Período Fim", "Relatório", "Posto", "Produto", "Total Abastecido"]
client.check_append_compat(dataset_id, table_id, headers)  # levanta ValidationError se não bater
client.upload("relatorio.xlsx", dataset_id=dataset_id, table_id=table_id, mode="append")

Só compara nomes e ordem de coluna — não valida tipo de dado (isso só o servidor descobre lendo o arquivo de verdade). Se a tabela ainda não existe (upload novo), não levanta erro.

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")