por lotes

api/customs/tasks/microservice_v2.py

@@ -1,3 +1,33 @@
+from celery import group
+@shared_task
+def procesar_pedimento_completo_individual(pedimento_id):
+    pedimento = Pedimento.objects.get(id=pedimento_id)
+    if not pedimento.documents.filter(document_type=2).exists():  # Tipo 2: Pedimento Completo
+        pedimento_dict = pedimento_to_dict(pedimento)
+        credenciales = Vucem.objects.filter(
+            id=CredencialesImportador.objects.filter(rfc=pedimento.contribuyente).first().vucem.id
+        ).first()
+        credenciales_dict = credenciales_to_dict(credenciales)
+        payload = {
+            "pedimento": pedimento_dict,
+            "credencial": credenciales_dict
+        }
+        response = requests.post(
+            f"{SERVICE_API_URL_V2}/services/pedimento_completo", 
+            data=json.dumps(payload),
+            headers={"Content-Type": "application/json"}
+        )
+        print(f"Servicio enviado para pedimento {pedimento.pedimento}")
+def procesar_pedimentos_completos_en_lotes(organizacion_id, batch_size=20):
+    pedimentos = Pedimento.objects.filter(organizacion_id=organizacion_id)
+    ids = list(pedimentos.values_list('id', flat=True))
+    for i in range(0, len(ids), batch_size):
+        batch = ids[i:i+batch_size]
+        job = group(
+            procesar_pedimento_completo_individual.s(ped_id)
+            for ped_id in batch
+        )
+        job.apply_async()
 
 from celery import shared_task, group
 from api.customs.models import *