Merge pull request #17 from drc38/available

ocpp: set unavailable state on charger disconnect

Author: drc38

custom_components/ocpp/api.py

@@ -6,7 +6,7 @@
 from typing import Dict
 
 from homeassistant.config_entries import ConfigEntry
-from homeassistant.const import TIME_MINUTES
+from homeassistant.const import STATE_OK, STATE_UNAVAILABLE, TIME_MINUTES
 from homeassistant.core import HomeAssistant
 from homeassistant.helpers import device_registry, entity_component, entity_registry
 import voluptuous as vol
@@ -163,6 +163,7 @@ async def on_connect(self, websocket, path: str):
         except Exception as e:
             _LOGGER.info(f"Exception occurred:\n{e}")
         finally:
+            self.charge_points[cp_id].status = STATE_UNAVAILABLE
             _LOGGER.info(f"Charger {cp_id} disconnected from {self.host}:{self.port}.")
 
     def get_metric(self, cp_id: str, measurand: str):
@@ -183,6 +184,12 @@ def get_extra_attr(self, cp_id: str, measurand: str):
             return self.charge_points[cp_id].get_extra_attr(measurand)
         return None
 
+    def get_available(self, cp_id: str):
+        """Return whether the charger is available."""
+        if cp_id in self.charge_points:
+            return self.charge_points[cp_id].status == STATE_OK
+        return False
+
     async def set_charger_state(
         self, cp_id: str, service_name: str, state: bool = True
     ):
@@ -349,6 +356,7 @@ async def handle_get_configuration(call):
                     handle_update_firmware,
                     UFW_SERVICE_DATA_SCHEMA,
                 )
+            self.status = STATE_OK
         except (NotImplementedError) as e:
             _LOGGER.error("Configuration of the charger failed: %s", e)
 

custom_components/ocpp/sensor.py

@@ -67,6 +67,11 @@ def state(self):
         """Return the state of the sensor."""
         return self.central_system.get_metric(self.cp_id, self.metric)
 
+    @property
+    def available(self) -> bool:
+        """Return if switch is available."""
+        return self.central_system.get_available(self.cp_id)
+
     @property
     def unit_of_measurement(self):
         """Return the unit the value is expressed in."""

custom_components/ocpp/switch.py

@@ -45,7 +45,7 @@ def unique_id(self):
     @property
     def available(self) -> bool:
         """Return if switch is available."""
-        return True  # type: ignore [no-any-return]
+        return self.central_system.get_available(self.cp_id)  # type: ignore [no-any-return]
 
     @property
     def is_on(self) -> bool: