log_context

Author: giancarloromeo

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

@@ -4,6 +4,7 @@
 from celery import Celery
 from celery.contrib.abortable import AbortableTask
 from models_library.progress_bar import ProgressReport
+from servicelib.logging_utils import log_context
 
 from .models import TaskID
 
@@ -22,11 +23,13 @@ def register_task(self, fn: Callable, task_name: str | None = None) -> None:
     def set_task_progress(
         self, task_name: str, task_id: TaskID, report: ProgressReport
     ) -> None:
-        _logger.debug(
-            "Setting progress for %s: %s", task_name, report.model_dump_json()
-        )
-        self.celery_app.tasks[task_name].update_state(
-            task_id=task_id,
-            state="PROGRESS",
-            meta=report.model_dump(mode="json"),
-        )
+        with log_context(
+            _logger,
+            logging.DEBUG,
+            msg=f"Setting progress for {task_name}: {report.model_dump_json()}",
+        ):
+            self.celery_app.tasks[task_name].update_state(
+                task_id=task_id,
+                state="PROGRESS",
+                meta=report.model_dump(mode="json"),
+            )