Se agregaron estados y update a cada una de las tareas

api/api_v2/modules/acuses/routers.py

@@ -19,16 +19,6 @@ async def obtener_acuse(acuse_request: AcuseSchema):
     # Ejecuta la tarea de Celery de forma asíncrona
     task = process_acuse_request.delay(acuse_dict)
     
-    # Registra la tarea en el servicio de seguimiento
-    await register_task(
-        task_id=task.id,
-        message=f"Procesando acuse para pedimento {acuse_dict.get('pedimento', {}).get('pedimento_app', 'N/A')}",
-        status="submitted",
-        pedimento_id=acuse_dict.get('pedimento', {}).get('id'),
-        organizacion_id=acuse_dict.get('pedimento', {}).get('organizacion'),
-        servicio=6  # 6 corresponde a "Acuse"
-    )
-    
     return {"task_id": task.id, "status": "submitted"}
 
 
@@ -49,14 +39,4 @@ async def obtener_acuses(acuse_request: AcuseMasivoSchema):
         task = process_acuse_request.delay(acuse_dict)
         task_ids.append(task.id)
         
-        # Registra cada tarea en el servicio de seguimiento
-        await register_task(
-            task_id=task.id,
-            message=f"Procesando acuse masivo para pedimento {acuse_request_dict.get('pedimento', {}).get('pedimento_app', 'N/A')}",
-            status="submitted",
-            pedimento_id=acuse_request_dict.get('pedimento', {}).get('id'),
-            organizacion_id=acuse_request_dict.get('pedimento', {}).get('organizacion'),
-            servicio=6  # 6 corresponde a "Acuse"
-        )
-
     return {"task_ids": task_ids, "status": "submitted", "total": len(task_ids)}

api/api_v2/modules/acuses/tasks.py

@@ -1,4 +1,4 @@
-from celery import Celery
+from celery import Celery, current_task
 from celery_app import celery_app
 import asyncio
 import logging
@@ -6,22 +6,82 @@ from typing import Dict, Any
 from contextlib import asynccontextmanager
 
 from .services import obtener_acuse
-from api.api_v2.modules.tasks.tasks import run_async_task
+from api.api_v2.modules.tasks.services import register_task, update_task
 
 
-@celery_app.task
-def process_acuse_request(acuse_request: Dict[str, Any]) -> Dict[str, Any]:
+@celery_app.task(bind=True)
+def process_acuse_request(self, acuse_request: Dict[str, Any]) -> Dict[str, Any]:
     """
-    Tarea de Celery para procesar la solicitud de acuse.
+    Tarea para procesar solicitudes de acuse.
+    """
+    task_id = self.request.id
+    pedimento_info = acuse_request.get('pedimento', {})
+    pedimento_id = pedimento_info.get('id')
+    organizacion_id = pedimento_info.get('organizacion')
+    pedimento_app = pedimento_info.get('pedimento_app', 'N/A')
     
-    Args:
-        acuse_request: Diccionario con los datos de la solicitud de acuse.
+    try:
+        # Registrar el inicio de la tarea
+        loop = asyncio.get_event_loop()
+        logging.info(f"[ACUSE] Registrando inicio de tarea {task_id}")
+        loop.run_until_complete(
+            register_task(
+                task_id=task_id,
+                status="submitted",
+                message=f"Iniciando proceso de acuse para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=6  # 6 corresponde a "Acuse"
+            )
+        )
+
+        # Esperar un momento breve para asegurar que el registro se complete
+        import time
+        time.sleep(1)
         
-    Returns:
-        Diccionario con la respuesta del acuse.
-    """
-    loop = asyncio.get_event_loop()
-    acuse_response = loop.run_until_complete(obtener_acuse(**acuse_request))
-
-    return {"status": "processed", "data": acuse_response}
+        # Actualizar estado: procesando
+        logging.info(f"[ACUSE] Actualizando estado a processing para tarea {task_id}")
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="processing",
+                message=f"Procesando acuse para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=6
+            )
+        )
+        
+        # Obtener el acuse
+        acuse_response = loop.run_until_complete(obtener_acuse(**acuse_request))
+        
+        # Actualizar estado: completado
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="completed",
+                message=f"Acuse obtenido exitosamente para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=6
+            )
+        )
+        
+        return {"status": "processed", "data": acuse_response}
+        
+    except Exception as e:
+        # En caso de error, actualizar estado
+        error_message = f"Error al procesar acuse para pedimento {pedimento_app}: {str(e)}"
+        logging.error(error_message)
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="failed",
+                message=error_message,
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=6
+            )
+        )
+        raise