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

Improve youless generic typing (#84739)

Browse Source
This commit is contained in:
Marc Mueller 2022-12-28 23:52:01 +01:00 committed by GitHub
parent 2512eb1e4c
commit 72671b93d2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 17 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
homeassistant/components/youless/__init__.py
View File

@ -27,7 +27,7 @@ async def async_setup_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
except URLError as exception:
raise ConfigEntryNotReady from exception
async def async_update_data():
async def async_update_data() -> YoulessAPI:
"""Fetch data from the API."""
await hass.async_add_executor_job(api.update)
return api

25
homeassistant/components/youless/sensor.py
View File

@ -1,6 +1,7 @@
"""The sensor entity for the Youless integration."""
from __future__ import annotations
from youless_api import YoulessAPI
from youless_api.youless_sensor import YoulessSensor
from homeassistant.components.sensor import (
@ -26,7 +27,7 @@ async def async_setup_entry(
hass: HomeAssistant, entry: ConfigEntry, async_add_entities: AddEntitiesCallback
) -> None:
"""Initialize the integration."""
coordinator = hass.data[DOMAIN][entry.entry_id]
coordinator: DataUpdateCoordinator[YoulessAPI] = hass.data[DOMAIN][entry.entry_id]
device = entry.data[CONF_DEVICE]
if (device := entry.data[CONF_DEVICE]) is None:
device = entry.entry_id
@ -50,12 +51,14 @@ async def async_setup_entry(
)
class YoulessBaseSensor(CoordinatorEntity, SensorEntity):
class YoulessBaseSensor(
CoordinatorEntity[DataUpdateCoordinator[YoulessAPI]], SensorEntity
):
"""The base sensor for Youless."""
def __init__(
self,
coordinator: DataUpdateCoordinator,
coordinator: DataUpdateCoordinator[YoulessAPI],
device: str,
device_group: str,
friendly_name: str,
@ -97,7 +100,9 @@ class GasSensor(YoulessBaseSensor):
_attr_device_class = SensorDeviceClass.GAS
_attr_state_class = SensorStateClass.TOTAL_INCREASING
def __init__(self, coordinator: DataUpdateCoordinator, device: str) -> None:
def __init__(
self, coordinator: DataUpdateCoordinator[YoulessAPI], device: str
) -> None:
"""Instantiate a gas sensor."""
super().__init__(coordinator, device, "gas", "Gas meter", "gas")
self._attr_name = "Gas usage"
@ -116,7 +121,9 @@ class CurrentPowerSensor(YoulessBaseSensor):
_attr_device_class = SensorDeviceClass.POWER
_attr_state_class = SensorStateClass.MEASUREMENT
def __init__(self, coordinator: DataUpdateCoordinator, device: str) -> None:
def __init__(
self, coordinator: DataUpdateCoordinator[YoulessAPI], device: str
) -> None:
"""Instantiate the usage meter."""
super().__init__(coordinator, device, "power", "Power usage", "usage")
self._device = device
@ -136,7 +143,7 @@ class DeliveryMeterSensor(YoulessBaseSensor):
_attr_state_class = SensorStateClass.TOTAL_INCREASING
def __init__(
self, coordinator: DataUpdateCoordinator, device: str, dev_type: str
self, coordinator: DataUpdateCoordinator[YoulessAPI], device: str, dev_type: str
) -> None:
"""Instantiate a delivery meter sensor."""
super().__init__(
@ -163,7 +170,7 @@ class EnergyMeterSensor(YoulessBaseSensor):
def __init__(
self,
coordinator: DataUpdateCoordinator,
coordinator: DataUpdateCoordinator[YoulessAPI],
device: str,
dev_type: str,
state_class: SensorStateClass,
@ -194,7 +201,7 @@ class ExtraMeterSensor(YoulessBaseSensor):
_attr_state_class = SensorStateClass.TOTAL_INCREASING
def __init__(
self, coordinator: DataUpdateCoordinator, device: str, dev_type: str
self, coordinator: DataUpdateCoordinator[YoulessAPI], device: str, dev_type: str
) -> None:
"""Instantiate an extra meter sensor."""
super().__init__(
@ -220,7 +227,7 @@ class ExtraMeterPowerSensor(YoulessBaseSensor):
_attr_state_class = SensorStateClass.MEASUREMENT
def __init__(
self, coordinator: DataUpdateCoordinator, device: str, dev_type: str
self, coordinator: DataUpdateCoordinator[YoulessAPI], device: str, dev_type: str
) -> None:
"""Instantiate an extra meter power sensor."""
super().__init__(