From c9e14feb3bb19aad4c4bdcf338f1f40bd6e12ec8 Mon Sep 17 00:00:00 2001
From: Alone <hi@anlo.ng>
Date: Sun, 28 Apr 2024 15:47:28 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=9B=80=20add=20sub-entities=20for=20xiaom?=
 =?UTF-8?q?i.waterheater.yms2=20(#1457)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 custom_components/xiaomi_miot/core/device_customizes.py | 8 ++++++++
 1 file changed, 8 insertions(+)

diff --git a/custom_components/xiaomi_miot/core/device_customizes.py b/custom_components/xiaomi_miot/core/device_customizes.py
index 3d9153b95..ff90a5843 100644
--- a/custom_components/xiaomi_miot/core/device_customizes.py
+++ b/custom_components/xiaomi_miot/core/device_customizes.py
@@ -1412,6 +1412,14 @@
     'xiaomi.watch.*': {
         'sensor_properties': 'current_step_count,current_distance',
     },
+    'xiaomi.waterheater.yms2': {
+        'sensor_properties': 'water_heater.status,water_level,temperature_tips,'
+                             'input_water_temp,water_pump_volt,remaining_time',
+        'switch_properties': 'on,hot_water_recirculation,heat_preservation,jog_function,pressure_boost.status,'
+                             'cloud_preheater,in_cloud_timer,reminder,enhanced',
+        'select_properties': 'light_off_time,effective,model',
+        'number_properties': 'target_temperature,boost_value',
+    },
     'xiaomi.wifispeaker.*': {
         'switch_properties': 'sleep_mode,no_disturb',
         'button_actions': 'wake_up,play_music,tv_switchon,stop_alarm',