mirror of
https://github.com/home-assistant/core.git
synced 2025-07-16 09:47:13 +00:00
Improve energy platform typing (#117003)
This commit is contained in:
parent
2db64c7e6d
commit
b21632ad05
@ -8,7 +8,6 @@ from collections.abc import Awaitable, Callable
|
||||
from datetime import timedelta
|
||||
import functools
|
||||
from itertools import chain
|
||||
from types import ModuleType
|
||||
from typing import Any, cast
|
||||
|
||||
import voluptuous as vol
|
||||
@ -64,13 +63,15 @@ async def async_get_energy_platforms(
|
||||
|
||||
@callback
|
||||
def _process_energy_platform(
|
||||
hass: HomeAssistant, domain: str, platform: ModuleType
|
||||
hass: HomeAssistant,
|
||||
domain: str,
|
||||
platform: EnergyPlatform,
|
||||
) -> None:
|
||||
"""Process energy platforms."""
|
||||
if not hasattr(platform, "async_get_solar_forecast"):
|
||||
return
|
||||
|
||||
platforms[domain] = cast(EnergyPlatform, platform).async_get_solar_forecast
|
||||
platforms[domain] = platform.async_get_solar_forecast
|
||||
|
||||
await async_process_integration_platforms(
|
||||
hass, DOMAIN, _process_energy_platform, wait_for_platforms=True
|
||||
|
Loading…
x
Reference in New Issue
Block a user