diff --git a/src/specsscan/metadata.py b/src/specsscan/metadata.py index e2c1776..6b45809 100644 --- 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", {}):