Allow updates to be made to data collection group experiment type

Author: d-j-hatton

src/murfey/server/__init__.py

@@ -2575,6 +2575,8 @@ def feedback_callback(header: dict, message: dict) -> None:
                     message["atlas"],
                     message["atlas_pixel_size"],
                     message["sample"],
+                    experiment_type=message["experiment_type"],
+                    experiment_type_id=message["experiment_type_id"],
                 )
                 _transport_object.transport.ack(header)
             return None

src/murfey/server/api/__init__.py

@@ -1327,6 +1327,8 @@ def register_dc_group(
                         "atlas": dcg_params.atlas,
                         "sample": dcg_params.sample,
                         "atlas_pixel_size": dcg_params.atlas_pixel_size,
+                        "experiment_type": dcg_params.experiment_type,
+                        "experiment_type_id": dcg_params.experiment_type_id,
                     },
                 )
             else:

src/murfey/server/ispyb.py

@@ -115,11 +115,28 @@ def do_insert_atlas(self, record: Atlas):
         return {"success": False, "return_value": None}
 
     def do_update_atlas(
-        self, atlas_id: int, atlas_image: str, pixel_size: float, slot: int
+        self,
+        atlas_id: int,
+        atlas_image: str,
+        pixel_size: float,
+        slot: int,
+        experiment_type: str = "",
+        experiment_type_id: Optional[int] = None,
     ):
         try:
             with Session() as db:
                 atlas = db.query(Atlas).filter(Atlas.atlasId == atlas_id).one()
+                dcg = (
+                    db.query(DataCollectionGroup)
+                    .filter(
+                        DataCollectionGroup.dataCollectionGroupId
+                        == atlas.dataCollectionGroupId
+                    )
+                    .one()
+                )
+                if experiment_type and experiment_type_id is not None:
+                    dcg.experimentTypeId = experiment_type_id
+                    db.add(dcg)
                 atlas.atlasImage = atlas_image or atlas.atlasImage
                 atlas.pixelSize = pixel_size or atlas.pixelSize
                 atlas.cassetteSlot = slot or atlas.cassetteSlot