Workflow to change experiment type id

Author: stephen-riggs

pyproject.toml

@@ -109,6 +109,7 @@ GitHub = "https://github.com/DiamondLightSource/python-murfey"
 "clem.register_preprocessing_result" = "murfey.workflows.clem.register_preprocessing_results:run"
 "data_collection" = "murfey.workflows.register_data_collection:run"
 "data_collection_group" = "murfey.workflows.register_data_collection_group:run"
+"experiment_type_update" = "murfey.workflows.register_experiment_type_update:run"
 "pato" = "murfey.workflows.notifications:notification_setup"
 "picked_particles" = "murfey.workflows.spa.picking:particles_picked"
 "picked_tomogram" = "murfey.workflows.tomo.picking:picked_tomogram"

src/murfey/server/api/workflow.py

@@ -155,6 +155,24 @@ def register_dc_group(
                 session_id, sm.name, search_map_params, db, close_db=False
             )
         db.close()
+    elif dcg_murfey := db.exec(
+        select(DataCollectionGroup)
+        .where(DataCollectionGroup.session_id == session_id)
+        .where(DataCollectionGroup.atlas == dcg_params.atlas)
+    ).all():
+        # Case where we switch from atlas to processing
+        dcg_murfey[0].tag = dcg_params.tag or dcg_murfey[0].tag
+        if _transport_object:
+            _transport_object.send(
+                _transport_object.feedback_queue,
+                {
+                    "register": "experiment_type_update",
+                    "experiment_type_id": dcg_params.experiment_type_id,
+                    "dcgid": dcg_murfey[0].id,
+                },
+            )
+        db.add(dcg_murfey[0])
+        db.commit()
     else:
         dcg_parameters = {
             "start_time": str(datetime.now()),

src/murfey/server/ispyb.py

@@ -137,6 +137,34 @@ def do_insert_data_collection_group(
             )
         return {"success": False, "return_value": None}
 
+    def do_update_data_collection_group(
+        self,
+        record: DataCollectionGroup,
+        message=None,
+        **kwargs,
+    ):
+        try:
+            with ISPyBSession() as db:
+                dcg = (
+                    db.query(DataCollectionGroup)
+                    .filter(
+                        DataCollectionGroup.dataCollectionGroupId
+                        == record.dataCollectionGroupId
+                    )
+                    .one()
+                )
+                dcg.experimentTypeId = record.experimentTypeId
+                db.add(dcg)
+                db.commit()
+                return {"success": True, "return_value": record.dataCollectionGroupId}
+        except ispyb.ISPyBException as e:
+            log.error(
+                "Updating Data Collection Group entry caused exception '%s'.",
+                e,
+                exc_info=True,
+            )
+        return {"success": False, "return_value": None}
+
     def do_insert_atlas(self, record: Atlas):
         try:
             with ISPyBSession() as db:

src/murfey/workflows/register_experiment_type_update.py

@@ -0,0 +1,39 @@
+import logging
+import time
+
+import ispyb.sqlalchemy._auto_db_schema as ISPyBDB
+from sqlmodel.orm.session import Session as SQLModelSession
+
+from murfey.server import _transport_object
+
+logger = logging.getLogger("murfey.workflows.register_data_collection_group")
+
+
+def run(
+    message: dict, murfey_db: SQLModelSession, demo: bool = False
+) -> dict[str, bool]:
+    # Fail immediately if no transport wrapper is found
+    if _transport_object is None:
+        logger.error("Unable to find transport manager")
+        return {"success": False, "requeue": False}
+
+    logger.info(f"Updating the experiment type for data collection group: \n{message}")
+
+    record = ISPyBDB.DataCollectionGroup(
+        dataCollectionGroupId=message["dcgid"],
+        experimentTypeId=message["experiment_type_id"],
+    )
+    dcgid = _transport_object.do_insert_data_collection_group(record).get(
+        "return_value", None
+    )
+
+    if dcgid is None:
+        time.sleep(2)
+        logger.error(
+            "Failed to update the following data collection group: \n"
+            f"{message} \n"
+            "Requeuing message"
+        )
+        return {"success": False, "requeue": True}
+
+    return {"success": True}