"""
AIKDATA_AI SDK
Auto-generated — v1.0
"""
import requests


class AikdataAiClient:
    def __init__(self, token: str, base_url: str = "https://ms.aikdata.com/api/gw/aikdata_ai"):
        self.token = token
        self.base_url = base_url.rstrip("/")
        self.headers = {
            "Authorization": f"Bearer {self.token}",
            "Content-Type": "application/json",
            "Accept": "application/json",
        }

    # ── Trámites ──

    def generar_token(self, body: dict = None) -> dict:
        """Create a new tramite and receive a unique verification token"""
        return self._request("POST", f"/generar-token", json_body=body)

    def consultar_trámite(self, token: str) -> dict:
        """Query the current status of a tramite by its token"""
        return self._request("GET", f"/consultar-tramite/{token}")

    # ── Aprobación ──

    def aprobar_trámite(self, body: dict = None) -> dict:
        """Approve or reject a pending tramite"""
        return self._request("POST", f"/aprobar", json_body=body)

    # ── Análisis ──

    def analizar_k_y_c(self, body: dict = None) -> dict:
        """AI-powered analysis of a KYC/Due-Diligence document"""
        return self._request("POST", f"/analizar-kyc", json_body=body)


    # ── HTTP helper ──

    def _request(self, method: str, path: str, params: dict = None, json_body: dict = None) -> dict:
        url = self.base_url + path
        resp = requests.request(
            method, url, headers=self.headers, params=params, json=json_body, timeout=30
        )
        return {"status": resp.status_code, "data": resp.json()}