se estan creando el registro de las tareas sin problemas
This commit is contained in:
@@ -3,6 +3,7 @@ from typing import Dict, Any, Optional
|
||||
from .schemas import EdocumentsSchema, EdocumentsMasivoSchema
|
||||
from .tasks import process_edoc_download_request
|
||||
from api.api_v2.modules.authentication.services import get_current_user
|
||||
from api.api_v2.modules.tasks.services import register_task
|
||||
|
||||
router = APIRouter()
|
||||
|
||||
@@ -17,6 +18,15 @@ async def download_edoc(edoc_request: EdocumentsSchema):
|
||||
edoc_dict = edoc_request.model_dump()
|
||||
# Ejecuta la tarea de Celery de forma asíncrona
|
||||
task = process_edoc_download_request.delay(edoc_dict)
|
||||
# Registrar la tarea en el servicio de seguimiento
|
||||
await register_task(
|
||||
task_id=task.id,
|
||||
message=f"Procesando descarga de E-Document {edoc_dict.get('edoc', {}).get('numero_edocument', 'N/A')}",
|
||||
status="submitted",
|
||||
pedimento_id=edoc_dict.get('pedimento', {}).get('id'),
|
||||
organizacion_id=edoc_dict.get('pedimento', {}).get('organizacion'),
|
||||
servicio=7 # 7 corresponde a "EDocument"
|
||||
)
|
||||
# Devuelve el ID de la tarea
|
||||
return {"task_id": task.id, "status": "submitted"}
|
||||
|
||||
@@ -37,5 +47,14 @@ async def download_edocs_masivo(edoc_request: EdocumentsMasivoSchema):
|
||||
}
|
||||
task = process_edoc_download_request.delay(edoc_dict)
|
||||
task_ids.append(task.id)
|
||||
# Registrar cada tarea en el servicio de seguimiento
|
||||
await register_task(
|
||||
task_id=task.id,
|
||||
message=f"Procesando descarga masiva de E-Document {edoc.get('numero_edocument', 'N/A')}",
|
||||
status="submitted",
|
||||
pedimento_id=edoc_dict.get('pedimento', {}).get('id'),
|
||||
organizacion_id=edoc_dict.get('pedimento', {}).get('organizacion'),
|
||||
servicio=7 # 7 corresponde a "EDocument"
|
||||
)
|
||||
|
||||
return {"task_ids": task_ids, "status": "submitted", "total": len(task_ids)}
|
||||
Reference in New Issue
Block a user