allows to setup plugins

Author: sanderegg

services/dask-sidecar/src/simcore_service_dask_sidecar/task_life_cycle_scheduler_plugin.py

@@ -1,6 +1,7 @@
 import logging
 from typing import Any
 
+import click
 from dask.typing import Key
 from dask_task_models_library.models import TASK_LIFE_CYCLE_EVENT, TaskLifeCycleState
 from distributed import Scheduler, SchedulerPlugin
@@ -49,3 +50,9 @@ def transition(
                     key, kwargs.get("worker"), finish
                 ).model_dump(mode="json"),
             )
+
+
+@click.command()
+def dask_setup(scheduler):
+    plugin = TaskLifecycleSchedulerPlugin()
+    scheduler.add_plugin(plugin)

services/dask-sidecar/src/simcore_service_dask_sidecar/task_life_cycle_worker_plugin.py

@@ -2,6 +2,7 @@
 from collections.abc import Awaitable
 from typing import Any
 
+import click
 from dask.typing import Key
 from dask_task_models_library.models import TASK_LIFE_CYCLE_EVENT, TaskLifeCycleState
 from distributed import WorkerPlugin
@@ -51,3 +52,9 @@ def transition(
                     key, kwargs.get("worker"), finish
                 ).model_dump(mode="json"),
             )
+
+
+@click.command()
+async def dask_setup(worker: Worker) -> None:
+    plugin = TaskLifecycleWorkerPlugin()
+    await worker.plugin_add(plugin)