chickliu/ha_xiaomi_home
1
0
Fork 0
mirror of https://github.com/XiaoMi/ha_xiaomi_home.git synced 2026-01-15 05:50:43 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

feat:add support for state_class and unit process

Browse Source
This commit is contained in:
tiger 2024-12-17 19:25:13 +08:00
parent 9f9226ad29
commit 679af9dd62
1 changed files with 20 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
custom_components/xiaomi_home/miot/miot_device.py
View File

@ -515,9 +515,21 @@ class MIoTDevice:
prop.icon = self.icon_convert(prop.unit)
device_class = SPEC_PROP_TRANS_MAP['properties'][prop_name][
'device_class']
prop.platform = device_class
result = {'platform': platform, 'device_class': device_class}
# optional:
prop_optional = SPEC_PROP_TRANS_MAP['properties'][prop_name].get('optional')
if prop_optional:
prop_optional_state_class = prop_optional.get('state_class')
if prop_optional_state_class:
result['state_class'] = prop_optional_state_class
return {'platform': platform, 'device_class': device_class}
prop_optional_unit = prop_optional.get('unit_of_measurement')
if prop_optional_unit and not prop.unit:
result['unit_of_measurement'] = prop_optional_unit
return result
def spec_transform(self) -> None:
"""Parse service, property, event, action from device spec."""
@ -544,6 +556,12 @@ class MIoTDevice:
if prop_entity:
prop.platform = prop_entity['platform']
prop.device_class = prop_entity['device_class']
if 'state_class' in prop_entity:
prop.state_class = prop_entity['state_class']
if 'unit_of_measurement' in prop_entity:
prop.external_unit = self.unit_convert(prop_entity['unit_of_measurement'])
prop.icon = self.icon_convert(prop_entity['unit_of_measurement'])
# general conversion
if not prop.platform:
if prop.writable: