Improve type hint in homeassistant trigger (#88596)

Imrpove type hint in homeassistant trigger
This commit is contained in:
epenet 2023-02-22 13:56:30 +01:00 committed by GitHub
parent 10bb0ab6b4
commit 209278d423
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1,16 +1,17 @@
"""Home Assistant trigger dispatcher."""
import importlib
from homeassistant.components.device_automation.trigger import (
DeviceAutomationTriggerProtocol,
)
from homeassistant.const import CONF_PLATFORM
from homeassistant.core import CALLBACK_TYPE, HomeAssistant
from homeassistant.helpers.trigger import TriggerActionType, TriggerInfo
from homeassistant.helpers.trigger import (
TriggerActionType,
TriggerInfo,
TriggerProtocol,
)
from homeassistant.helpers.typing import ConfigType
def _get_trigger_platform(config: ConfigType) -> DeviceAutomationTriggerProtocol:
def _get_trigger_platform(config: ConfigType) -> TriggerProtocol:
return importlib.import_module(f"..triggers.{config[CONF_PLATFORM]}", __name__)
@ -20,7 +21,7 @@ async def async_validate_trigger_config(
"""Validate config."""
platform = _get_trigger_platform(config)
if hasattr(platform, "async_validate_trigger_config"):
return await getattr(platform, "async_validate_trigger_config")(hass, config)
return await platform.async_validate_trigger_config(hass, config)
return platform.TRIGGER_SCHEMA(config)