fix beam status

OpenCOMPES · Feb 10, 2025 · 4efb02d · 4efb02d
1 parent a988e6b
commit 4efb02d
Showing 1 changed file with 16 additions and 0 deletions.
diff --git a/src/specsscan/metadata.py b/src/specsscan/metadata.py
@@ -286,6 +286,22 @@ def fetch_elab_metadata(self, scan: int, metadata: dict) -> dict:
             elif metadata["elabFTW"]["scan"]["pump2_polarization"] == "p":
                 metadata["elabFTW"]["scan"]["pump2_polarization"] = 0
 
+        # fix pump status
+        if "scan" in metadata["elabFTW"] and "pump_status" in metadata["elabFTW"]["scan"]:
+            try:
+                metadata["elabFTW"]["scan"]["pump_status"] = (
+                    "opened" if int(metadata["elabFTW"]["scan"]["pump_status"]) else "closed"
+                )
+            except ValueError:
+                pass
+        if "scan" in metadata["elabFTW"] and "pump2_status" in metadata["elabFTW"]["scan"]:
+            try:
+                metadata["elabFTW"]["scan"]["pump2_status"] = (
+                    "opened" if int(metadata["elabFTW"]["scan"]["pump2_status"]) else "closed"
+                )
+            except ValueError:
+                pass
+
         # remove pump information if pump not applied:
         if metadata["elabFTW"]["scan"].get("pump_status", "closed") == "closed":
             if "pump_photon_energy" in metadata["elabFTW"].get("laser_status", {}):