Only load rated information on boot

Author: ArnyminerZ

custom_components/axpert_inverter/coordinator.py

@@ -34,6 +34,7 @@ def __init__(self, hass: HomeAssistant, inverter: AxpertInverter, entry: ConfigE
         self.firmware_version = None
         self.model_id = None
         self.model_name = None
+        self.rated_information = None
 
     async def _async_update_data(self):
         """Update data via library."""
@@ -56,6 +57,9 @@ def _io_update(self):
         if not self.model_name:
             self.model_name = self.inverter.get_model_name()
 
+        if not self.rated_information:
+            self.rated_information = self.inverter.get_rated_information()
+
         data = self.inverter.get_general_status()
         if not data:
             raise UpdateFailed("Received empty data from QPIGS")
@@ -70,9 +74,8 @@ def _io_update(self):
         if mode:
             data["mode"] = mode
 
-        # Get QPIRI for selectors
-        qpiri_data = self.inverter.get_rated_information()
-        if qpiri_data:
-            data.update(qpiri_data)
+        # Add rated information (QPIRI) for selectors
+        if self.rated_information:
+            data.update(self.rated_information)
 
         return data

custom_components/axpert_inverter/number.py

@@ -47,7 +47,8 @@ async def _async_set_value_generic(self, func, value):
             self._attr_available = False
             self.async_write_ha_state()
         else:
-            # Refresh data
+            # Invalidate cache and refresh data
+            self.coordinator.rated_information = None
             await self.coordinator.async_request_refresh()
 
 class AxpertMaxChargingCurrent(AxpertNumberEntity):

custom_components/axpert_inverter/select.py

@@ -58,7 +58,9 @@ async def async_select_option(self, option: str) -> None:
 
         if success:
             self._attr_current_option = option
+            self.coordinator.rated_information = None
             self.async_write_ha_state()
+            await self.coordinator.async_request_refresh()
             _LOGGER.info(f"Set AC Input Range to {option}")
         else:
             _LOGGER.error(f"Failed to set AC Input Range to {option}")
@@ -116,9 +118,8 @@ async def async_select_option(self, option: str) -> None:
             self.coordinator.inverter.set_output_source_priority, val
         )
         if success:
-            # Optimistically update? Or wait for poll.
-            # Wait for poll is safer but slower. 
-            pass
+            self.coordinator.rated_information = None
+            await self.coordinator.async_request_refresh()
 
 class AxpertChargerPrioritySelect(AxpertEntity, SelectEntity):
     """Select entity for Charger Source Priority."""
@@ -155,7 +156,8 @@ async def async_select_option(self, option: str) -> None:
             self.coordinator.inverter.set_charger_source_priority, val
         )
         if success:
-            pass
+            self.coordinator.rated_information = None
+            await self.coordinator.async_request_refresh()
 
 
 class AxpertBatteryTypeSelect(AxpertEntity, SelectEntity):
@@ -191,7 +193,10 @@ async def async_select_option(self, option: str) -> None:
         success = await self.hass.async_add_executor_job(
             self.coordinator.inverter.set_battery_type, val
         )
-        if not success:
+        if success:
+            self.coordinator.rated_information = None
+            await self.coordinator.async_request_refresh()
+        else:
             _LOGGER.warning(f"Failed to set Battery Type to {option}")
             self._attr_available = False
             self.async_write_ha_state()