Migrated 'atlas_update' feedback-callback block to its own workflows module

Author: tieneupin

pyproject.toml

@@ -101,6 +101,7 @@ GitHub = "https://github.com/DiamondLightSource/python-murfey"
 [project.entry-points."murfey.config.extraction"]
 "murfey_machine" = "murfey.util.config:get_extended_machine_config"
 [project.entry-points."murfey.workflows"]
+"atlas_update" = "murfey.workflows.register_atlas_update:run"
 "clem.align_and_merge" = "murfey.workflows.clem.align_and_merge:submit_cluster_request"
 "clem.process_raw_lifs" = "murfey.workflows.clem.process_raw_lifs:zocalo_cluster_request"
 "clem.process_raw_tiffs" = "murfey.workflows.clem.process_raw_tiffs:zocalo_cluster_request"

src/murfey/server/feedback.py

@@ -2013,26 +2013,6 @@ def feedback_callback(header: dict, message: dict, _db=murfey_db) -> None:
             if murfey.server._transport_object:
                 murfey.server._transport_object.transport.ack(header)
             return None
-        elif message["register"] == "atlas_update":
-            if murfey.server._transport_object:
-                murfey.server._transport_object.do_update_atlas(
-                    message["atlas_id"],
-                    message["atlas"],
-                    message["atlas_pixel_size"],
-                    message["sample"],
-                )
-                murfey.server._transport_object.transport.ack(header)
-            if dcg_hooks := entry_points().select(
-                group="murfey.hooks", name="data_collection_group"
-            ):
-                try:
-                    for hook in dcg_hooks:
-                        hook.load()(message["dcgid"], session_id=message["session_id"])
-                except Exception:
-                    logger.error(
-                        "Call to data collection group hook failed", exc_info=True
-                    )
-            return None
         elif message["register"] == "processing_job":
             murfey_session_id = message["session_id"]
             logger.info("registering processing job")

src/murfey/workflows/register_atlas_update.py

@@ -0,0 +1,33 @@
+import logging
+
+from backports.entry_points_selectable import entry_points
+from sqlmodel.orm.session import Session as SQLModelSession
+
+from murfey.server import _transport_object
+
+logger = logging.getLogger("murfey.workflows.register_atlas_update")
+
+
+def run(
+    message: dict,
+    murfey_db: SQLModelSession,  # Defined for compatibility but unused
+    demo: bool = False,
+):
+    if _transport_object is None:
+        logger.error("Unable to find transport manager")
+        return {"success": False, "requeue": False}
+    _transport_object.do_update_atlas(
+        message["atlas_id"],
+        message["atlas"],
+        message["atlas_pixel_size"],
+        message["sample"],
+    )
+    if dcg_hooks := entry_points().select(
+        group="murfey.hooks", name="data_collection_group"
+    ):
+        try:
+            for hook in dcg_hooks:
+                hook.load()(message["dcgid"], session_id=message["session_id"])
+        except Exception:
+            logger.error("Call to data collection group hook failed", exc_info=True)
+    return {"success": True}