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

Improve energy platform typing (#117003)

Browse Source
This commit is contained in:
Marc Mueller 2024-05-07 18:28:42 +02:00 committed by GitHub
parent 2db64c7e6d
commit b21632ad05
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
homeassistant/components/energy/websocket_api.py
View File

@ -8,7 +8,6 @@ from collections.abc import Awaitable, Callable
from datetime import timedelta from datetime import timedelta
import functools import functools
from itertools import chain from itertools import chain
from types import ModuleType
from typing import Any, cast from typing import Any, cast
import voluptuous as vol import voluptuous as vol
@ -64,13 +63,15 @@ async def async_get_energy_platforms(
@callback @callback
def _process_energy_platform( def _process_energy_platform(
hass: HomeAssistant, domain: str, platform: ModuleType hass: HomeAssistant,
domain: str,
platform: EnergyPlatform,
) -> None: ) -> None:
"""Process energy platforms.""" """Process energy platforms."""
if not hasattr(platform, "async_get_solar_forecast"): if not hasattr(platform, "async_get_solar_forecast"):
return return
platforms[domain] = cast(EnergyPlatform, platform).async_get_solar_forecast platforms[domain] = platform.async_get_solar_forecast
await async_process_integration_platforms( await async_process_integration_platforms(
hass, DOMAIN, _process_energy_platform, wait_for_platforms=True hass, DOMAIN, _process_energy_platform, wait_for_platforms=True