Fixed unknown archive interval

MarcoGos · Dec 24, 2024 · 50d9ef0 · 50d9ef0
1 parent fdb0622
commit 50d9ef0
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/custom_components/davis_vantage/client.py b/custom_components/davis_vantage/client.py
@@ -90,10 +90,13 @@ def get_current_data(
         except Exception:
             pass
 
+        if self._hass.data.get(DATA_ARCHIVE_PERIOD, None) is None:
+            self._hass.data.setdefault(DATA_ARCHIVE_PERIOD, self._vantagepro2.archive_period)
+
         try:
             end_datetime = datetime.now()
             start_datetime = end_datetime - \
-                timedelta(minutes=self._hass.data.get(DATA_ARCHIVE_PERIOD) * 2)  # type: ignore
+                timedelta(minutes=self._vantagepro2.archive_period * 2)  # type: ignore
             archives = self._vantagepro2.get_archives(start_datetime, end_datetime)  # type: ignore
         except Exception:
             pass
@@ -146,7 +149,7 @@ async def async_get_current_data(self) -> LoopDataParserRevB | None:
         return data
 
     def __get_full_raw_data(self, data: LoopDataParserRevB) -> DataParser:
-        raw_data = DataParser(data.raw_bytes, LoopDataParserRevB.LOOP_FORMAT)
+        raw_data = DataParser(data.raw_bytes, LoopDataParserRevB.LOOP_FORMAT)  # type: ignore
         raw_data["HumExtra"] = struct.unpack(b"7B", raw_data["HumExtra"])  # type: ignore
         raw_data["ExtraTemps"] = struct.unpack(b"7B", raw_data["ExtraTemps"])  # type: ignore
         raw_data["SoilMoist"] = struct.unpack(b"4B", raw_data["SoilMoist"])  # type: ignore
@@ -162,7 +165,7 @@ def __get_full_raw_data(self, data: LoopDataParserRevB) -> DataParser:
         return raw_data
 
     def __get_full_raw_data_hilows(self, data: HighLowParserRevB) -> DataParser:
-        raw_data = DataParser(data.raw_bytes, HighLowParserRevB.HILOWS_FORMAT)
+        raw_data = DataParser(data.raw_bytes, HighLowParserRevB.HILOWS_FORMAT)  # type: ignore
         return raw_data
 
     def get_davis_time(self) -> datetime | None: