2f474a0ed8
bellows 0.3.0 changes the API to have both, renaming the attribute which used to be for input clusters in the process. This is in preparation for remotes.
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
"""All constants related to the ZHA component."""
|
|
|
|
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(), set())
|
|
clusters = profile.CLUSTERS[device_type]
|
|
COMPONENT_CLUSTERS[component][0].update(clusters[0])
|
|
COMPONENT_CLUSTERS[component][1].update(clusters[1])
|