evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Cleanup deprecated async_get_registry in xbox (#72079)

Browse source
This commit is contained in:
epenet 2022-05-18 12:37:27 +02:00 • committed by GitHub
parent 8b5803735f
commit 8492f282cb
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

6
homeassistant/components/xbox/binary_sensor.py
View file

@ -6,10 +6,8 @@ from functools import partial
from homeassistant.components.binary_sensor import BinarySensorEntity from homeassistant.components.binary_sensor import BinarySensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.entity_registry import (
async_get_registry as async_get_entity_registry,
)
from . import XboxUpdateCoordinator from . import XboxUpdateCoordinator
from .base_sensor import XboxBaseSensorEntity from .base_sensor import XboxBaseSensorEntity
@ -80,7 +78,7 @@ async def async_remove_entities(
current: dict[str, XboxBinarySensorEntity], current: dict[str, XboxBinarySensorEntity],
) -> None: ) -> None:
"""Remove friend sensors from Home Assistant.""" """Remove friend sensors from Home Assistant."""
registry = await async_get_entity_registry(coordinator.hass) registry = er.async_get(coordinator.hass)
entities = current[xuid] entities = current[xuid]
for entity in entities: for entity in entities:
if entity.entity_id in registry.entities: if entity.entity_id in registry.entities:

6
homeassistant/components/xbox/sensor.py
View file

@ -6,10 +6,8 @@ from functools import partial
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorEntity
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers import entity_registry as er
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.entity_registry import (
async_get_registry as async_get_entity_registry,
)
from . import XboxUpdateCoordinator from . import XboxUpdateCoordinator
from .base_sensor import XboxBaseSensorEntity from .base_sensor import XboxBaseSensorEntity
@ -82,7 +80,7 @@ async def async_remove_entities(
current: dict[str, XboxSensorEntity], current: dict[str, XboxSensorEntity],
) -> None: ) -> None:
"""Remove friend sensors from Home Assistant.""" """Remove friend sensors from Home Assistant."""
registry = await async_get_entity_registry(coordinator.hass) registry = er.async_get(coordinator.hass)
entities = current[xuid] entities = current[xuid]
for entity in entities: for entity in entities:
if entity.entity_id in registry.entities: if entity.entity_id in registry.entities: