225 lines
9.1 KiB
Python
225 lines
9.1 KiB
Python
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}") |