import pandas as pd
import joblib
import os
import pyodbc
from datetime import datetime, timedelta
import sys

def insertar_alerta_en_db(cursor, tipo_alerta, id_entidad, entidad, mensaje, fecha_anomalia, cant_enviada=None, cant_devuelta=None, porc_devolucion=None):
    """Función centralizada para insertar en la nueva tabla Sistema_Alertas."""
    insert_query = """
        INSERT INTO Sistema_Alertas 
            (TipoAlerta, IdEntidad, Entidad, Mensaje, FechaAnomalia, CantidadEnviada, CantidadDevuelta, PorcentajeDevolucion, Leida)
        VALUES (?, ?, ?, ?, ?, ?, ?, ?, 0)
    """
    try:
        # Asegurarse de que los valores numéricos opcionales sean None si no se proporcionan
        p_dev = float(porc_devolucion) if porc_devolucion is not None else None
        c_env = int(cant_enviada) if cant_enviada is not None else None
        c_dev = int(cant_devuelta) if cant_devuelta is not None else None

        cursor.execute(insert_query, tipo_alerta, id_entidad, entidad, mensaje, fecha_anomalia, c_env, c_dev, p_dev)
        print(f"INFO: Alerta '{tipo_alerta}' para '{entidad}' ID {id_entidad} registrada.")
    except Exception as e:
        print(f"ERROR: No se pudo insertar la alerta para '{entidad}' ID {id_entidad}. Error: {e}")

print("--- INICIANDO SCRIPT DE DETECCIÓN COMPLETO ---")

# --- 1. Configuración ---
DB_SERVER = 'TECNICA3'
DB_DATABASE = 'SistemaGestion'
CONNECTION_STRING = f'DRIVER={{ODBC Driver 18 for SQL Server}};SERVER={DB_SERVER};DATABASE={DB_DATABASE};Trusted_Connection=yes;TrustServerCertificate=yes;'
MODEL_INDIVIDUAL_FILE = 'modelo_anomalias.joblib'
MODEL_SISTEMA_FILE = 'modelo_sistema_anomalias.joblib'

# --- 2. Determinar Fecha ---
if len(sys.argv) > 1:
    target_date = datetime.strptime(sys.argv[1], '%Y-%m-%d')
else:
    target_date = datetime.now() - timedelta(days=1)
print(f"--- FECHA DE ANÁLISIS: {target_date.date()} ---")

try:
    cnxn = pyodbc.connect(CONNECTION_STRING)
    cursor = cnxn.cursor()
except Exception as e:
    print(f"CRITICAL: No se pudo conectar a la base de datos. Error: {e}")
    exit()

# --- 3. DETECCIÓN INDIVIDUAL (CANILLITAS) ---
print("\n--- FASE 1: Detección de Anomalías Individuales (Canillitas) ---")
if not os.path.exists(MODEL_INDIVIDUAL_FILE):
    print(f"ADVERTENCIA: Modelo individual '{MODEL_INDIVIDUAL_FILE}' no encontrado.")
