fix: fix error for cars without heated steering wheels and seats

zabuldon · Apr 24, 2022 · b5cf9bb · b5cf9bb
1 parent 26025b2
commit b5cf9bb
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/teslajsonpy/homeassistant/heated_seats.py b/teslajsonpy/homeassistant/heated_seats.py
@@ -70,7 +70,7 @@ def refresh(self) -> None:
         if last_update >= self.__manual_update_time:
             data = self._controller.get_climate_params(self._id)
             self.__seat_heat_level = (
-                data[f"seat_heater_{self.__seat_name}"] if data else None
+                data.get(f"seat_heater_{self.__seat_name}") if data else None
             )
 
     async def set_seat_heat_level(self, level):

diff --git a/teslajsonpy/homeassistant/heated_steering_wheel.py b/teslajsonpy/homeassistant/heated_steering_wheel.py
@@ -59,7 +59,7 @@ def refresh(self) -> None:
         if last_update >= self.__manual_update_time:
             data = self._controller.get_climate_params(self._id)
             self.__steering_wheel_heated = (
-                data["steering_wheel_heater"] if data else None
+                data.get("steering_wheel_heater") if data else None
             )
 
     async def set_steering_wheel_heat(self, value: bool):