fix: use final states

Author: giancarloromeo

packages/celery-library/src/celery_library/task_manager.py

@@ -7,6 +7,7 @@
 from common_library.async_tools import make_async
 from models_library.progress_bar import ProgressReport
 from servicelib.celery.models import (
+    TASK_FINAL_STATES,
     Task,
     TaskFilter,
     TaskID,
@@ -110,6 +111,11 @@ async def _get_task_progress_report(
             if progress is not None:
                 return progress
 
+        if task_state in TASK_FINAL_STATES:
+            return ProgressReport(
+                actual_value=_MAX_PROGRESS_VALUE, total=_MAX_PROGRESS_VALUE
+            )
+
         # task is pending
         return ProgressReport(
             actual_value=_MIN_PROGRESS_VALUE, total=_MAX_PROGRESS_VALUE

packages/service-library/src/servicelib/celery/models.py

@@ -26,6 +26,9 @@ class TaskState(StrEnum):
     ABORTED = "ABORTED"
 
 
+TASK_FINAL_STATES = {TaskState.SUCCESS, TaskState.FAILURE, TaskState.ABORTED}
+
+
 class TasksQueue(StrEnum):
     CPU_BOUND = "cpu_bound"
     DEFAULT = "default"
@@ -78,9 +81,6 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
     model_config = ConfigDict(json_schema_extra=_update_json_schema_extra)
 
 
-_TASK_DONE = {TaskState.SUCCESS, TaskState.FAILURE, TaskState.ABORTED}
-
-
 class TaskInfoStore(Protocol):
     async def create_task(
         self,
@@ -138,4 +138,4 @@ def _update_json_schema_extra(schema: JsonDict) -> None:
 
     @property
     def is_done(self) -> bool:
-        return self.task_state in _TASK_DONE
+        return self.task_state in TASK_FINAL_STATES