python.py

"""CFE API - Python example.

Requirements: requests (`pip install requests`).
"""
import os
import requests

API = "https://cfe-api.fly.dev"
API_KEY = os.environ["CFE_API_KEY"]


def consulta(rpu: str, nombre: str, provider: str | None = None,
             total_a_pagar: str | None = None) -> dict:
    body = {"rpu": rpu, "nombre": nombre}
    if provider:
        body["provider"] = provider
    # total_a_pagar (monto actual a pagar, sin decimales) es requerido cuando el
    # proveedor efectivo es "micfe": CFE valida rpu + nombre (razón social) + total.
    if total_a_pagar:
        body["total_a_pagar"] = total_a_pagar
    r = requests.post(
        f"{API}/api/v1/consulta",
        headers={"X-API-Key": API_KEY},
        json=body,
        timeout=120,
    )
    r.raise_for_status()
    return r.json()


def balance() -> dict:
    r = requests.get(f"{API}/api/v1/balance", headers={"X-API-Key": API_KEY})
    r.raise_for_status()
    return r.json()


if __name__ == "__main__":
    result = consulta("123456789012", "JUAN PEREZ")
    data = result["data"]
    print(f"Tarifa: {data['tarifa']}  Anual: {data['annual_kwh']} kWh")
    print(f"Hilos: {data.get('hilos') or 'no disponible'}")
    print(f"Cobrado: {result['charged_cents']/100:.2f} MXN  Cache: {result['cached']}")