fix: don't set the inverter obj directly to domain obj

mukaschultze · Jan 20, 2024 · ffd8cd0 · ffd8cd0
1 parent 4e81e0c
commit ffd8cd0
Show file tree

Hide file tree

Showing 7 changed files with 8 additions and 8 deletions.
diff --git a/custom_components/must_inverter/__init__.py b/custom_components/must_inverter/__init__.py
@@ -33,7 +33,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry):
         if not successReading:
             raise ConfigEntryNotReady(f"Unable to read from modbus device")
 
-        hass.data[DOMAIN] = inverter
+        hass.data[DOMAIN][entry.entry_id] = inverter
 
         for component in PLATFORMS:
             hass.async_create_task(
@@ -59,7 +59,7 @@ async def async_unload_entry(hass, entry):
     if not unload_ok:
         return False
 
-    hass.data[DOMAIN] = None
+    hass.data[DOMAIN][entry.entry_id] = None
     return True
 
 class MustInverter:

diff --git a/custom_components/must_inverter/binary_sensor.py b/custom_components/must_inverter/binary_sensor.py
@@ -11,7 +11,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, entry, async_add_entities):
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for sensor_info in SENSORS_ARRAY:

diff --git a/custom_components/must_inverter/button.py b/custom_components/must_inverter/button.py
@@ -10,7 +10,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, entry, async_add_entities):
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for sensor_info in SENSORS_ARRAY:

diff --git a/custom_components/must_inverter/number.py b/custom_components/must_inverter/number.py
@@ -11,7 +11,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, entry, async_add_entities):
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for sensor_info in SENSORS_ARRAY:

diff --git a/custom_components/must_inverter/select.py b/custom_components/must_inverter/select.py
@@ -11,7 +11,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, entry, async_add_entities):
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for sensor_info in SENSORS_ARRAY:

diff --git a/custom_components/must_inverter/sensor.py b/custom_components/must_inverter/sensor.py
@@ -11,7 +11,7 @@
 _LOGGER = logging.getLogger(__name__)
 
 async def async_setup_entry(hass, entry, async_add_entities):
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for sensor_info in SENSORS_ARRAY:

diff --git a/custom_components/must_inverter/switch.py b/custom_components/must_inverter/switch.py
@@ -29,7 +29,7 @@ async def async_setup_entry(hass, entry, async_add_entities):
         Setting(7, "RecordFault",      True,  True,  'mdi:alert'),
     ]
 
-    inverter = hass.data[DOMAIN]
+    inverter = hass.data[DOMAIN][entry.entry_id]
     entities = []
 
     for setting in settings: