from celery import shared_task, group
from api.customs.models import ProcesamientoPedimento, Pedimento, Cove, EDocument
from core.utils import xml_controller
import requests
from core.utils import xml_remesas_controller

def obtener_pedimentos(organizacion_id):
    return Pedimento.objects.filter(organizacion_id=organizacion_id)

def extraer_coves(pedimento):
    remesas = pedimento.documents.filter(document_type=3).first()
    with open(f'./media/{remesas.archivo}', 'r') as f:
        xml_content = f.read()

    xml_data = xml_remesas_controller.extract_remesas(xml_content)

    return xml_data

@shared_task
def auditar_procesamiento_remesas(organizacion_id):
    pedimentos = obtener_pedimentos(organizacion_id)

    for pedimento in pedimentos:
        if pedimento.remesas:
            # Tipo 3: Remesa
            if not pedimento.documents.filter(document_type=3).exists():
                ProcesamientoPedimento.objects.get_or_create(
                    pedimento=pedimento,
                    servicio_id=5,  # ID del servicio de remesas
                    organizacion=organizacion_id
                )
            else:
                xml_data = extraer_coves(pedimento)
                if xml_data:
                    for remesa in xml_data:
                        Cove.objects.get_or_create(
                            pedimento=pedimento,
                            numero_cove=remesa.get('remesaSA'),
                            organizacion=organizacion_id
                        )

@shared_task
def auditar_procesamiento_remesa_por_pedimento(pedimento_id):
    """
    Audita el procesamiento de remesa para un pedimento específico.
    Args:
        pedimento_id: UUID del pedimento a auditar
    Returns:
        dict: Resultado de la auditoría con detalles del procesamiento
    """
    try:
        pedimento = Pedimento.objects.get(id=pedimento_id)
        resultado = {
            'pedimento_id': str(pedimento_id),
            'pedimento_numero': pedimento.pedimento,
            'tiene_remesas': pedimento.remesas,
            'procesamiento_creado': False,
            'coves_creados': []
        }

        if not pedimento.remesas:
            resultado['mensaje'] = 'El pedimento no tiene remesas para procesar'
            return resultado

        # Verificar documento tipo remesa
        if not pedimento.documents.filter(document_type=3).exists():
            # Crear procesamiento si no existe documento de remesa
            procesamiento, creado = ProcesamientoPedimento.objects.get_or_create(
                pedimento=pedimento,
                servicio_id=5,  # ID del servicio de remesas
                organizacion=pedimento.organizacion_id
            )
            resultado['procesamiento_creado'] = creado
            resultado['mensaje'] = 'Procesamiento de remesa creado - documento no encontrado'
        else:
            # Procesar XML de remesas
            xml_data = extraer_coves(pedimento)
            if xml_data:
                for remesa in xml_data:
                    numero_cove = remesa.get('remesaSA')
                    cove, creado = Cove.objects.get_or_create(
                        pedimento=pedimento,
                        numero_cove=numero_cove,
                        organizacion=pedimento.organizacion_id
                    )
                    if creado:
                        resultado['coves_creados'].append(numero_cove)
                
                resultado['mensaje'] = f"Procesados {len(xml_data)} remesas, creados {len(resultado['coves_creados'])} COVEs nuevos"
            else:
                resultado['mensaje'] = 'No se encontraron datos de remesas en el XML'

        return resultado

    except Pedimento.DoesNotExist:
        return {
            'error': f'Pedimento con ID {pedimento_id} no encontrado',
            'pedimento_id': str(pedimento_id)
        }
    except Exception as e:
        return {
            'error': f'Error procesando pedimento {pedimento_id}: {str(e)}',
            'pedimento_id': str(pedimento_id)
        }

    
@shared_task
def auditar_partidas(organizacion_id):
    pedimentos = Pedimento.objects.filter(organizacion_id=organizacion_id)
    for pedimento in pedimentos:
        # Validar que numero_partidas no sea None y sea mayor que 0
        if pedimento.numero_partidas is not None and pedimento.numero_partidas > 0:
            partidas_descargadas = pedimento.documents.filter(document_type=1)

            partidas = {str(documento.archivo).split('_')[-1].split('.')[0]: documento.archivo for documento in partidas_descargadas}
            partidas_faltantes = []

            for i in range(1, pedimento.numero_partidas + 1):
                if str(i) not in partidas.keys():
                    partidas_faltantes.append(i)
                    # crear servicio individual para cada partida faltante en microservicios
            
            
