161 lines
6.6 KiB
Python
161 lines
6.6 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_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}") |