else:
    model_individual = joblib.load(MODEL_INDIVIDUAL_FILE)
    query_individual = f"""
        SELECT esc.Id_Canilla AS id_canilla, esc.Fecha AS fecha, esc.CantSalida AS cantidad_enviada, esc.CantEntrada AS cantidad_devuelta, c.NomApe AS nombre_canilla
        FROM dist_EntradasSalidasCanillas esc
        JOIN dist_dtCanillas c ON esc.Id_Canilla = c.Id_Canilla
        WHERE CAST(Fecha AS DATE) = '{target_date.strftime('%Y-%m-%d')}' AND CantSalida > 0
    """
    df_new = pd.read_sql(query_individual, cnxn)
    
    if not df_new.empty:
        df_new['porcentaje_devolucion'] = (df_new['cantidad_devuelta'] / df_new['cantidad_enviada']).fillna(0) * 100
        df_new['dia_semana'] = pd.to_datetime(df_new['fecha']).dt.dayofweek
        features = ['id_canilla', 'porcentaje_devolucion', 'dia_semana']
        X_new = df_new[features]
        df_new['anomalia'] = model_individual.predict(X_new)
        anomalias_detectadas = df_new[df_new['anomalia'] == -1]
        
        if not anomalias_detectadas.empty:
            for index, row in anomalias_detectadas.iterrows():
                mensaje = f"Devolución del {row['porcentaje_devolucion']:.2f}% para '{row['nombre_canilla']}'."
                insertar_alerta_en_db(cursor,
                                      tipo_alerta='DevolucionAnomala',
                                      id_entidad=row['id_canilla'],
                                      entidad='Canillita',
                                      mensaje=mensaje,
                                      fecha_anomalia=row['fecha'].date(),
                                      cant_enviada=row['cantidad_enviada'],
                                      cant_devuelta=row['cantidad_devuelta'],
                                      porc_devolucion=row['porcentaje_devolucion'])
        else:
            print("INFO: No se encontraron anomalías individuales significativas.")
    else:
        print("INFO: No hay datos de canillitas para analizar en la fecha seleccionada.")

# --- 4. DETECCIÓN DE SISTEMA ---
print("\n--- FASE 2: Detección de Anomalías de Sistema ---")
if not os.path.exists(MODEL_SISTEMA_FILE):
    print(f"ADVERTENCIA: Modelo de sistema '{MODEL_SISTEMA_FILE}' no encontrado.")
else:
    model_sistema = joblib.load(MODEL_SISTEMA_FILE)
    query_agregada = f"""
        SELECT CAST(Fecha AS DATE) AS fecha_dia, DATEPART(weekday, Fecha) as dia_semana,
               COUNT(DISTINCT Id_Canilla) as total_canillitas_activos,
               SUM(CantSalida) as total_salidas, SUM(CantEntrada) as total_devoluciones
        FROM dist_EntradasSalidasCanillas
        WHERE CAST(Fecha AS DATE) = '{target_date.strftime('%Y-%m-%d')}' AND CantSalida > 0
        GROUP BY CAST(Fecha AS DATE), DATEPART(weekday, Fecha)
    """
    df_system = pd.read_sql(query_agregada, cnxn)
    
    if not df_system.empty and df_system['total_salidas'].iloc[0] > 0:
        df_system['ratio_devolucion'] = (df_system['total_devoluciones'] / df_system['total_salidas']).fillna(0)
        df_system['salidas_por_canillita'] = (df_system['total_salidas'] / df_system['total_canillitas_activos']).fillna(0)
        features_system = ['dia_semana', 'total_salidas', 'ratio_devolucion', 'salidas_por_canillita']
        X_system = df_system[features_system]
        df_system['anomalia_sistema'] = model_sistema.predict(X_system)
        
        if df_system['anomalia_sistema'].iloc[0] == -1:
            ratio_hoy = df_system['ratio_devolucion'].iloc[0] * 100
            mensaje = f"El ratio de devolución global fue del {ratio_hoy:.2f}%, un valor atípico para este día de la semana."
            insertar_alerta_en_db(cursor,
                                  tipo_alerta='ComportamientoSistema',
                                  id_entidad=0,
                                  entidad='Sistema',
                                  mensaje=mensaje,
                                  fecha_anomalia=target_date.date())
        else:
            print("INFO: El comportamiento agregado del sistema fue normal.")
    else:
        mensaje = f"ALERTA GRAVE: No se registraron movimientos de salida para ningún canillita en la fecha {target_date.date()}."
        insertar_alerta_en_db(cursor,
                              tipo_alerta='FaltaDeDatos',
                              id_entidad=0,
                              entidad='Sistema',
                              mensaje=mensaje,
                              fecha_anomalia=target_date.date())

# --- 5. Finalización ---
cnxn.commit()
cnxn.close()
print("\n--- DETECCIÓN COMPLETA ---")