Improve renson typing (#94390)

This commit is contained in:
Marc Mueller 2023-06-11 02:28:32 +02:00 committed by GitHub
parent b45659eb84
commit 3b08d5f0c3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 8 additions and 10 deletions

View File

@ -4,6 +4,7 @@ from __future__ import annotations
from dataclasses import dataclass
from datetime import timedelta
import logging
from typing import Any
import async_timeout
from renson_endura_delta.renson import RensonVentilation
@ -81,7 +82,7 @@ class RensonCoordinator(DataUpdateCoordinator):
)
self.api = api
async def _async_update_data(self):
async def _async_update_data(self) -> dict[str, Any]:
"""Fetch data from API endpoint."""
async with async_timeout.timeout(30):
return await self.hass.async_add_executor_job(self.api.get_all_data)

View File

@ -16,7 +16,7 @@ from . import RensonCoordinator
from .const import DOMAIN
class RensonEntity(CoordinatorEntity):
class RensonEntity(CoordinatorEntity[RensonCoordinator]):
"""Renson entity."""
def __init__(

View File

@ -46,7 +46,7 @@ from homeassistant.const import (
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.entity_platform import AddEntitiesCallback
from . import RensonCoordinator
from . import RensonCoordinator, RensonData
from .const import DOMAIN
from .entity import RensonEntity
@ -305,13 +305,10 @@ async def async_setup_entry(
) -> None:
"""Set up the Renson sensor platform."""
api: RensonVentilation = hass.data[DOMAIN][config_entry.entry_id].api
coordinator: RensonCoordinator = hass.data[DOMAIN][
config_entry.entry_id
].coordinator
data: RensonData = hass.data[DOMAIN][config_entry.entry_id]
entities: list = []
for description in SENSORS:
entities.append(RensonSensor(description, api, coordinator))
entities = [
RensonSensor(description, data.api, data.coordinator) for description in SENSORS
]
async_add_entities(entities)