Fix none issue with session energy usage

Author: dan-r

custom_components/ohme/const.py

@@ -1,7 +1,7 @@
 """Component constants"""
 DOMAIN = "ohme"
 USER_AGENT = "dan-r-homeassistant-ohme"
-INTEGRATION_VERSION = "0.6.1"
+INTEGRATION_VERSION = "0.6.2"
 CONFIG_VERSION = 1
 ENTITY_TYPES = ["sensor", "binary_sensor", "switch", "button", "number", "time"]
 

custom_components/ohme/sensor.py

@@ -293,7 +293,7 @@ def _handle_coordinator_update(self) -> None:
             new_state = self.coordinator.data['batterySoc']['wh']
 
             # Let the state reset to 0, but not drop otherwise
-            if new_state <= 0:
+            if not new_state or new_state <= 0:
                 self._state = 0
             else:
                 self._state = max(0, self._state or 0, new_state)