Se agregaron estados y update a cada una de las tareas

api/api_v2/modules/coves/tasks.py

@@ -1,11 +1,12 @@
 import asyncio
 import logging
+import time
 from celery import Celery
 from celery_app import celery_app
 from typing import Dict, Any
 
 from .services import consume_ws_get_cove, consume_ws_get_acuse_cove
-from api.api_v2.modules.tasks.tasks import run_async_task
+from api.api_v2.modules.tasks.services import update_task, register_task
 
 # Logger para el módulo
 logger = logging.getLogger(__name__)
@@ -13,16 +14,156 @@ logger = logging.getLogger(__name__)
 
 @celery_app.task(bind=True)
 def process_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str, Any]:
-    loop = asyncio.get_event_loop()
-    cove_response = loop.run_until_complete(consume_ws_get_cove(**cove_request))
+    """
+    Tarea para procesar solicitudes de COVE.
+    """
+    task_id = self.request.id
+    pedimento_info = cove_request.get('pedimento', {})
+    pedimento_id = pedimento_info.get('id')
+    organizacion_id = pedimento_info.get('organizacion')
+    pedimento_app = pedimento_info.get('pedimento_app', 'N/A')
+    cove_info = cove_request.get('cove', {})
+    cove_number = cove_info.get('numero_cove', 'N/A')
+    
+    try:
+        # Registrar el inicio de la tarea
+        loop = asyncio.get_event_loop()
+        logger.info(f"[COVE] Registrando inicio de tarea {task_id}")
+        loop.run_until_complete(
+            register_task(
+                task_id=task_id,
+                status="submitted",
+                message=f"Iniciando proceso de COVE {cove_number} para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=8  # 8 corresponde a "Cove"
+            )
+        )
 
-    return {"status": "processed", "data": cove_response}
+        # Esperar un momento breve para asegurar que el registro se complete
+        time.sleep(1)
+        
+        # Actualizar estado: procesando
+        logger.info(f"[COVE] Actualizando estado a processing para tarea {task_id}")
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="processing",
+                message=f"Procesando COVE {cove_number} para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=8
+            )
+        )
+        
+        # Obtener el COVE
+        cove_response = loop.run_until_complete(consume_ws_get_cove(**cove_request))
+        
+        # Actualizar estado: completado
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="completed",
+                message=f"COVE {cove_number} obtenido exitosamente para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=8
+            )
+        )
+        
+        return {"status": "processed", "data": cove_response}
+        
+    except Exception as e:
+        # En caso de error, actualizar estado
+        error_message = f"Error al procesar COVE {cove_number} para pedimento {pedimento_app}: {str(e)}"
+        logger.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=8
+            )
+        )
+        raise
 
 
 @celery_app.task(bind=True)
 def process_acuse_cove_request(self, cove_request: Dict[str, Any]) -> Dict[str, Any]:
-    loop = asyncio.get_event_loop()
-    acuse_response = loop.run_until_complete(consume_ws_get_acuse_cove(**cove_request))
+    """
+    Tarea para procesar solicitudes de acuse de COVE.
+    """
+    task_id = self.request.id
+    pedimento_info = cove_request.get('pedimento', {})
+    pedimento_id = pedimento_info.get('id')
+    organizacion_id = pedimento_info.get('organizacion')
+    pedimento_app = pedimento_info.get('pedimento_app', 'N/A')
+    cove_info = cove_request.get('cove', {})
+    cove_number = cove_info.get('numero_cove', 'N/A')
+    
+    try:
+        # Registrar el inicio de la tarea
+        loop = asyncio.get_event_loop()
+        logger.info(f"[COVE] Registrando inicio de tarea de acuse {task_id}")
+        loop.run_until_complete(
+            register_task(
+                task_id=task_id,
+                status="submitted",
+                message=f"Iniciando proceso de acuse de COVE {cove_number} para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=9  # 9 corresponde a "Acuse Cove"
+            )
+        )
 
-    return {"status": "processed", "data": acuse_response}
+        # Esperar un momento breve para asegurar que el registro se complete
+        time.sleep(1)
+        
+        # Actualizar estado: procesando
+        logger.info(f"[COVE] Actualizando estado a processing para tarea de acuse {task_id}")
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="processing",
+                message=f"Procesando acuse de COVE {cove_number} para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=9
+            )
+        )
+        
+        # Obtener el acuse del COVE
+        acuse_response = loop.run_until_complete(consume_ws_get_acuse_cove(**cove_request))
+        
+        # Actualizar estado: completado
+        loop.run_until_complete(
+            update_task(
+                task_id=task_id,
+                status="completed",
+                message=f"Acuse de COVE {cove_number} obtenido exitosamente para pedimento {pedimento_app}",
+                pedimento_id=pedimento_id,
+                organizacion_id=organizacion_id,
+                servicio=9
+            )
+        )
+        
+        return {"status": "processed", "data": acuse_response}
+        
+    except Exception as e:
+        # En caso de error, actualizar estado
+        error_message = f"Error al procesar acuse de COVE {cove_number} para pedimento {pedimento_app}: {str(e)}"
+        logger.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=9
+            )
+        )
+        raise