From f716b77554a169e56d1fd6eed5460794535c8a2f Mon Sep 17 00:00:00 2001
From: rettigl <l.rettig@web.de>
Date: Tue, 11 Feb 2025 12:11:47 +0100
Subject: [PATCH] fix polarization and pump status conversions

---
 src/specsscan/metadata.py | 25 +++++++++++++++++++++++--
 1 file changed, 23 insertions(+), 2 deletions(-)

diff --git a/src/specsscan/metadata.py b/src/specsscan/metadata.py
index 6b45809..b5052fe 100644
--- a/src/specsscan/metadata.py
+++ b/src/specsscan/metadata.py
@@ -265,6 +265,13 @@ def fetch_elab_metadata(self, scan: int, metadata: dict) -> dict:
                 metadata["elabFTW"]["scan"]["pump_polarization"] = 90
             elif metadata["elabFTW"]["scan"]["pump_polarization"] == "p":
                 metadata["elabFTW"]["scan"]["pump_polarization"] = 0
+            else:
+                try:
+                    metadata["elabFTW"]["scan"]["pump_polarization"] = float(
+                        metadata["elabFTW"]["scan"]["pump_polarization"],
+                    )
+                except ValueError:
+                    pass
 
         if (
             "scan" in metadata["elabFTW"]
@@ -275,6 +282,13 @@ def fetch_elab_metadata(self, scan: int, metadata: dict) -> dict:
                 metadata["elabFTW"]["scan"]["probe_polarization"] = 90
             elif metadata["elabFTW"]["scan"]["probe_polarization"] == "p":
                 metadata["elabFTW"]["scan"]["probe_polarization"] = 0
+            else:
+                try:
+                    metadata["elabFTW"]["scan"]["probe_polarization"] = float(
+                        metadata["elabFTW"]["scan"]["probe_polarization"],
+                    )
+                except ValueError:
+                    pass
 
         if (
             "scan" in metadata["elabFTW"]
@@ -285,19 +299,26 @@ def fetch_elab_metadata(self, scan: int, metadata: dict) -> dict:
                 metadata["elabFTW"]["scan"]["pump2_polarization"] = 90
             elif metadata["elabFTW"]["scan"]["pump2_polarization"] == "p":
                 metadata["elabFTW"]["scan"]["pump2_polarization"] = 0
+            else:
+                try:
+                    metadata["elabFTW"]["scan"]["pump2_polarization"] = float(
+                        metadata["elabFTW"]["scan"]["pump2_polarization"],
+                    )
+                except ValueError:
+                    pass
 
         # 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"
+                    "open" 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"
+                    "open" if int(metadata["elabFTW"]["scan"]["pump2_status"]) else "closed"
                 )
             except ValueError:
                 pass