Fix for crash reading in textual UVOT mode (#26)

Swift-SOT · Nov 13, 2024 · 2e8333d · 2e8333d
1 parent 7d24f58
commit 2e8333d
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/swifttools/swift_too/swift_instruments.py b/swifttools/swift_too/swift_instruments.py
@@ -1,4 +1,7 @@
 # Lookup table for XRT modes
+import re
+
+
 XRTMODES = {
     0: "Auto",
     1: "Null",
@@ -42,6 +45,11 @@ class TOOAPI_Instruments:
     def uvot_mode_setter(self, attr, mode):
         if type(mode) == str and "0x" in mode:
             """Convert hex string to int"""
+            uvot_mode = re.match(r"0x([0-9a-fA-F]+)", mode)
+            if uvot_mode is not None:
+                setattr(self, f"_{attr}", int(uvot_mode.group(0), 16))
+            else:
+                setattr(self, f"_{attr}", mode)
             setattr(self, f"_{attr}", int(mode.split(":")[0], 16))
         elif type(mode) == str:
             """Convert decimal string to int"""