add task

Author: giancarloromeo

services/storage/src/simcore_service_storage/modules/celery/client.py

@@ -1,6 +1,6 @@
 import contextlib
 import logging
-from typing import Any, Final, Type
+from typing import Any, Final
 from uuid import uuid4
 
 from celery import Celery
@@ -10,7 +10,7 @@
 from pydantic import TypeAdapter, ValidationError
 from servicelib.logging_utils import log_context
 
-from .models import TaskContext, TaskError, TaskID, TaskResult, TaskState, TaskStatus, TaskUUID
+from .models import TaskContext, TaskID, TaskResult, TaskState, TaskStatus, TaskUUID
 
 _logger = logging.getLogger(__name__)
 

services/storage/src/simcore_service_storage/modules/celery/tasks.py

@@ -1,16 +1,20 @@
 import logging
 import time
 
+
 from celery import Task
+from common_library.errors_classes import OsparcErrorMixin
 from models_library.progress_bar import ProgressReport
 from models_library.projects_nodes_io import StorageFileID
 from servicelib.logging_utils import log_context
-from simcore_service_storage.modules.celery.utils import get_celery_worker
+
+from .utils import get_celery_worker
 
 _logger = logging.getLogger(__name__)
 
 
 def export_data(task: Task, files: list[StorageFileID]):
+    _logger.info("Exporting files: %s", files)
     for n, file in enumerate(files, start=1):
         with log_context(
             _logger,
@@ -25,3 +29,12 @@ def export_data(task: Task, files: list[StorageFileID]):
             )
             time.sleep(10)
     return "done"
+
+
+class MyError(OsparcErrorMixin, Exception):
+   msg_template = "Something strange happened: {msg}"
+
+
+def export_data_with_error(task: Task, files: list[StorageFileID]):
+    msg = "BOOM!"
+    raise MyError(msg=msg)