rips_manager/app/routes/transaccion.py

import json as json_lib
import httpx
from datetime import datetime
from fastapi import APIRouter, Request, Form, Depends
from fastapi.responses import JSONResponse
from app.database import get_connection
from app.auth import get_current_user
from app.services.firebird_service import get_firebird_from_config
from app.services.json_generator import generar_rda_paciente, agrupar_por_recepcion
from app.services.api_client import get_tns_token, TNS_BASE

router = APIRouter(prefix="/transaccion", tags=["transaccion"])


def _cfg_and_fb():
    conn = get_connection()
    cfg = {r["key"]: r["value"] for r in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()
    return cfg


def _query_rows(cfg, query_text, factura, fecha_inicio, fecha_fin):
    import re as _re
    fb, ok, msg = get_firebird_from_config(cfg)
    if not ok:
        return None, msg, None
    nums = _re.findall(r'\d+', factura or "")
    num_val = int(nums[-1]) if nums else (factura or "")
    prefix = _re.sub(r'[\d\s]', '', factura or "").strip().upper()
    params = {"fecha_ini": f"{fecha_inicio} 00:00:00", "fecha_fin": f"{fecha_fin} 23:59:59"}
    if ":num_factura" in query_text:
        params["num_factura"] = num_val
    if ":prefijo" in query_text:
        params["prefijo"] = prefix
    ok2, err, rows = fb.execute_query(query_text, params)
    fb.disconnect()
    if not ok2:
        return None, err, None
    # Si el usuario escribió prefijo (ej. "LHXC03726") y el query devuelve PREFIJO, filtramos
    if prefix and rows and "PREFIJO" in rows[0]:
        rows = [r for r in rows if str(r.get("PREFIJO") or "").strip().upper() == prefix]
    return rows, None, fb


def _is_sent(idrecepcion: int) -> dict:
    """Returns the latest envio record for this idrecepcion, or None."""
    conn = get_connection()
    row = conn.execute(
        "SELECT status, mensaje_tns, created_at FROM envios WHERE idrecepcion=? AND tipo='transaccion' ORDER BY id DESC LIMIT 1",
        (idrecepcion,)
    ).fetchone()
    conn.close()
    if row:
        return {"status": row["status"], "mensaje": row["mensaje_tns"], "at": row["created_at"]}
    return None


@router.get("")
async def transaccion_page(request: Request, user: dict = Depends(get_current_user)):
    conn = get_connection()
    queries = conn.execute(
        "SELECT * FROM queries WHERE query_type = 'transaccion' ORDER BY name"
    ).fetchall()
    configs = {row["key"]: row["value"] for row in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()
    return request.app.state.templates.TemplateResponse("transaccion.html", {
        "request": request, "user": user, "queries": queries, "configs": configs,
    })


@router.post("/preview")
async def preview_transaccion(
    request: Request, user: dict = Depends(get_current_user),
    query_id: int = Form(...), factura: str = Form(""),
    fecha_inicio: str = Form(...), fecha_fin: str = Form(...),
    contrato: str = Form(""),
):
    conn = get_connection()
    q = conn.execute("SELECT * FROM queries WHERE id = ?", (query_id,)).fetchone()
    cfg = {r["key"]: r["value"] for r in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()
    if not q:
        return JSONResponse({"success": False, "message": "Consulta no encontrada"})

    rows, err, _ = _query_rows(cfg, q["query_text"], factura, fecha_inicio, fecha_fin)
    if rows is None:
        return JSONResponse({"success": False, "message": err})
    if not rows:
        return JSONResponse({"success": False, "message": "Sin datos"})

    grupos = agrupar_por_recepcion(rows)
    if contrato:
        grupos = {k: v for k, v in grupos.items()
                  if str(v[0].get("CODCONTRATO") or "").strip() == contrato.strip()}
    prof_def = cfg.get("profesional_default", "")
    esp_def = cfg.get("especialidad_default", "")
    remis_def = cfg.get("remisionante_default", "00")
    prefijo_def = cfg.get("prefijo_tns_default", "00")

    items = []
    for id_rec, grupo_rows in grupos.items():
        enviado = _is_sent(id_rec)
        rda = generar_rda_paciente(grupo_rows, prof_def, esp_def, remis_def, prefijo_def)
        items.append({
            "idrecepcion": id_rec,
            "factura": grupo_rows[0].get("NUM_FACTURA", ""),
            "paciente": grupo_rows[0].get("COD_PACIENTE", ""),
            "contrato": grupo_rows[0].get("CODCONTRATO", ""),
            "examenes": [r.get("COD_EXAMEN", "") for r in grupo_rows],
            "valor": float(grupo_rows[0].get("VALORTOTAL") or 0),
            "enviado": enviado,
            "json": rda,
        })

    pendientes = sum(1 for i in items if not i["enviado"])
    enviados_ok = sum(1 for i in items if i["enviado"] and i["enviado"]["status"] == "success")
    enviados_err = sum(1 for i in items if i["enviado"] and i["enviado"]["status"] == "error")

    return JSONResponse({
        "success": True,
        "total": len(items),
        "pendientes": pendientes,
        "enviados_ok": enviados_ok,
        "enviados_err": enviados_err,
        "items": items,
    })


@router.post("/send-one")
async def send_one(
    request: Request, user: dict = Depends(get_current_user),
    idrecepcion: int = Form(...), query_id: int = Form(...),
    factura: str = Form(""), fecha_inicio: str = Form(...), fecha_fin: str = Form(...),
    contrato: str = Form(""),
):
    conn = get_connection()
    q = conn.execute("SELECT * FROM queries WHERE id = ?", (query_id,)).fetchone()
    cfg = {r["key"]: r["value"] for r in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    rows, err, _ = _query_rows(cfg, q["query_text"], factura, fecha_inicio, fecha_fin)
    if rows is None:
        return JSONResponse({"success": False, "message": err})

    grupos = agrupar_por_recepcion(rows)
    grupo_rows = grupos.get(idrecepcion)
    if not grupo_rows:
        return JSONResponse({"success": False, "message": f"IDRECEPCION {idrecepcion} no encontrado"})

    token, token_err = await get_tns_token(
        cfg.get("tns_empresa", ""), cfg.get("tns_usuario", ""), cfg.get("tns_password", "")
    )
    if not token:
        return JSONResponse({"success": False, "message": f"Error login TNS: {token_err}"})

    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {token}"}
    api_sucursal = cfg.get("api_sucursal", "") or "00"
    endpoint = f"{TNS_BASE}/v2/rda/RdaPaciente/Insertar?codigosucursal={api_sucursal}"

    num_factura_one = str(grupo_rows[0].get("NUM_FACTURA") or "").strip()
    numero_override_one = num_factura_one
    rda_json = generar_rda_paciente(
        grupo_rows,
        cfg.get("profesional_default", ""),
        cfg.get("especialidad_default", ""),
        cfg.get("remisionante_default", "00"),
        cfg.get("prefijo_tns_default", "00"),
        numero_override_one,
    )

    raw_resp = ""
    ok_rda = False
    msg_tns = ""
    try:
        async with httpx.AsyncClient(timeout=int(cfg.get("api_timeout", 30))) as client:
            r = await client.post(endpoint, json=rda_json, headers=headers)
        raw_resp = r.text
        try:
            data = r.json()
            ok_rda = bool(data.get("status") or (data.get("data") or {}).get("success", False))
            msg_tns = ((data.get("data") or {}).get("response") or data.get("message") or raw_resp[:300])
        except Exception:
            ok_rda = r.status_code < 300
            msg_tns = raw_resp[:300]
    except Exception as ex:
        msg_tns = str(ex)

    conn = get_connection()
    conn.execute("""
        INSERT INTO envios (user_id, tipo, factura, idrecepcion, contrato,
            fecha_inicio, fecha_fin, pacientes_count, servicios_count,
            status, json_enviado, respuesta_api, mensaje_tns, created_at)
        VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?)
    """, (
        user["user_id"], "transaccion",
        str(grupo_rows[0].get("NUM_FACTURA", idrecepcion)),
        idrecepcion,
        str(grupo_rows[0].get("CODCONTRATO", "")),
        fecha_inicio, fecha_fin, 1, len(grupo_rows),
        "success" if ok_rda else "error",
        json_lib.dumps(rda_json, ensure_ascii=False)[:5000],
        raw_resp[:2000],
        msg_tns,
        datetime.now().isoformat(),
    ))
    conn.commit()
    conn.close()

    return JSONResponse({"success": ok_rda, "message": msg_tns, "raw_tns": raw_resp, "idrecepcion": idrecepcion})


@router.post("/send")
async def send_transaccion(
    request: Request, user: dict = Depends(get_current_user),
    query_id: int = Form(...), factura: str = Form(""),
    fecha_inicio: str = Form(...), fecha_fin: str = Form(...),
    solo_pendientes: str = Form("0"), contrato: str = Form(""),
):
    conn = get_connection()
    q = conn.execute("SELECT * FROM queries WHERE id = ?", (query_id,)).fetchone()
    cfg = {r["key"]: r["value"] for r in conn.execute("SELECT * FROM config").fetchall()}
    conn.close()

    rows, err, _ = _query_rows(cfg, q["query_text"], factura, fecha_inicio, fecha_fin)
    if rows is None:
        return JSONResponse({"success": False, "message": err})
    if not rows:
        return JSONResponse({"success": False, "message": "Sin datos"})

    grupos = agrupar_por_recepcion(rows)
    if contrato:
        grupos = {k: v for k, v in grupos.items()
                  if str(v[0].get("CODCONTRATO") or "").strip() == contrato.strip()}
    if solo_pendientes == "1":
        grupos = {k: v for k, v in grupos.items() if not _is_sent(k)}

    token, token_err = await get_tns_token(
        cfg.get("tns_empresa", ""), cfg.get("tns_usuario", ""), cfg.get("tns_password", "")
    )
    if not token:
        return JSONResponse({"success": False, "message": f"Error login TNS: {token_err}"})

    headers = {"Content-Type": "application/json", "Authorization": f"Bearer {token}"}
    api_sucursal = cfg.get("api_sucursal", "") or "00"
    endpoint = f"{TNS_BASE}/v2/rda/RdaPaciente/Insertar?codigosucursal={api_sucursal}"

    resultados = []
    async with httpx.AsyncClient(timeout=int(cfg.get("api_timeout", 30))) as client:
        for id_rec, grupo_rows in grupos.items():
            num_factura = str(grupo_rows[0].get("NUM_FACTURA") or "").strip()
            numero_override = num_factura
            rda_json = generar_rda_paciente(
                grupo_rows,
                cfg.get("profesional_default", ""),
                cfg.get("especialidad_default", ""),
                cfg.get("remisionante_default", "00"),
                cfg.get("prefijo_tns_default", "00"),
                numero_override,
            )
            raw_resp = ""
            ok_rda = False
            msg_tns = ""
            try:
                r = await client.post(endpoint, json=rda_json, headers=headers)
                raw_resp = r.text
                try:
                    data = r.json()
                    ok_rda = bool(data.get("status") or (data.get("data") or {}).get("success", False))
                    msg_tns = ((data.get("data") or {}).get("response") or data.get("message") or raw_resp[:300])
                except Exception:
                    ok_rda = r.status_code < 300
                    msg_tns = raw_resp[:300]
            except Exception as ex:
                msg_tns = str(ex)

            conn = get_connection()
            conn.execute("""
                INSERT INTO envios (user_id, tipo, factura, idrecepcion, contrato,
                    fecha_inicio, fecha_fin, pacientes_count, servicios_count,
                    status, json_enviado, respuesta_api, mensaje_tns, created_at)
                VALUES (?,?,?,?,?,?,?,?,?,?,?,?,?,?)
            """, (
                user["user_id"], "transaccion",
                str(grupo_rows[0].get("NUM_FACTURA", id_rec)),
                id_rec,
                str(grupo_rows[0].get("CODCONTRATO", "")),
                fecha_inicio, fecha_fin, 1, len(grupo_rows),
                "success" if ok_rda else "error",
                json_lib.dumps(rda_json, ensure_ascii=False)[:5000],
                raw_resp[:2000], msg_tns,
                datetime.now().isoformat(),
            ))
            conn.commit()
            conn.close()
            resultados.append({"idrecepcion": id_rec, "success": ok_rda, "msg": msg_tns})

    ok_count = sum(1 for r in resultados if r["success"])
    err_count = len(resultados) - ok_count
    return JSONResponse({
        "success": err_count == 0,
        "total_enviados": ok_count,
        "total_errores": err_count,
        "resultados": resultados,
    })