diff --git a/custom_components/xiaomi_home/miot/specs/specv2entity.py b/custom_components/xiaomi_home/miot/specs/specv2entity.py index 4a57867..30528f8 100644 --- a/custom_components/xiaomi_home/miot/specs/specv2entity.py +++ b/custom_components/xiaomi_home/miot/specs/specv2entity.py @@ -47,6 +47,7 @@ Conversion rules of MIoT-Spec-V2 instance to Home Assistant entity. """ from homeassistant.components.sensor import SensorDeviceClass from homeassistant.components.event import EventDeviceClass +from homeassistant.components.binary_sensor import BinarySensorDeviceClass # pylint: disable=pointless-string-statement """SPEC_DEVICE_TRANS_MAP @@ -391,6 +392,14 @@ SPEC_PROP_TRANS_MAP: dict[str, dict | str] = { 'device_class': SensorDeviceClass.DURATION, 'entity': 'sensor' }, + 'submersion-state': { + 'device_class': BinarySensorDeviceClass.MOISTURE, + 'entity': 'binary_sensor' + }, + 'contact-state': { + 'device_class': BinarySensorDeviceClass.DOOR, + 'entity': 'binary_sensor' + }, 'has-someone-duration': 'no-one-determine-time', 'no-one-duration': 'no-one-determine-time' }