jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-24 21:57:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Cleanup deprecated async_get_registry in rfxtrx (#72073)

Browse Source
This commit is contained in:
epenet 2022-05-18 12:00:01 +02:00 committed by GitHub
parent fc84e4061f
commit 8b5803735f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 10 additions and 16 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
homeassistant/components/rfxtrx/config_flow.py
View File

@ -24,16 +24,10 @@ from homeassistant.const import (
CONF_TYPE, CONF_TYPE,
) )
from homeassistant.core import callback from homeassistant.core import callback
import homeassistant.helpers.config_validation as cv from homeassistant.helpers import (
from homeassistant.helpers.device_registry import ( config_validation as cv,
DeviceEntry, device_registry as dr,
DeviceRegistry, entity_registry as er,
async_entries_for_config_entry,
async_get_registry as async_get_device_registry,
)
from homeassistant.helpers.entity_registry import (
async_entries_for_device,
async_get_registry as async_get_entity_registry,
) )
from . import ( from . import (
@ -80,8 +74,8 @@ def none_or_int(value, base):
class OptionsFlow(config_entries.OptionsFlow): class OptionsFlow(config_entries.OptionsFlow):
"""Handle Rfxtrx options.""" """Handle Rfxtrx options."""
_device_registry: DeviceRegistry _device_registry: dr.DeviceRegistry
_device_entries: list[DeviceEntry] _device_entries: list[dr.DeviceEntry]
def __init__(self, config_entry: ConfigEntry) -> None: def __init__(self, config_entry: ConfigEntry) -> None:
"""Initialize rfxtrx options flow.""" """Initialize rfxtrx options flow."""
@ -135,8 +129,8 @@ class OptionsFlow(config_entries.OptionsFlow):
return self.async_create_entry(title="", data={}) return self.async_create_entry(title="", data={})
device_registry = await async_get_device_registry(self.hass) device_registry = dr.async_get(self.hass)
device_entries = async_entries_for_config_entry( device_entries = dr.async_entries_for_config_entry(
device_registry, self._config_entry.entry_id device_registry, self._config_entry.entry_id
) )
self._device_registry = device_registry self._device_registry = device_registry
@ -320,8 +314,8 @@ class OptionsFlow(config_entries.OptionsFlow):
old_device_id = "_".join(x for x in old_device_data[CONF_DEVICE_ID]) old_device_id = "_".join(x for x in old_device_data[CONF_DEVICE_ID])
new_device_id = "_".join(x for x in new_device_data[CONF_DEVICE_ID]) new_device_id = "_".join(x for x in new_device_data[CONF_DEVICE_ID])
entity_registry = await async_get_entity_registry(self.hass) entity_registry = er.async_get(self.hass)
entity_entries = async_entries_for_device( entity_entries = er.async_entries_for_device(
entity_registry, old_device, include_disabled_entities=True entity_registry, old_device, include_disabled_entities=True
) )
entity_migration_map = {} entity_migration_map = {}