Allow extra configuration components to be added by a key found in the murfey.config entry point and validate against the provided model

Also provide a convenience function for extracting the relevant part of the extended configuration from other packages if Murfey is installed

Author: d-j-hatton

pyproject.toml

@@ -95,6 +95,8 @@ murfey = "murfey.client:run"
 "murfey.transfer" = "murfey.cli.transfer:run"
 [project.entry-points."murfey.auth.token_validation"]
 "password" = "murfey.server.api.auth:password_token_validation"
+[project.entry-points."murfey.config.extraction"]
+"murfey_machine" = "murfey.util.config:get_extended_machine_config"
 [project.entry-points."murfey.workflows"]
 "lif_to_stack" = "murfey.workflows.lif_to_stack:zocalo_cluster_request"
 "tiff_to_stack" = "murfey.workflows.tiff_to_stack:zocalo_cluster_request"

src/murfey/util/config.py

@@ -7,6 +7,7 @@
 from typing import Dict, List, Literal, Optional, Union
 
 import yaml
+from backports.entry_points_selectable import entry_points
 from pydantic import BaseModel, BaseSettings
 
 
@@ -157,3 +158,16 @@ def get_machine_config(instrument_name: str = "") -> Dict[str, MachineConfig]:
             Path(settings.murfey_machine_configuration), microscope
         )
     return machine_config
+
+
+def get_extended_machine_config(
+    extension_name: str, instrument_name: str = ""
+) -> Optional[BaseModel]:
+    machine_config = get_machine_config(instrument_name=instrument_name).get(
+        instrument_name
+    )
+    if not machine_config:
+        return None
+    model = entry_points.select(group="murfey.config", name=extension_name)[0].load()
+    data = getattr(machine_config, extension_name, {})
+    return model(data)