evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Improve device_automation typing (#66621)

Browse source
This commit is contained in:
Marc Mueller 2022-02-17 22:08:43 +01:00 • committed by GitHub
parent 8bf19655f1
commit 1a247f7d1b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 148 additions and 19 deletions
Download patch file Download diff file Expand all files Collapse all files

16
homeassistant/helpers/condition.py
View file

@ -875,12 +875,7 @@ async def async_device_from_config(
platform = await async_get_device_automation_platform(
hass, config[CONF_DOMAIN], DeviceAutomationType.CONDITION
)
return trace_condition_function(
cast(
ConditionCheckerType,
platform.async_condition_from_config(hass, config),
)
)
return trace_condition_function(platform.async_condition_from_config(hass, config))
async def async_trigger_from_config(
@ -943,14 +938,15 @@ async def async_validate_condition_config(
hass, config[CONF_DOMAIN], DeviceAutomationType.CONDITION
)
if hasattr(platform, "async_validate_condition_config"):
return await platform.async_validate_condition_config(hass, config) # type: ignore
return await platform.async_validate_condition_config(hass, config)
return cast(ConfigType, platform.CONDITION_SCHEMA(config))
if condition in ("numeric_state", "state"):
validator = getattr(
sys.modules[__name__], VALIDATE_CONFIG_FORMAT.format(condition)
validator = cast(
Callable[[HomeAssistant, ConfigType], ConfigType],
getattr(sys.modules[__name__], VALIDATE_CONFIG_FORMAT.format(condition)),
)
return validator(hass, config) # type: ignore
return validator(hass, config)
return config