Homologação de fornecedores: certidões automáticas por API

import base64
import os
import time
from datetime import date, timedelta
from pathlib import Path

import requests

BASE = "https://api.fiscalapi.com.br/api"
CHAVE = os.environ["FISCALAPI_KEY"]  # nunca no código-fonte
REGULAR = {"negativa", "positiva_com_efeitos_de_negativa"}
ANTECEDENCIA = timedelta(days=15)  # defina com compras


def consultar(rota: str, params: dict, tentativas: int = 4) -> dict:
    for _ in range(tentativas):
        resp = requests.get(f"{BASE}/{rota}", headers={"X-API-Key": CHAVE}, params=params, timeout=120)
        espera = int(resp.headers.get("Retry-After", "60"))
        if resp.status_code == 429:  # limite por minuto do plano
            time.sleep(espera)
            continue
        if resp.status_code >= 500:  # o órgão ou a API não responderam
            return {"source_status": {"status": "error", "error_code": f"HTTP {resp.status_code}"}}
        resp.raise_for_status()  # 400, 402, 403: repetir não resolve
        dados = resp.json()
        if dados["source_status"]["error_code"] == "EM_PROCESSAMENTO":  # CND Federal em emissão
            time.sleep(espera)
            continue
        return dados
    return {"source_status": {"status": "error", "error_code": "SEM_RESPOSTA_FINAL"}}


def homologar(cnpj_matriz: str, uf: str, pasta: Path) -> dict:
    rotas = {
        "cnd_federal": ("consultar-cnd-federal", {"cnpj": cnpj_matriz}),
        "cndt": ("consultar-cndt", {"cnpj": cnpj_matriz}),
        "crf_fgts": ("consultar-crf-fgts", {"cnpj": cnpj_matriz}),
        "cnd_estadual": ("consultar-cnd", {"uf": uf, "cnpj": cnpj_matriz}),
    }
    resultado = {}
    for nome, (rota, params) in rotas.items():
        dados = consultar(rota, params)
        certidao = dados.get("result")
        if dados["source_status"]["status"] != "success" or not certidao:
            resultado[nome] = {"situacao": "consultar_de_novo", "motivo": dados["source_status"]["error_code"]}
            continue
        arquivo = None
        if certidao.get("pdf_base64"):
            arquivo = pasta / f"{cnpj_matriz}-{nome}-{certidao['emissao_iso'] or date.today()}.pdf"
            arquivo.write_bytes(base64.b64decode(certidao["pdf_base64"]))
        resultado[nome] = {
            "situacao": "regular" if certidao["status"] in REGULAR else certidao["status"],
            "validade": certidao["validade_iso"],
            "protocolo": certidao["protocolo"],
            "pdf": str(arquivo) if arquivo else None,
            "request_id": dados["request_id"],
        }
    validades = [r["validade"] for r in resultado.values() if r.get("validade")]
    if validades:
        resultado["renovar_em"] = (date.fromisoformat(min(validades)) - ANTECEDENCIA).isoformat()
    return resultado