From 209278d423b3ee8cccf0bd83a40bb0ad732de08c Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Wed, 22 Feb 2023 13:56:30 +0100 Subject: [PATCH] Improve type hint in homeassistant trigger (#88596) Imrpove type hint in homeassistant trigger --- homeassistant/components/homeassistant/trigger.py | 13 +++++++------ 1 file changed, 7 insertions(+), 6 deletions(-) diff --git a/homeassistant/components/homeassistant/trigger.py b/homeassistant/components/homeassistant/trigger.py index 588b6713007..3160af58079 100644 --- a/homeassistant/components/homeassistant/trigger.py +++ b/homeassistant/components/homeassistant/trigger.py @@ -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)