# Dados da Tabela 10331 do IBGE SIDRA (https://apisidra.ibge.gov.br)
# Requisitos: pip install requests pandas
import pandas as pd
import requests

url = "https://apisidra.ibge.gov.br/values/t/10331/n1/all/v/all/p/all/c2/all/h/y/f/a/d/m"

resposta = requests.get(url, timeout=120)
resposta.raise_for_status()
linhas = resposta.json()

# A primeira linha traz os nomes amigáveis das colunas (/h/y)
df = pd.DataFrame(linhas[1:]).rename(columns=linhas[0])
print(df.head())

# Dados da Tabela 10331 do IBGE SIDRA via sidra-fetcher
# https://github.com/Quantilica/sidra-fetcher
# Requisitos: pip install httpx polars
#             pip install "sidra-fetcher @ git+https://github.com/Quantilica/sidra-fetcher.git"
import httpx
import polars as pl
from sidra_fetcher.sidra import Parametro

parametro = Parametro(
    agregado="10331",
    territorios={"1": ["all"]},
    variaveis=["all"],
    periodos=["all"],
    classificacoes={"2": ["all"]},
)

resposta = httpx.get(parametro.url(), timeout=120)
resposta.raise_for_status()
linhas = resposta.json()

# A primeira linha traz os nomes amigáveis das colunas (/h/y)
df = pl.DataFrame(linhas[1:]).rename(linhas[0])
print(df.head())

[[tabelas]]
tabela_sidra = "10331"
variables = ["allxp"]
territories = {1 = []}
classifications = {537 = [], 2 = ["allxt"], 86 = [], 386 = []}
# c537 (Tempo habitual de deslocamento do domicílio para o trabalho principal) não tem categoria "Total"; [] baixa todas as categorias.
# c86 (Cor ou raça) não tem categoria "Total"; [] baixa todas as categorias.
# c386 (Classes de rendimento nominal mensal domiciliar per capita) não tem categoria "Total"; [] baixa todas as categorias.

Cole o bloco acima em um fetch.toml existente do seu plugin sidra-sql, ou instale e rode um pipeline novo já configurado para esta tabela:

sidra-sql plugin install <url-do-seu-repo> --alias meu_plugin
sidra-sql run meu_plugin tabela_10331
sidra-sql export meu_plugin tabela_10331

-- Tabela 10331 — Pessoas de 10 anos ou mais de idade residentes em domicílios particulares, ocupadas na semana de referência, que, no trabalho principal, trabalhavam fora do domicílio, por tempo habitual de deslocamento do domicílio para o trabalho principal, sexo, cor ou raça e classes de rendimento nominal mensal domiciliar per capita
-- Consulta ao banco carregado pelo sidra-sql (schema ibge_sidra)
-- https://github.com/Quantilica/sidra-sql
-- Pressupõe `sidra-sql run` já executado para a tabela 10331.
SELECT
    p.codigo AS periodo,
    p.ano,
    l.nc  AS nivel_territorial,
    l.d1c AS localidade_codigo,
    l.d1n AS localidade_nome,
    dim.d2c AS variavel_codigo,
    dim.d2n AS variavel,
    dim.mn  AS unidade,
    dim.d4c AS tempo_habitual_de_deslocamento_do_domici_codigo,
    dim.d4n AS tempo_habitual_de_deslocamento_do_domici,
    dim.d5c AS sexo_codigo,
    dim.d5n AS sexo,
    dim.d6c AS cor_ou_raca_codigo,
    dim.d6n AS cor_ou_raca,
    dim.d7c AS classes_de_rendimento_nominal_mensal_dom_codigo,
    dim.d7n AS classes_de_rendimento_nominal_mensal_dom,
    CASE WHEN d.v ~ '^-?[0-9]' THEN d.v::numeric END AS valor
FROM ibge_sidra.dados AS d
JOIN ibge_sidra.periodo    AS p   ON d.periodo_id = p.id
JOIN ibge_sidra.dimensao   AS dim ON d.dimensao_id = dim.id
JOIN ibge_sidra.localidade AS l   ON d.localidade_id = l.id
WHERE d.tabela_sidra_id = '10331'
  AND d.ativo = TRUE
  AND (l.nc = '1')
ORDER BY p.codigo, l.d1c;