mirror of
https://github.com/jdejaegh/irm-kmi-ha.git
synced 2025-06-27 11:39:26 +02:00
25 lines
1.1 KiB
Python
25 lines
1.1 KiB
Python
import logging
|
|
|
|
from homeassistant.config_entries import ConfigEntry
|
|
from homeassistant.core import HomeAssistant
|
|
from homeassistant.helpers import device_registry, entity_registry
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
|
|
def disable_from_config(hass: HomeAssistant, config_entry: ConfigEntry):
|
|
modify_from_config(hass, config_entry, False)
|
|
|
|
|
|
def enable_from_config(hass: HomeAssistant, config_entry: ConfigEntry):
|
|
modify_from_config(hass, config_entry, True)
|
|
|
|
|
|
def modify_from_config(hass: HomeAssistant, config_entry: ConfigEntry, enable: bool):
|
|
dr = device_registry.async_get(hass)
|
|
devices = device_registry.async_entries_for_config_entry(dr, config_entry.entry_id)
|
|
_LOGGER.info(f"Trying to {'enable' if enable else 'disable'} {config_entry.entry_id}: {len(devices)} device(s)")
|
|
for device in devices:
|
|
_LOGGER.info(f"Disabling device {device.name} because it is out of Benelux")
|
|
dr.async_update_device(device_id=device.id,
|
|
disabled_by=None if enable else device_registry.DeviceEntryDisabler.INTEGRATION)
|