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 dataclasses import dataclass
from datetime import timedelta from datetime import timedelta
import logging import logging
from typing import Any
import async_timeout import async_timeout
from renson_endura_delta.renson import RensonVentilation from renson_endura_delta.renson import RensonVentilation
@ -81,7 +82,7 @@ class RensonCoordinator(DataUpdateCoordinator):
) )
self.api = api self.api = api
async def _async_update_data(self): async def _async_update_data(self) -> dict[str, Any]:
"""Fetch data from API endpoint.""" """Fetch data from API endpoint."""
async with async_timeout.timeout(30): async with async_timeout.timeout(30):
return await self.hass.async_add_executor_job(self.api.get_all_data) 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 from .const import DOMAIN
class RensonEntity(CoordinatorEntity): class RensonEntity(CoordinatorEntity[RensonCoordinator]):
"""Renson entity.""" """Renson entity."""
def __init__( def __init__(

View File

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