0.0.6 Fixes a bug causing error messages in log

gadget-man · Jul 16, 2021 · cc0d26f · cc0d26f
1 parent ea3ec48
commit cc0d26f
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 7 deletions.
diff --git a/binary_sensor.py b/binary_sensor.py
@@ -148,7 +148,7 @@ async def async_will_remove_from_hass(self) -> None:
     @callback
     def _update_callback(self, data):
         """Call update method."""
-        if (data[self._sensor]!= ''):
+        if self._sensor in data:
             _LOGGER.debug("Binary Sensor: %s - %s", self._sensor, data[self._sensor])
             self._state = data[self._sensor]
             self.async_schedule_update_ha_state(True)

diff --git a/sensor.py b/sensor.py
@@ -130,19 +130,19 @@ async def async_will_remove_from_hass(self) -> None:
 
     @callback
     def _update_callback(self, data):
-        if (data[self._sensor]!= ''):
+        if self._sensor in data:
             """Call update method."""
             _LOGGER.debug("iParcelBox Sensor callback: %s", data[self._sensor])
             self._state = data[self._sensor]
             self.async_schedule_update_ha_state(True)
 
     @callback
     def _update_message_callback(self, data):
-        # if (data[self._sensor]!= ''):
-        """Call update method."""
-        _LOGGER.debug("iParcelBox Message callback: %s", data[self._sensor])
-        self._state = data[self._sensor]
-        self.async_schedule_update_ha_state(True)
+        if self._sensor in data:
+            """Call update method."""
+            _LOGGER.debug("iParcelBox Message callback: %s", data[self._sensor])
+            self._state = data[self._sensor]
+            self.async_schedule_update_ha_state(True)
 
 
 # @asyncio.coroutine