fix: handle missing prices (#116)

firstof9 · Nov 22, 2024 · 6574956 · 6574956
1 parent d63eb5d
commit 6574956
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/custom_components/gasbuddy/sensor.py b/custom_components/gasbuddy/sensor.py
@@ -80,12 +80,18 @@ def native_value(self) -> Any:
             self._state = None
         if self._type in data.keys():
             if self._cash and "cash_price" in data[self._type]:
-                if self.coordinator.data["unit_of_measure"] == "cents_per_liter":
+                if (
+                    self.coordinator.data["unit_of_measure"] == "cents_per_liter"
+                    and data[self._type]["cash_price"]
+                ):
                     self._state = data[self._type]["cash_price"] / 100
                 else:
                     self._state = data[self._type]["cash_price"]
             else:
-                if self.coordinator.data["unit_of_measure"] == "cents_per_liter":
+                if (
+                    self.coordinator.data["unit_of_measure"] == "cents_per_liter"
+                    and data[self._type]["price"]
+                ):
                     self._state = data[self._type]["price"] / 100
                 else:
                     self._state = data[self._type]["price"]