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

Improve oncue generic typing (#84761)

Browse source 
Improve oncue generic typing
This commit is contained in:
Marc Mueller 2022-12-29 12:26:17 +01:00 • committed by GitHub
parent 7768084655
commit d849dab7ba
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 7 deletions
Download patch file Download diff file Expand all files Collapse all files

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

@ -34,9 +34,11 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Set up sensors.""" """Set up sensors."""
coordinator: DataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id] coordinator: DataUpdateCoordinator[dict[str, OncueDevice]] = hass.data[DOMAIN][
config_entry.entry_id
]
entities: list[OncueBinarySensorEntity] = [] entities: list[OncueBinarySensorEntity] = []
devices: dict[str, OncueDevice] = coordinator.data devices = coordinator.data
for device_id, device in devices.items(): for device_id, device in devices.items():
entities.extend( entities.extend(
OncueBinarySensorEntity( OncueBinarySensorEntity(

6
homeassistant/components/oncue/entity.py
View file

@ -14,12 +14,14 @@ from homeassistant.helpers.update_coordinator import (
from .const import CONNECTION_ESTABLISHED_KEY, DOMAIN, VALUE_UNAVAILABLE from .const import CONNECTION_ESTABLISHED_KEY, DOMAIN, VALUE_UNAVAILABLE
class OncueEntity(CoordinatorEntity, Entity): class OncueEntity(
CoordinatorEntity[DataUpdateCoordinator[dict[str, OncueDevice]]], Entity
):
"""Representation of an Oncue entity.""" """Representation of an Oncue entity."""
def __init__( def __init__(
self, self,
coordinator: DataUpdateCoordinator, coordinator: DataUpdateCoordinator[dict[str, OncueDevice]],
device_id: str, device_id: str,
device: OncueDevice, device: OncueDevice,
sensor: OncueSensor, sensor: OncueSensor,

8
homeassistant/components/oncue/sensor.py
View file

@ -183,9 +183,11 @@ async def async_setup_entry(
async_add_entities: AddEntitiesCallback, async_add_entities: AddEntitiesCallback,
) -> None: ) -> None:
"""Set up sensors.""" """Set up sensors."""
coordinator: DataUpdateCoordinator = hass.data[DOMAIN][config_entry.entry_id] coordinator: DataUpdateCoordinator[dict[str, OncueDevice]] = hass.data[DOMAIN][
config_entry.entry_id
]
entities: list[OncueSensorEntity] = [] entities: list[OncueSensorEntity] = []
devices: dict[str, OncueDevice] = coordinator.data devices = coordinator.data
for device_id, device in devices.items(): for device_id, device in devices.items():
entities.extend( entities.extend(
OncueSensorEntity(coordinator, device_id, device, sensor, SENSOR_MAP[key]) OncueSensorEntity(coordinator, device_id, device, sensor, SENSOR_MAP[key])
@ -201,7 +203,7 @@ class OncueSensorEntity(OncueEntity, SensorEntity):
def __init__( def __init__(
self, self,
coordinator: DataUpdateCoordinator, coordinator: DataUpdateCoordinator[dict[str, OncueDevice]],
device_id: str, device_id: str,
device: OncueDevice, device: OncueDevice,
sensor: OncueSensor, sensor: OncueSensor,