Small fixes

Author: Bre77

custom_components/teslemetry/climate.py

@@ -259,7 +259,10 @@ async def async_added_to_hass(self) -> None:
         """Handle entity which will be added."""
         await super().async_added_to_hass()
         if (state := await self.async_get_last_state()) is not None:
-            self._attr_hvac_mode = HVACMode(state.state) if (state.state and state.state != "unknown") else None
+            try:
+                self._attr_hvac_mode = HVACMode(state.state)
+            except ValueError:
+                self._attr_hvac_mode = None
             self._attr_current_temperature = state.attributes.get('current_temperature')
             self._attr_target_temperature = state.attributes.get('target_temperature')
             #self._attr_fan_mode = state.attributes.get('fan_mode')

custom_components/teslemetry/media_player.py

@@ -185,7 +185,10 @@ async def async_added_to_hass(self) -> None:
 
         await super().async_added_to_hass()
         if (state := await self.async_get_last_state()) is not None:
-            self._attr_state = None if state.state == "Unknown" else MediaPlayerState(state.state)
+            try:
+                self._attr_state = MediaPlayerState(state.state)
+            except ValueError:
+                self._attr_state = None
             self._attr_volume_level = state.attributes.get("volume_level")
             self._attr_media_title = state.attributes.get("media_title")
             self._attr_media_artist = state.attributes.get("media_artist")