Upload api.py

api.py

@@ -22,9 +22,11 @@ import imaplib
 import email
 from email.header import decode_header
 import re
-import hashlib
-import time
-import pandas as pd
+import hashlib
+import time
+import threading
+from datetime import datetime
+import pandas as pd
 import io
 from urllib.parse import urljoin
 from dotenv import load_dotenv
@@ -120,9 +122,25 @@ logger.addHandler(log_handler)
 
 # --- 2. SEGURIDAD Y CIFRADO ---
 # Se utiliza el módulo centralizado `crypto.py`
-INTERNAL_API_TOKEN = os.getenv("INTERNAL_API_TOKEN", "default-dev-token")
-
-def verify_internal_token(x_internal_token: str = Header(None)):
+INTERNAL_API_TOKEN = os.getenv("INTERNAL_API_TOKEN", "default-dev-token")
+RADAR_AUTO_SCAN_ENABLED = os.getenv("RADAR_AUTO_SCAN_ENABLED", "true").strip().lower() in ["1", "true", "yes", "si", "sí", "on"]
+RADAR_AUTO_SCAN_INTERVAL_MINUTES = max(5, int(os.getenv("RADAR_AUTO_SCAN_INTERVAL_MINUTES", "25") or 25))
+RADAR_AUTO_SCAN_ON_STARTUP = os.getenv("RADAR_AUTO_SCAN_ON_STARTUP", "false").strip().lower() in ["1", "true", "yes", "si", "sí", "on"]
+RADAR_SCHEDULER_STATE = {
+    "enabled": RADAR_AUTO_SCAN_ENABLED,
+    "interval_minutes": RADAR_AUTO_SCAN_INTERVAL_MINUTES,
+    "running": False,
+    "last_started": None,
+    "last_finished": None,
+    "last_result": None,
+    "last_error": None,
+    "next_run_at": None,
+}
+_radar_scheduler_stop = threading.Event()
+_radar_scan_lock = threading.Lock()
+_radar_scheduler_thread = None
+
+def verify_internal_token(x_internal_token: str = Header(None)):
     if x_internal_token != INTERNAL_API_TOKEN:
         raise HTTPException(status_code=403, detail="Acceso denegado: Token interno inválido.")
     return x_internal_token
@@ -139,7 +157,76 @@ app.add_middleware(
     allow_headers=["*"],
 )
 
-@app.get("/")
+def run_radar_auto_scan(source="scheduler"):
+    """Ejecuta el escaneo del Radar SLI de forma segura para jobs automáticos."""
+    if not _radar_scan_lock.acquire(blocking=False):
+        logger.info("[RADAR AUTO] Escaneo omitido: ya hay un escaneo en curso.")
+        return {"status": "skipped", "reason": "scan_already_running"}
+
+    RADAR_SCHEDULER_STATE["running"] = True
+    RADAR_SCHEDULER_STATE["last_started"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+    RADAR_SCHEDULER_STATE["last_error"] = None
+
+    try:
+        import sli_scraper
+        logger.info(f"[RADAR AUTO] Iniciando escaneo SLI ({source}).")
+        result = sli_scraper.ejecutar_radar_detallado(db_module=db)
+        RADAR_SCHEDULER_STATE["last_result"] = result
+        RADAR_SCHEDULER_STATE["last_finished"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        logger.info(f"[RADAR AUTO] Escaneo completado: {result}")
+        return {"status": "success", "result": result}
+    except Exception as exc:
+        error_text = str(exc)
+        RADAR_SCHEDULER_STATE["last_error"] = error_text
+        RADAR_SCHEDULER_STATE["last_finished"] = datetime.now().strftime("%Y-%m-%d %H:%M:%S")
+        try:
+            db.registrar_escaneo_radar(0, 0, f"Auto radar error: {error_text}")
+        except Exception:
+            logger.exception("[RADAR AUTO] No se pudo registrar error de escaneo.")
+        logger.exception(f"[RADAR AUTO] Error en escaneo: {error_text}")
+        return {"status": "error", "error": error_text}
+    finally:
+        RADAR_SCHEDULER_STATE["running"] = False
+        _radar_scan_lock.release()
+
+def radar_scheduler_loop():
+    interval_seconds = RADAR_AUTO_SCAN_INTERVAL_MINUTES * 60
+    if RADAR_AUTO_SCAN_ON_STARTUP:
+        run_radar_auto_scan(source="startup")
+    while not _radar_scheduler_stop.is_set():
+        next_run_ts = time.time() + interval_seconds
+        RADAR_SCHEDULER_STATE["next_run_at"] = datetime.fromtimestamp(next_run_ts).strftime("%Y-%m-%d %H:%M:%S")
+        if _radar_scheduler_stop.wait(interval_seconds):
+            break
+        run_radar_auto_scan()
+
+@app.on_event("startup")
+def start_radar_scheduler():
+    global _radar_scheduler_thread
+    if not RADAR_AUTO_SCAN_ENABLED:
+        logger.info("[RADAR AUTO] Scheduler desactivado por variable de entorno.")
+        return
+    if _radar_scheduler_thread and _radar_scheduler_thread.is_alive():
+        return
+    _radar_scheduler_stop.clear()
+    _radar_scheduler_thread = threading.Thread(target=radar_scheduler_loop, name="radar-sli-scheduler", daemon=True)
+    _radar_scheduler_thread.start()
+    logger.info(f"[RADAR AUTO] Scheduler iniciado cada {RADAR_AUTO_SCAN_INTERVAL_MINUTES} minutos.")
+
+@app.on_event("shutdown")
+def stop_radar_scheduler():
+    _radar_scheduler_stop.set()
+    logger.info("[RADAR AUTO] Scheduler detenido.")
+
+@app.get("/api/v1/radar/scheduler")
+def radar_scheduler_status(_token: str = Depends(verify_internal_token)):
+    return RADAR_SCHEDULER_STATE
+
+@app.post("/api/v1/radar/scan-now")
+def radar_scan_now(_token: str = Depends(verify_internal_token)):
+    return run_radar_auto_scan(source="manual_api")
+
+@app.get("/")
 def estado():
     return {"status": "Online", "engine": "Proyelec Core v6.1 — Token-Optimized"}