@shared_task
def auditar_coves(organizacion_id):
    # crear servicio individual para cada cove faltante en microservicios
    pass

@shared_task
def auditar_edocuments(organizacion_id):
    # crear servicio individual para cada Edocument faltante en microservicios
    pass

@shared_task
def auditar_acuse_coves(organizacion_id):
    # crear servicio individual para cada cove faltante en microservicios

    pass



@shared_task
def auditar_acuse_edocuments(organizacion_id):
    # crear servicio individual para cada acuse de edocument faltante en microservicios
    pedimentos = obtener_pedimentos(organizacion_id)

    for pedimento in pedimentos:
        acuses_descargados = pedimento.documents.filter(document_type=4)
        edocs = pedimento.documentos.all()



@shared_task
def crear_partidas(organizacion_id):
    pedimentos = obtener_pedimentos(organizacion_id)
    total_pedimentos = pedimentos.count()
    pedimentos_procesados = 0
    total_partidas_agregadas = 0

    print(f"Iniciando procesamiento de {total_pedimentos} pedimentos para organización {organizacion_id}")

    for pedimento in pedimentos:
        pedimentos_procesados += 1
        partidas_agregadas_pedimento = 0
        
        # Validar que numero_partidas no sea None y sea mayor que 0
        if pedimento.numero_partidas is not None and pedimento.numero_partidas > 0:
            partidas_existentes = pedimento.partidas.count()
            if pedimento.numero_partidas > partidas_existentes:
                print(f"Procesando pedimento {pedimento.id} ({pedimentos_procesados}/{total_pedimentos}) - Partidas existentes: {partidas_existentes}, Requeridas: {pedimento.numero_partidas}")
                
                for i in range(1, pedimento.numero_partidas + 1):
                    from api.customs.models import Partida
                    partida, created = Partida.objects.get_or_create(
                        pedimento=pedimento,
                        numero_partida=i,
                        organizacion_id=organizacion_id
                    )
                    if created:
                        partidas_agregadas_pedimento += 1
                        total_partidas_agregadas += 1
                
                print(f"  → Partidas agregadas para pedimento {pedimento.id}: {partidas_agregadas_pedimento}")
            else:
                print(f"Pedimento {pedimento.id} ya tiene todas sus partidas ({partidas_existentes}/{pedimento.numero_partidas})")
        else:
            print(f"Pedimento {pedimento.id} omitido - numero_partidas: {pedimento.numero_partidas} (inválido)")

    print(f"\n=== RESUMEN ===")
    print(f"Pedimentos procesados: {pedimentos_procesados}")
    print(f"Total de partidas agregadas: {total_partidas_agregadas}")
    print(f"Procesamiento completado para organización {organizacion_id}")

@shared_task
def crear_partidas_por_pedimento(pedimento_id):
    try:
        pedimento = Pedimento.objects.get(id=pedimento_id)
    except Pedimento.DoesNotExist:
        print(f"Error: Pedimento con ID {pedimento_id} no encontrado")
        return

    print(f"Procesando pedimento individual {pedimento_id}...")
    partidas_agregadas = 0
    
    # Validar que numero_partidas no sea None y sea mayor que 0
    if pedimento.numero_partidas is not None and pedimento.numero_partidas > 0:
        partidas_existentes = pedimento.partidas.count()
        if pedimento.numero_partidas > partidas_existentes:
            print(f"Pedimento {pedimento_id} - Partidas existentes: {partidas_existentes}, Requeridas: {pedimento.numero_partidas}")
            
            for i in range(1, pedimento.numero_partidas + 1):
                from api.customs.models import Partida
                partida, created = Partida.objects.get_or_create(
                    pedimento=pedimento,
                    numero_partida=i,
                    organizacion_id=pedimento.organizacion_id
                )
                if created:
                    partidas_agregadas += 1
            
            print(f"✓ Partidas agregadas para pedimento {pedimento_id}: {partidas_agregadas}")
        else:
            print(f"Pedimento {pedimento_id} ya tiene todas sus partidas ({partidas_existentes}/{pedimento.numero_partidas})")
    else:
        print(f"Error: Pedimento {pedimento_id} tiene numero_partidas inválido: {pedimento.numero_partidas}")