mirror of
https://github.com/home-assistant/core.git
synced 2026-07-05 21:55:36 +01:00
Simone Chemelli
82 lines
2.8 KiB
Python
82 lines
2.8 KiB
Python
"""Utils for Alexa Devices."""
|
|
|
|
from aioamazondevices.const.devices import SPEAKER_GROUP_FAMILY
|
|
from aioamazondevices.const.schedules import (
|
|
NOTIFICATION_ALARM,
|
|
NOTIFICATION_REMINDER,
|
|
NOTIFICATION_TIMER,
|
|
)
|
|
|
|
from homeassistant.components.sensor import DOMAIN as SENSOR_DOMAIN
|
|
from homeassistant.components.switch import DOMAIN as SWITCH_DOMAIN
|
|
from homeassistant.core import HomeAssistant
|
|
import homeassistant.helpers.entity_registry as er
|
|
|
|
from .const import _LOGGER, DOMAIN
|
|
from .coordinator import AmazonDevicesCoordinator
|
|
|
|
|
|
async def async_update_unique_id(
|
|
hass: HomeAssistant,
|
|
coordinator: AmazonDevicesCoordinator,
|
|
platform: str,
|
|
old_key: str,
|
|
new_key: str,
|
|
) -> None:
|
|
"""Update unique id for entities created with old format."""
|
|
entity_registry = er.async_get(hass)
|
|
|
|
for serial_num in coordinator.data:
|
|
unique_id = f"{serial_num}-{old_key}"
|
|
if entity_id := entity_registry.async_get_entity_id(
|
|
DOMAIN, platform, unique_id
|
|
):
|
|
_LOGGER.debug("Updating unique_id for %s", entity_id)
|
|
new_unique_id = unique_id.replace(old_key, new_key)
|
|
|
|
# Update the registry with the new unique_id
|
|
entity_registry.async_update_entity(entity_id, new_unique_id=new_unique_id)
|
|
|
|
|
|
async def async_remove_dnd_from_virtual_group(
|
|
hass: HomeAssistant,
|
|
coordinator: AmazonDevicesCoordinator,
|
|
key: str,
|
|
) -> None:
|
|
"""Remove entity DND from virtual group."""
|
|
entity_registry = er.async_get(hass)
|
|
|
|
for serial_num in coordinator.data:
|
|
unique_id = f"{serial_num}-{key}"
|
|
entity_id = entity_registry.async_get_entity_id(
|
|
DOMAIN, SWITCH_DOMAIN, unique_id
|
|
)
|
|
is_group = coordinator.data[serial_num].device_family == SPEAKER_GROUP_FAMILY
|
|
if entity_id and is_group:
|
|
entity_registry.async_remove(entity_id)
|
|
_LOGGER.debug("Removed DND switch from virtual group %s", entity_id)
|
|
|
|
|
|
async def async_remove_unsupported_notification_sensors(
|
|
hass: HomeAssistant,
|
|
coordinator: AmazonDevicesCoordinator,
|
|
) -> None:
|
|
"""Remove notification sensors from unsupported devices."""
|
|
entity_registry = er.async_get(hass)
|
|
|
|
for serial_num in coordinator.data:
|
|
for notification_key in (
|
|
NOTIFICATION_ALARM,
|
|
NOTIFICATION_REMINDER,
|
|
NOTIFICATION_TIMER,
|
|
):
|
|
unique_id = f"{serial_num}-{notification_key}"
|
|
entity_id = entity_registry.async_get_entity_id(
|
|
DOMAIN, SENSOR_DOMAIN, unique_id=unique_id
|
|
)
|
|
is_unsupported = not coordinator.data[serial_num].notifications_supported
|
|
|
|
if entity_id and is_unsupported:
|
|
entity_registry.async_remove(entity_id)
|
|
_LOGGER.debug("Removed unsupported notification sensor %s", entity_id)
|