d79f89e168
* Add support for Zigbee Home Automation * Fewer magic numbers * Make optional device config work * Remove non-zha device_tracker stuff * Always return boolean from is_on * Only pass through JSON serializable discovery_info * Update to bellows 0.2.4 * Fewer magic numbers in binary sensor * Populate const structures from a function * Update bellows to 0.2.6 * Fewer magic numbers in light * Take all possible clusters when overriding * Update bellows to 0.2.7
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""Constants related to the zha component."""
|
|
|
|
# Populated by populate_data() when zha component is initialized
|
|
DEVICE_CLASS = {}
|
|
SINGLE_CLUSTER_DEVICE_CLASS = {}
|
|
COMPONENT_CLUSTERS = {}
|
|
|
|
|
|
def populate_data():
|
|
"""Populate data using constants from bellows.
|
|
|
|
These cannot be module level, as importing bellows must be done in a
|
|
in a function.
|
|
"""
|
|
from bellows.zigbee import zcl
|
|
from bellows.zigbee.profiles import PROFILES, zha, zll
|
|
|
|
DEVICE_CLASS[zha.PROFILE_ID] = {
|
|
zha.DeviceType.ON_OFF_SWITCH: 'switch',
|
|
zha.DeviceType.SMART_PLUG: 'switch',
|
|
|
|
zha.DeviceType.ON_OFF_LIGHT: 'light',
|
|
zha.DeviceType.DIMMABLE_LIGHT: 'light',
|
|
zha.DeviceType.COLOR_DIMMABLE_LIGHT: 'light',
|
|
zha.DeviceType.ON_OFF_LIGHT_SWITCH: 'light',
|
|
zha.DeviceType.DIMMER_SWITCH: 'light',
|
|
zha.DeviceType.COLOR_DIMMER_SWITCH: 'light',
|
|
}
|
|
DEVICE_CLASS[zll.PROFILE_ID] = {
|
|
zll.DeviceType.ON_OFF_LIGHT: 'light',
|
|
zll.DeviceType.ON_OFF_PLUGIN_UNIT: 'switch',
|
|
zll.DeviceType.DIMMABLE_LIGHT: 'light',
|
|
zll.DeviceType.DIMMABLE_PLUGIN_UNIT: 'light',
|
|
zll.DeviceType.COLOR_LIGHT: 'light',
|
|
zll.DeviceType.EXTENDED_COLOR_LIGHT: 'light',
|
|
zll.DeviceType.COLOR_TEMPERATURE_LIGHT: 'light',
|
|
}
|
|
|
|
SINGLE_CLUSTER_DEVICE_CLASS.update({
|
|
zcl.clusters.general.OnOff: 'switch',
|
|
zcl.clusters.measurement.TemperatureMeasurement: 'sensor',
|
|
zcl.clusters.security.IasZone: 'binary_sensor',
|
|
})
|
|
|
|
# A map of hass components to all Zigbee clusters it could use
|
|
for profile_id, classes in DEVICE_CLASS.items():
|
|
profile = PROFILES[profile_id]
|
|
for device_type, component in classes.items():
|
|
if component not in COMPONENT_CLUSTERS:
|
|
COMPONENT_CLUSTERS[component] = set()
|
|
clusters = profile.CLUSTERS[device_type]
|
|
COMPONENT_CLUSTERS[component].update(clusters)
|