Move get_custom_ids_by_entity_keys to device (#1518)

SukramJ · Apr 19, 2024 · 364a2b9 · 364a2b9
1 parent cf886f5
commit 364a2b9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 17 deletions.
diff --git a/hahomematic/central/__init__.py b/hahomematic/central/__init__.py
@@ -616,23 +616,6 @@ def get_readable_generic_entities(
             )
         )
 
-    def get_custom_ids_by_entity_keys(self, entity_keys: set[ENTITY_KEY]) -> set[str]:
-        """Return the custom ids by entity keys."""
-        return {
-            entity.custom_id
-            for entity in self.get_entities(exclude_no_create=False, registered=True)
-            if (
-                (
-                    (isinstance(entity, BaseParameterEntity) and entity.entity_key in entity_keys)
-                    or (
-                        isinstance(entity, CustomEntity)
-                        and entity.has_entity_key(entity_keys=entity_keys)
-                    )
-                )
-                and entity.custom_id is not None
-            )
-        }
-
     def _get_primary_client(self) -> hmcl.Client | None:
         """Return the client by interface_id or the first with a virtual remote."""
         client: hmcl.Client | None = None

diff --git a/hahomematic/platforms/device.py b/hahomematic/platforms/device.py
@@ -352,6 +352,23 @@ def get_sub_device_channel(self, channel_no: int) -> int | None:
         """Return the sub device channel."""
         return self._sub_device_channels.get(channel_no)
 
+    def get_custom_ids_by_entity_keys(self, entity_keys: set[ENTITY_KEY]) -> set[str]:
+        """Return the custom ids by entity keys."""
+        return {
+            entity.custom_id
+            for entity in self.get_entities(exclude_no_create=False, registered=True)
+            if (
+                (
+                    (isinstance(entity, BaseParameterEntity) and entity.entity_key in entity_keys)
+                    or (
+                        isinstance(entity, hmce.CustomEntity)
+                        and entity.has_entity_key(entity_keys=entity_keys)
+                    )
+                )
+                and entity.custom_id is not None
+            )
+        }
+
     def add_entity(self, entity: CallbackEntity) -> None:
         """Add a hm entity to a device."""
         if isinstance(entity, BaseParameterEntity):