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

Use NamedTuple for device_automation details (#55697)

Browse Source
This commit is contained in:
Marc Mueller 2021-09-04 13:42:36 +02:00 committed by GitHub
parent d8b85b2067
commit 0700108278
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 20 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
homeassistant/components/device_automation/__init__.py
View File

@ -6,7 +6,7 @@ from collections.abc import Iterable, Mapping
from functools import wraps from functools import wraps
import logging import logging
from types import ModuleType from types import ModuleType
from typing import Any from typing import Any, NamedTuple
import voluptuous as vol import voluptuous as vol
import voluptuous_serialize import voluptuous_serialize
@ -36,19 +36,31 @@ DEVICE_TRIGGER_BASE_SCHEMA = cv.TRIGGER_BASE_SCHEMA.extend(
} }
) )
class DeviceAutomationDetails(NamedTuple):
"""Details for device automation."""
section: str
get_automations_func: str
get_capabilities_func: str
TYPES = { TYPES = {
# platform name, get automations function, get capabilities function "trigger": DeviceAutomationDetails(
"trigger": (
"device_trigger", "device_trigger",
"async_get_triggers", "async_get_triggers",
"async_get_trigger_capabilities", "async_get_trigger_capabilities",
), ),
"condition": ( "condition": DeviceAutomationDetails(
"device_condition", "device_condition",
"async_get_conditions", "async_get_conditions",
"async_get_condition_capabilities", "async_get_condition_capabilities",
), ),
"action": ("device_action", "async_get_actions", "async_get_action_capabilities"), "action": DeviceAutomationDetails(
"device_action",
"async_get_actions",
"async_get_action_capabilities",
),
} }
@ -92,7 +104,7 @@ async def async_get_device_automation_platform(
Throws InvalidDeviceAutomationConfig if the integration is not found or does not support device automation. Throws InvalidDeviceAutomationConfig if the integration is not found or does not support device automation.
""" """
platform_name = TYPES[automation_type][0] platform_name = TYPES[automation_type].section
try: try:
integration = await async_get_integration_with_requirements(hass, domain) integration = await async_get_integration_with_requirements(hass, domain)
platform = integration.get_platform(platform_name) platform = integration.get_platform(platform_name)
@ -119,7 +131,7 @@ async def _async_get_device_automations_from_domain(
except InvalidDeviceAutomationConfig: except InvalidDeviceAutomationConfig:
return {} return {}
function_name = TYPES[automation_type][1] function_name = TYPES[automation_type].get_automations_func
return await asyncio.gather( return await asyncio.gather(
*( *(
@ -196,7 +208,7 @@ async def _async_get_device_automation_capabilities(hass, automation_type, autom
except InvalidDeviceAutomationConfig: except InvalidDeviceAutomationConfig:
return {} return {}
function_name = TYPES[automation_type][2] function_name = TYPES[automation_type].get_capabilities_func
if not hasattr(platform, function_name): if not hasattr(platform, function_name):
# The device automation has no capabilities # The device automation has no capabilities