Bug fixed on load

custom_components/foxess_em/__init__.py

@@ -140,6 +140,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
     hass.data[DOMAIN][entry.entry_id]["unload"] = entry.add_update_listener(
         async_reload_entry
     )
+
     return True
 
 

custom_components/foxess_em/average/average_controller.py

@@ -29,9 +29,6 @@ def __init__(
         house_power: str,
         aux_power: list[str],
     ) -> None:
-        UnloadController.__init__(self)
-        CallbackController.__init__(self)
-        HassLoadController.__init__(self, hass, self.async_refresh)
         self._hass = hass
         self._last_update = None
 
@@ -47,6 +44,11 @@ def __init__(
 
         self._model = AverageModel(hass, entities, eco_start_time, eco_end_time)
 
+        # Setup mixins
+        UnloadController.__init__(self)
+        CallbackController.__init__(self)
+        HassLoadController.__init__(self, hass, self.async_refresh)
+
         # Refresh every hour on the half hour
         midnight_refresh = async_track_utc_time_change(
             self._hass,

custom_components/foxess_em/battery/battery_controller.py

@@ -37,9 +37,6 @@ def __init__(
         schedule: Schedule,
         peak_utils: PeakPeriodUtils,
     ) -> None:
-        UnloadController.__init__(self)
-        CallbackController.__init__(self)
-        HassLoadController.__init__(self, hass, self.async_refresh)
         self._hass = hass
         self._schedule = schedule
         self._peak_utils = peak_utils
@@ -60,6 +57,11 @@ def __init__(
         self._average_controller = average_controller
         self._last_update = None
 
+        # Setup mixins
+        UnloadController.__init__(self)
+        CallbackController.__init__(self)
+        HassLoadController.__init__(self, hass, self.async_refresh)
+
         # Refresh on SoC change
         battery_refresh = async_track_state_change(
             self._hass, battery_soc, self.refresh

custom_components/foxess_em/battery/schedule.py

@@ -18,11 +18,13 @@ class Schedule(HassLoadController, UnloadController):
 
     def __init__(self, hass: HomeAssistant) -> None:
         """Get persisted schedule from states"""
-        UnloadController.__init__(self)
-        HassLoadController.__init__(self, hass, self.load)
         self._hass = hass
         self._schedule = {}
 
+        # Setup mixins
+        UnloadController.__init__(self)
+        HassLoadController.__init__(self, hass, self.load)
+
         # Housekeeping on schedule
         housekeeping = async_track_utc_time_change(
             self._hass,
@@ -34,7 +36,7 @@ def __init__(self, hass: HomeAssistant) -> None:
         )
         self._unload_listeners.append(housekeeping)
 
-    def load(self, *args) -> None:
+    async def load(self, *args) -> None:
         """Load schedule from state"""
         schedule = self._hass.states.get(_SCHEDULE)
 

custom_components/foxess_em/const.py

@@ -20,7 +20,7 @@
 # Configuration and options
 SOLCAST_API_KEY = "key"
 SOLCAST_SCAN_INTERVAL = "scan_interval"
-# SOLCAST_URL = "https://238beaac-50d9-4aa4-89eb-24f49d786205.mock.pstmn.io"
+# SOLCAST_URL = "https://06db0776-e926-42bb-a6a2-85f88da8b0c8.mock.pstmn.io"
 SOLCAST_URL = "https://api.solcast.com.au"
 FOX_USERNAME = "fox_username"
 FOX_PASSWORD = "fox_password"

custom_components/foxess_em/forecast/forecast_controller.py

@@ -29,14 +29,16 @@ class ForecastController(UnloadController, CallbackController, HassLoadControlle
     """Class to manage forecast retrieval"""
 
     def __init__(self, hass: HomeAssistant, api: SolcastApiClient) -> None:
-        UnloadController.__init__(self)
-        CallbackController.__init__(self)
-        HassLoadController.__init__(self, hass, self.async_refresh)
         self._hass = hass
         self._api = ForecastModel(api)
         self._api_count = 0
         self._last_update = None
 
+        # Setup mixins
+        UnloadController.__init__(self)
+        CallbackController.__init__(self)
+        HassLoadController.__init__(self, hass, self.async_refresh)
+
         async_call_later(hass, 5, self.setup_refresh)
 
     async def setup_refresh(self, *args):