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

Improve gios generic typing (#89321)

Browse Source
This commit is contained in:
Marc Mueller 2023-03-08 08:39:15 +01:00 committed by GitHub
parent ff83b8adb8
commit 58280dc2ec
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
homeassistant/components/gios/__init__.py
View File

@ -2,13 +2,13 @@
from __future__ import annotations from __future__ import annotations
import logging import logging
from typing import Any, cast
from aiohttp import ClientSession from aiohttp import ClientSession
from aiohttp.client_exceptions import ClientConnectorError from aiohttp.client_exceptions import ClientConnectorError
from async_timeout import timeout from async_timeout import timeout
from gios import Gios from gios import Gios
from gios.exceptions import GiosError from gios.exceptions import GiosError
from gios.model import GiosSensors
from homeassistant.components.air_quality import DOMAIN as AIR_QUALITY_PLATFORM from homeassistant.components.air_quality import DOMAIN as AIR_QUALITY_PLATFORM
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
@ -74,7 +74,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
return unload_ok return unload_ok
class GiosDataUpdateCoordinator(DataUpdateCoordinator): class GiosDataUpdateCoordinator(DataUpdateCoordinator[GiosSensors]):
"""Define an object to hold GIOS data.""" """Define an object to hold GIOS data."""
def __init__( def __init__(
@ -85,10 +85,10 @@ class GiosDataUpdateCoordinator(DataUpdateCoordinator):
super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL) super().__init__(hass, _LOGGER, name=DOMAIN, update_interval=SCAN_INTERVAL)
async def _async_update_data(self) -> dict[str, Any]: async def _async_update_data(self) -> GiosSensors:
"""Update data via library.""" """Update data via library."""
try: try:
async with timeout(API_TIMEOUT): async with timeout(API_TIMEOUT):
return cast(dict[str, Any], await self.gios.async_update()) return await self.gios.async_update()
except (GiosError, ClientConnectorError) as error: except (GiosError, ClientConnectorError) as error:
raise UpdateFailed(error) from error raise UpdateFailed(error) from error