Add files via upload

Preparing for release 2.6.1

Author: HAEdwin

custom_components/apsystems_ecu_reader/__init__.py

@@ -31,7 +31,7 @@
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORMS = ["sensor", "binary_sensor", "switch", "number"]
+PLATFORMS = ["sensor", "binary_sensor", "switch", "number", "button"]
 
 class ECUREADER:
     """ECU Reader"""

custom_components/apsystems_ecu_reader/button.py

@@ -0,0 +1,62 @@
+"""Button platform for APsystems ECU Reader."""
+
+from homeassistant.components.button import ButtonEntity, ButtonDeviceClass
+from homeassistant.helpers.entity import EntityCategory
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN, ECU_REBOOT_ICON
+from .gui_helpers import pers_notification
+
+
+import logging
+
+_LOGGER = logging.getLogger(__name__)
+
+async def async_setup_entry(hass, config_entry, async_add_entities):
+    """Set up the button platform."""
+    ecu = hass.data[DOMAIN]["ecu"]
+    async_add_entities([RebootECUButton(ecu)])
+
+
+class RebootECUButton(ButtonEntity):
+    """Representation of a button to reboot the ECU."""
+
+    def __init__(self, ecu):
+        """Initialize the button."""
+        self._ecu = ecu
+        self._attr_icon = ECU_REBOOT_ICON
+        self._attr_name = f"ECU {ecu.ecu.ecu_id} Reboot"
+        self._attr_unique_id = f"ECU_{ecu.ecu.ecu_id}_reboot_button"
+        self._attr_device_class = ButtonDeviceClass.RESTART
+
+    @property
+    def device_info(self):
+        """Return the device info for the ECU."""
+        return {
+            "identifiers": {
+                (DOMAIN, f"ecu_{self._ecu.ecu.ecu_id}"),
+            },
+            "name": f"ECU {self._ecu.ecu.ecu_id}",
+            "manufacturer": "APsystems",
+            "model": self._ecu.ecu.firmware,
+            "sw_version": self._ecu.ecu.firmware,
+        }
+
+    @property
+    def entity_category(self):
+        """Return the category of the entity."""
+        return EntityCategory.DIAGNOSTIC
+    
+    async def async_press(self):
+        """Handle the button press."""
+        try:
+            response = await self._ecu.reboot_ecu()
+            pers_notification(
+                self.hass,
+                f"Rebooted ECU {self._ecu.ecu.ecu_id}"
+            )
+        except Exception as e:
+            pers_notification(
+                self.hass,
+                f"Failed to reboot ECU: {e}"
+            )

custom_components/apsystems_ecu_reader/manifest.json

@@ -9,5 +9,5 @@
   "iot_class": "local_polling",
   "issue_tracker": "https://github.com/haedwin/homeassistant-apsystems_ecu_reader/issues",
   "loggers": ["custom_components.apsystems_ecu_reader"],
-  "version": "2.6.0"
+  "version": "2.6.1"
 }

custom_components/apsystems_ecu_reader/number.py

@@ -24,28 +24,29 @@ def __init__(self, coordinator, ecu, inverter_id, inverter_data):
         """Initialize the number entity."""
         super().__init__(coordinator)
         self._ecu = ecu
-        self.inverter_id = inverter_id
-        self.inverter_data = inverter_data
+        self._uid = inverter_id
+        self._inv_data = inverter_data
         self._attr_name = f"Inverter {inverter_id} Maxpwr"
         self._attr_unique_id = f"{DOMAIN}_inverter_{inverter_id}_maxpwr"
         self._attr_native_min_value = 20
         self._attr_native_max_value = 500
         self._attr_native_step = 1
-        self._attr_native_value = inverter_data.get("number_value", 0)
+        self._attr_native_value = self._inv_data.get("number_value", 0)
         self._attr_device_class = "power"  # Set device class
         self._attr_mode = "slider"  # Set mode to slider
 
     @property
     def device_info(self):
         """Return the device info."""
