Handle token exiration time not being correct type

nickknissen · Jul 12, 2023 · 25c1a85 · 25c1a85
1 parent a4508ab
commit 25c1a85
Showing 1 changed file with 11 additions and 1 deletion.
diff --git a/custom_components/monta/api.py b/custom_components/monta/api.py
@@ -283,6 +283,11 @@ def _is_access_token_valid(self):
         if self._prefs[STORAGE_ACCESS_EXPIRE_TIME] is None:
             return False
 
+        _LOGGER.debug("[_is_access_token_valid] STORAGE_ACCESS_EXPIRE_TIME: %s, type: %s", self._prefs[STORAGE_ACCESS_EXPIRE_TIME], type(self._prefs[STORAGE_ACCESS_EXPIRE_TIME]))
+
+        if not isinstance(self._prefs[STORAGE_ACCESS_EXPIRE_TIME], str):
+            return False
+
         expire_time = dt_util.parse_datetime(self._prefs[STORAGE_ACCESS_EXPIRE_TIME])
         preemptive_expire_time = expire_time - timedelta(
             seconds=PREEMPTIVE_REFRESH_TTL_IN_SECONDS
@@ -295,7 +300,12 @@ def _is_refresh_token_valid(self):
         if not self._prefs[STORAGE_REFRESH_TOKEN]:
             return False
 
-        if self._prefs[STORAGE_ACCESS_EXPIRE_TIME] is None:
+        if self._prefs[STORAGE_REFRESH_EXPIRE_TIME] is None:
+            return False
+
+        _LOGGER.debug("[_is_refresh_token_valid] STORAGE_REFRESH_EXPIRE_TIME: %s, type: %s", self._prefs[STORAGE_ACCESS_EXPIRE_TIME], type(self._prefs[STORAGE_REFRESH_EXPIRE_TIME]))
+
+        if not isinstance(self._prefs[STORAGE_REFRESH_EXPIRE_TIME], str):
             return False
 
         expire_time = dt_util.parse_datetime(self._prefs[STORAGE_REFRESH_EXPIRE_TIME])