try to parse datetime columns as iso-format string first

dylanljones · Oct 7, 2023 · 991b0f7 · 991b0f7
1 parent ad9911c
commit 991b0f7
Showing 1 changed file with 12 additions and 8 deletions.
diff --git a/pyrekordbox/db6/tables.py b/pyrekordbox/db6/tables.py
@@ -73,14 +73,18 @@ def process_bind_param(self, value, dialect):
 
     def process_result_value(self, value, dialect):
         if value:
-            if len(value.strip()) > 23:
-                datestr, tzinfo = value[:23], value[23:]
-                datestr = datestr.strip()
-                tzinfo = tzinfo.strip()
-                assert tzinfo == "+00:00", tzinfo
-            else:
-                datestr, tzinfo = value, ""
-            return datetime.fromisoformat(datestr)
+            try:
+                dt = datetime.fromisoformat(value)
+            except ValueError:
+                if len(value.strip()) > 23:
+                    datestr, tzinfo = value[:23], value[23:]
+                    datestr = datestr.strip()
+                    tzinfo = tzinfo.strip()
+                    assert tzinfo == "+00:00", tzinfo
+                else:
+                    datestr, tzinfo = value, ""
+                dt = datetime.fromisoformat(datestr)
+            return dt
         return None