Add alias to DOMAIN import (part 1) (#125560)

* Add alias to DOMAIN import (part 1)

* Revert tomato/xiaomi
This commit is contained in:
epenet 2024-09-09 22:05:27 +02:00 committed by GitHub
parent e750f8f457
commit ce4a62574a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
22 changed files with 53 additions and 45 deletions

View file

@ -18,7 +18,11 @@ from aiounifi.models.client import Client
from aiounifi.models.device import Device
from aiounifi.models.event import Event, EventKey
from homeassistant.components.device_tracker import DOMAIN, ScannerEntity, SourceType
from homeassistant.components.device_tracker import (
DOMAIN as DEVICE_TRACKER_DOMAIN,
ScannerEntity,
SourceType,
)
from homeassistant.core import Event as core_Event, HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity_platform import AddEntitiesCallback
@ -198,11 +202,15 @@ def async_update_unique_id(hass: HomeAssistant, config_entry: UnifiConfigEntry)
def update_unique_id(obj_id: str) -> None:
"""Rework unique ID."""
new_unique_id = f"{hub.site}-{obj_id}"
if ent_reg.async_get_entity_id(DOMAIN, UNIFI_DOMAIN, new_unique_id):
if ent_reg.async_get_entity_id(
DEVICE_TRACKER_DOMAIN, UNIFI_DOMAIN, new_unique_id
):
return
unique_id = f"{obj_id}-{hub.site}"
if entity_id := ent_reg.async_get_entity_id(DOMAIN, UNIFI_DOMAIN, unique_id):
if entity_id := ent_reg.async_get_entity_id(
DEVICE_TRACKER_DOMAIN, UNIFI_DOMAIN, unique_id
):
ent_reg.async_update_entity(entity_id, new_unique_id=new_unique_id)
for obj_id in list(hub.api.clients) + list(hub.api.clients_all):