v1

al-one · Sep 24, 2024 · c589738 · c589738
1 parent 8a95118
commit c589738
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/custom_components/xiaomi_miot/core/device.py b/custom_components/xiaomi_miot/core/device.py
@@ -23,7 +23,7 @@
 from miio.miot_device import MiotDevice as MiotDeviceBase
 
 if TYPE_CHECKING:
-    from . import XEntity
+    from . import BasicEntity
 
 _LOGGER = logging.getLogger(__name__)
 
@@ -121,7 +121,7 @@ def __init__(self, info: DeviceInfo, entry: HassEntry):
         self.info = info
         self.hass = entry.hass
         self.entry = entry
-        self.entities: dict[str, 'XEntity'] = {}
+        self.entities: dict[str, 'BasicEntity'] = {}
         self.listeners: list[Callable] = []
         self.converters: list[BaseConv] = []
 
@@ -266,7 +266,7 @@ def init_converters(self):
                 for action in srv.get_actions(*als):
                     self.converters.append(MiotActionConv(action.full_name, d, action=action))
 
-    def add_entity(self, entity: 'XEntity'):
+    def add_entity(self, entity: 'BasicEntity'):
         if entity not in self.entities:
             self.entities[entity.unique_id] = entity
 

diff --git a/custom_components/xiaomi_miot/core/hass_entry.py b/custom_components/xiaomi_miot/core/hass_entry.py
@@ -70,7 +70,7 @@ def new_adder(self, domain, adder: AddEntitiesCallback):
                     if not cls:
                         continue
                     entity = cls(device, conv)
-                    device.entities[key] = entity
+                    device.add_entity(entity)
                     adder([entity], update_before_add=False)
                     _LOGGER.info('New entity: %s', entity)