Merge pull request #96 from dougiteixeira/fix-unique_id-multiple-nodes

Fixes unique id for multiple nodes
dougiteixeira · Jun 16, 2023 · 8d531c9 · 8d531c9
2 parents f7683b4 + be82028
commit 8d531c9
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 10 deletions.
diff --git a/custom_components/proxmoxve/binary_sensor.py b/custom_components/proxmoxve/binary_sensor.py
@@ -85,7 +85,7 @@ async def async_setup_entry(
                             node=node,
                         ),
                         description=description,
-                        vm_id=None,
+                        resource_id=node,
                     )
                 )
 
@@ -107,7 +107,7 @@ async def async_setup_entry(
                             vm_id=vm_id,
                         ),
                         description=description,
-                        vm_id=vm_id,
+                        resource_id=vm_id,
                     )
                 )
 
@@ -129,7 +129,7 @@ async def async_setup_entry(
                             vm_id=container_id,
                         ),
                         description=description,
-                        vm_id=container_id,
+                        resource_id=container_id,
                     )
                 )
 
@@ -138,15 +138,15 @@ async def async_setup_entry(
 
 def create_binary_sensor(
     coordinator,
-    vm_id,
+    resource_id,
     config_entry,
     info_device,
     description,
 ) -> ProxmoxBinarySensorEntity:
     """Create a binary sensor based on the given data."""
     return ProxmoxBinarySensorEntity(
         coordinator=coordinator,
-        unique_id=f"{config_entry.entry_id}_{vm_id}_{description.key}",
+        unique_id=f"{config_entry.entry_id}_{resource_id}_{description.key}",
         description=description,
         info_device=info_device,
     )

diff --git a/custom_components/proxmoxve/sensor.py b/custom_components/proxmoxve/sensor.py
@@ -432,7 +432,7 @@ async def async_setup_entry(
                             node=node,
                         ),
                         description=description,
-                        vm_id=None,
+                        resource_id=node,
                         config_entry=config_entry,
                     )
                 )
@@ -454,7 +454,7 @@ async def async_setup_entry(
                             vm_id=vm_id,
                         ),
                         description=description,
-                        vm_id=vm_id,
+                        resource_id=vm_id,
                         config_entry=config_entry,
                     )
                 )
@@ -476,7 +476,7 @@ async def async_setup_entry(
                             vm_id=ct_id,
                         ),
                         description=description,
-                        vm_id=ct_id,
+                        resource_id=ct_id,
                         config_entry=config_entry,
                     )
                 )
@@ -489,13 +489,13 @@ def create_sensor(
     info_device: DeviceInfo,
     description: ProxmoxSensorEntityDescription,
     config_entry: ConfigEntry,
-    vm_id: str | None = None,
+    resource_id: str | None = None,
 ) -> ProxmoxSensorEntity:
     """Create a sensor based on the given data."""
     return ProxmoxSensorEntity(
         coordinator=coordinator,
         description=description,
-        unique_id=f"{config_entry.entry_id}_{vm_id}_{description.key}",
+        unique_id=f"{config_entry.entry_id}_{resource_id}_{description.key}",
         info_device=info_device,
     )