+        parent = f"inverter_{self._uid}"
         return {
             "identifiers": {
-                (DOMAIN, f"ecu_{self._ecu.ecu.ecu_id}"),
+                (DOMAIN, parent),
             },
-            "name": f"ECU {self._ecu.ecu.ecu_id}",
+            "name": f"Inverter {self._uid}",
             "manufacturer": "APsystems",
-            "model": self._ecu.ecu.firmware,
-            "sw_version": self._ecu.ecu.firmware,
+            "model": self._inv_data.get("model", "Unknown"),
+            "via_device": (DOMAIN, f"ecu_{self._ecu.ecu.ecu_id}"),
         }
 
     @property
@@ -55,7 +56,7 @@ def entity_category(self):
 
     async def async_set_native_value(self, value: float):
         """Update the current value."""
-        await self._ecu.set_inverter_max_power(self.inverter_id, value)
+        await self._ecu.set_inverter_max_power(self._uid, value)
         self._attr_native_value = value
         self.async_write_ha_state()
 

custom_components/apsystems_ecu_reader/switch.py

@@ -8,7 +8,7 @@
 from homeassistant.helpers.restore_state import RestoreEntity
 from homeassistant.helpers.update_coordinator import CoordinatorEntity
 
-from .const import DOMAIN, POWER_ICON, ECU_REBOOT_ICON
+from .const import DOMAIN, POWER_ICON
 from .gui_helpers import pers_notification
 
 _LOGGER = logging.getLogger(__name__)
@@ -22,7 +22,6 @@ async def async_setup_entry(hass, config_entry, async_add_entities):
     for uid, inv_data in inverters.items():
         switches.append(APsystemsECUInverterSwitch(coordinator, ecu, uid, inv_data))
     switches.append(APsystemsZeroExportSwitch(coordinator, ecu))
-    switches.append(RebootECUSwitch(ecu))
     switches.append(APsystemsAllInvertersSwitch(coordinator, ecu))
     async_add_entities(switches)
 
@@ -198,78 +197,3 @@ async def async_turn_off(self, **kwargs):
             self.async_write_ha_state()
         except Exception as e:
             _LOGGER.error("Failed to turn off zero export: %s", e)
-
-class RebootECUSwitch(SwitchEntity, RestoreEntity):
-    """Momentary switch to reboot the ECU."""
-
-    def __init__(self, ecu):
-        """Initialize the switch."""
-        self._ecu = ecu
-        self._state = False
-        self._name = f"ECU {ecu.ecu.ecu_id} Reboot Switch"
-        self._unique_id = f"ECU_{ecu.ecu.ecu_id}_reboot_switch"
-
-    @property
-    def unique_id(self):
-        """Return the unique ID of the switch."""
-        return self._unique_id
-
-    @property
-    def name(self):
-        """Return the name of the switch."""
-        return self._name
-
-    @property
-    def icon(self):
-        """Return the icon to use in the UI."""
-        return ECU_REBOOT_ICON
-
-    @property
-    def device_info(self):
-        """Return the device info for the ECU."""
-        return {
-            "identifiers": {
-                (DOMAIN, f"ecu_{self._ecu.ecu.ecu_id}"),
-            },
-            "name": f"ECU {self._ecu.ecu.ecu_id}",
-            "manufacturer": "APsystems",
-            "model": self._ecu.ecu.firmware,
-            "sw_version": self._ecu.ecu.firmware,
-        }
-
-    @property
-    def entity_category(self):
-        """Return the category of the entity."""
-        return EntityCategory.DIAGNOSTIC
-
-    @property
-    def is_on(self):
-        """Return the state of the switch."""
-        return self._state
-
-    async def async_turn_on(self, **kwargs):
-        """Reboot the ECU."""
-        self._state = True
-        self.async_write_ha_state()
-
-        try:
-            await self._ecu.reboot_ecu()
-            pers_notification(
-                self.hass,
-                f"Rebooted ECU {self._ecu.ecu.ecu_id}"
-            )
-        except Exception as e:
-            _LOGGER.error("Failed to reboot ECU: %s", e)
-            pers_notification(
-                self.hass,
-                f"Failed to reboot ECU: {e}"
-            )
-
-        # Turn off the switch after 2 seconds
-        await asyncio.sleep(2)
-        self._state = False
-        self.async_write_ha_state()
-
-    async def async_turn_off(self, **kwargs):
-        """Handle turning the switch off."""
-        pass  # Momentary switch, no manual turn-off