sytle: change logic to suit conversation

This commit is contained in:
DoraTiger 2024-12-26 20:12:05 +08:00
parent 2b2a670e2d
commit 07716ec9a8

View File

@ -517,15 +517,12 @@ class MIoTDevice:
'device_class'] 'device_class']
result = {'platform': platform, 'device_class': device_class} result = {'platform': platform, 'device_class': device_class}
# optional: # optional:
optional = SPEC_PROP_TRANS_MAP['properties'][prop_name].get('optional') if 'optional' in SPEC_PROP_TRANS_MAP['properties'][prop_name]:
if optional: optional = SPEC_PROP_TRANS_MAP['properties'][prop_name]['optional']
prop_optional_state_class = optional.get('state_class') if 'state_class' in optional:
if prop_optional_state_class: result['state_class'] = optional['state_class']
result['state_class'] = prop_optional_state_class if not prop.unit and 'unit_of_measurement' in optional:
result['unit_of_measurement'] = optional['unit_of_measurement']
prop_optional_unit = optional.get('unit_of_measurement')
if prop_optional_unit and not prop.unit:
result['unit_of_measurement'] = prop_optional_unit
return result return result
def spec_transform(self) -> None: def spec_transform(self) -> None: