41 lines
1.2 KiB
Python
41 lines
1.2 KiB
Python
"""Helper methods for components within Home Assistant."""
|
|
import re
|
|
from typing import Any, Dict, Iterable, Sequence, Tuple
|
|
|
|
from homeassistant.const import CONF_PLATFORM
|
|
|
|
ConfigType = Dict[str, Any]
|
|
|
|
|
|
def config_per_platform(config: ConfigType, domain: str) -> Iterable[Tuple[Any, Any]]:
|
|
"""Break a component config into different platforms.
|
|
|
|
For example, will find 'switch', 'switch 2', 'switch 3', .. etc
|
|
Async friendly.
|
|
"""
|
|
for config_key in extract_domain_configs(config, domain):
|
|
platform_config = config[config_key]
|
|
|
|
if not platform_config:
|
|
continue
|
|
|
|
if not isinstance(platform_config, list):
|
|
platform_config = [platform_config]
|
|
|
|
for item in platform_config:
|
|
try:
|
|
platform = item.get(CONF_PLATFORM)
|
|
except AttributeError:
|
|
platform = None
|
|
|
|
yield platform, item
|
|
|
|
|
|
def extract_domain_configs(config: ConfigType, domain: str) -> Sequence[str]:
|
|
"""Extract keys from config for given domain name.
|
|
|
|
Async friendly.
|
|
"""
|
|
pattern = re.compile(fr"^{domain}(| .+)$")
|
|
return [key for key in config.keys() if pattern.match(key)]
|