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

Add typing for EVENT_COMPONENT_LOADED (#110812)

Browse Source
This commit is contained in:
J. Nick Koston 2024-02-17 16:03:52 -06:00 committed by GitHub
parent bf1b5252c0
commit 0d6f4058ca
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 15 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

25
homeassistant/setup.py
View File

@ -8,7 +8,7 @@ from datetime import timedelta
import logging.handlers import logging.handlers
from timeit import default_timer as timer from timeit import default_timer as timer
from types import ModuleType from types import ModuleType
from typing import Any from typing import Any, Final, TypedDict
from . import config as conf_util, core, loader, requirements from . import config as conf_util, core, loader, requirements
from .const import ( from .const import (
@ -20,12 +20,12 @@ from .const import (
from .core import CALLBACK_TYPE, DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, callback from .core import CALLBACK_TYPE, DOMAIN as HOMEASSISTANT_DOMAIN, HomeAssistant, callback
from .exceptions import DependencyError, HomeAssistantError from .exceptions import DependencyError, HomeAssistantError
from .helpers.issue_registry import IssueSeverity, async_create_issue from .helpers.issue_registry import IssueSeverity, async_create_issue
from .helpers.typing import ConfigType from .helpers.typing import ConfigType, EventType
from .util import dt as dt_util, ensure_unique_string from .util import dt as dt_util, ensure_unique_string
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
ATTR_COMPONENT = "component" ATTR_COMPONENT: Final = "component"
BASE_PLATFORMS = {platform.value for platform in Platform} BASE_PLATFORMS = {platform.value for platform in Platform}
@ -66,6 +66,12 @@ SLOW_SETUP_WARNING = 10
SLOW_SETUP_MAX_WAIT = 300 SLOW_SETUP_MAX_WAIT = 300
class EventComponentLoaded(TypedDict):
"""EventComponentLoaded data."""
component: str
@callback @callback
def async_notify_setup_error( def async_notify_setup_error(
hass: HomeAssistant, component: str, display_link: str | None = None hass: HomeAssistant, component: str, display_link: str | None = None
@ -516,28 +522,27 @@ def _async_when_setup(
listeners: list[CALLBACK_TYPE] = [] listeners: list[CALLBACK_TYPE] = []
async def _matched_event(event: core.Event) -> None: async def _matched_event(event: EventType[EventComponentLoaded]) -> None:
"""Call the callback when we matched an event.""" """Call the callback when we matched an event."""
for listener in listeners: for listener in listeners:
listener() listener()
await when_setup() await when_setup()
@callback @callback
def _async_is_component_filter(event: core.Event) -> bool: def _async_is_component_filter(event: EventType[EventComponentLoaded]) -> bool:
"""Check if the event is for the component.""" """Check if the event is for the component."""
event_comp: str = event.data[ATTR_COMPONENT] return event.data[ATTR_COMPONENT] == component
return event_comp == component
listeners.append( listeners.append(
hass.bus.async_listen( hass.bus.async_listen(
EVENT_COMPONENT_LOADED, EVENT_COMPONENT_LOADED,
_matched_event, _matched_event, # type: ignore[arg-type]
event_filter=_async_is_component_filter, event_filter=_async_is_component_filter, # type: ignore[arg-type]
) )
) )
if start_event: if start_event:
listeners.append( listeners.append(
hass.bus.async_listen(EVENT_HOMEASSISTANT_START, _matched_event) hass.bus.async_listen(EVENT_HOMEASSISTANT_START, _matched_event) # type: ignore[arg-type]
) )