Add files via upload

Release 2.5.1

Author: HAEdwin

custom_components/apsystems_ecu_reader/__init__.py

@@ -18,15 +18,14 @@
     DataUpdateCoordinator,
     UpdateFailed,
 )
-from homeassistant.helpers.event import async_track_state_change_event
 
 from .const import DOMAIN
 from .ecu_api import APsystemsSocket, APsystemsInvalidData
-from .gui_helpers import set_inverter_state, set_zero_export, reboot_ecu, inverter_max_power
+from .gui_helpers import set_inverter_state, set_zero_export, reboot_ecu, set_inverter_max_power
 
 _LOGGER = logging.getLogger(__name__)
 
-PLATFORMS = ["sensor", "binary_sensor", "switch"]
+PLATFORMS = ["sensor", "binary_sensor", "switch", "number"]
 
 class ECUREADER:
     """ECU Reader"""
@@ -40,6 +39,11 @@ def __init__(self, ipaddr, wifi_ssid, wifi_password, show_graphs):
         self.data_from_cache_count = 0
         self.cached_data = {}
 
+    # called from number.py
+    async def set_inverter_max_power(self, inverter_uid, max_panel_power):
+        """Set the max power for an inverter."""
+        return await set_inverter_max_power(self.ipaddr, inverter_uid, max_panel_power)
+
     # called from switch.py
     async def set_inverter_state(self, inverter_id, state):
         """Set the on/off state of an inverter. 1=on, 2=off"""
@@ -97,7 +101,7 @@ async def update_listener(hass, config):
 
 
 async def async_setup_entry(hass, config):
-    """Setup APsystems platform."""
+    """ Setup APsystems platform """
     hass.data.setdefault(DOMAIN, {})
     interval = timedelta(seconds=config.data.get("scan_interval", 300))
     ecu = ECUREADER(
@@ -108,13 +112,14 @@ async def async_setup_entry(hass, config):
     )
 
     async def do_ecu_update():
-        """Pass current port_retries value dynamically."""
+        """ Pass current port_retries value dynamically. """
         return await ecu.update(
             config.data.get("port_retries", 2),
             config.data.get("cache_reboot", 3),
             config.data.get("show_graphs", True)
         )
 
+
     coordinator = DataUpdateCoordinator(
         hass,
         _LOGGER,
@@ -138,7 +143,7 @@ async def do_ecu_update():
     # If not, the user should be notified and devices should not be created.
     if not ecu.ecu.ecu_id:
         _LOGGER.error(
-            "Unable to connect with ECU @ %s. Check IP-Address or wait "
+           "Unable to connect with ECU @ %s. Check IP-Address or wait "
             "10 minutes because the ECU might be recovering from reboot.",
             config.data["ecu_host"]
         )
@@ -171,25 +176,9 @@ async def do_ecu_update():
     # Forward platform setup requests.
     await hass.config_entries.async_forward_entry_setups(config, PLATFORMS)
     config.async_on_unload(config.add_update_listener(update_listener))
-
-    # Add listeners for state changes of the input_number entities
-    async def max_panel_power_listener(event):
-        """Handle state changes of the input_number entity."""
-        new_state = event.data.get("new_state")
-        if new_state is None:
-            return
-        new_value = new_state.state
-        await inverter_max_power(ecu.ipaddr, event.data["entity_id"].split('.')[-1], new_value)
-
-    # Create listeners for each inverter UID
-    max_panel_power_entity_ids = [
-        f"max_panel_power.{uid}" for uid in inverters.keys()
-    ]
-    _LOGGER.debug("Adding listeners for input_number entities: %s",max_panel_power_entity_ids)
-    for entity_id in max_panel_power_entity_ids:
-        async_track_state_change_event(hass, entity_id, max_panel_power_listener)
     return True
 
+
 async def async_remove_config_entry_device(hass, _, device_entry) -> bool:
     """ Handle device removal """	
     if device_entry:

custom_components/apsystems_ecu_reader/ecu_api.py

@@ -173,7 +173,10 @@ def finalize_data(self, show_graphs):
             # apply filters for ECU firmware bug where sometimes values are zero
             if self.qty_of_inverters:
                 self.data["qty_of_inverters"] = self.qty_of_inverters
-            if self.today_energy and self.qty_of_online_inverters:
+            if (
+                self.today_energy != 0 or
+                (self.today_energy == 0 and self.qty_of_online_inverters == 0)
+            ):
                 self.data["today_energy"] = self.today_energy
             if self.lifetime_energy:
                 self.data["lifetime_energy"] = self.lifetime_energy

custom_components/apsystems_ecu_reader/gui_helpers.py

@@ -56,7 +56,7 @@ async def set_zero_export(ipaddr, state):
             "This switch is only compatible with ECU-C models", err
         )
 
-async def inverter_max_power(ipaddr, inverter_uid, max_panel_power):
+async def set_inverter_max_power(ipaddr, inverter_uid, max_panel_power):
     """Set the max power for an inverter."""
     action = {
         "id": inverter_uid,

custom_components/apsystems_ecu_reader/number.py

@@ -0,0 +1,52 @@
+"""Number platform for APsystems ECU Reader."""
+
+from homeassistant.components.number import RestoreNumber
+from homeassistant.helpers.update_coordinator import CoordinatorEntity
+
+from .const import DOMAIN
+
+async def async_setup_entry(hass, _, async_add_entities):
+    """Set up the number platform."""
+    coordinator = hass.data[DOMAIN]["coordinator"]
+    ecu = hass.data[DOMAIN]["ecu"]
+
+    entities = []
+    for inverter_id, inverter_data in coordinator.data.get("inverters", {}).items():
+        entities.append(InverterMaxPwrNumber(coordinator, ecu, inverter_id, inverter_data))
+
+    async_add_entities(entities, True)
+
+class InverterMaxPwrNumber(CoordinatorEntity, RestoreNumber):
+    """Representation of an Inverter_MaxPwr Number entity."""
+
+    #def set_native_value(self, value: float) -> None:
+    #    """Set new value."""
+    #    self._attr_native_value = value
+    #    self.async_write_ha_state()
+
+    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._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_device_class = "power"  # Set device class
+        self._attr_mode = "slider"  # Set mode to slider
+
+    async def async_set_native_value(self, value: float):
+        """Update the current value."""
+        await self.ecu.set_inverter_max_power(self.inverter_id, value)
+        self._attr_native_value = value
+        self.async_write_ha_state()
+
+    async def async_added_to_hass(self):
+        """Handle entity which value will be restored."""
+        await super().async_added_to_hass()
+        if (last_state := await self.async_get_last_state()) is not None:
+            self._attr_native_value = last_state.state

custom_components/apsystems_ecu_reader/sensor.py

@@ -36,7 +36,6 @@
     CACHE_COUNTER_ICON
 )
 
-
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, _, add_entities):