add enum

Author: giancarloromeo

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

@@ -13,6 +13,7 @@
 class TaskState(StrEnum):
     PENDING = auto()
     STARTED = auto()
+    PROGRESS = auto()
     SUCCESS = auto()
     FAILURE = auto()
     ABORTED = auto()

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

@@ -6,7 +6,7 @@
 from models_library.progress_bar import ProgressReport
 from servicelib.logging_utils import log_context
 
-from .models import TaskID
+from .models import TaskID, TaskState
 
 _logger = logging.getLogger(__name__)
 
@@ -30,6 +30,6 @@ def set_task_progress(
         ):
             self.celery_app.tasks[task_name].update_state(
                 task_id=task_id,
-                state="PROGRESS",
+                state=TaskState.PROGRESS.value,
                 meta=report.model_dump(mode="json"),
             )