From b09499826431a9b6720ab477039e1293a819d89b Mon Sep 17 00:00:00 2001
From: chris4824 <chris482422@gmail.com>
Date: Wed, 25 Dec 2024 11:20:27 -0500
Subject: [PATCH] add heater model WH517S (#237)

---
 README.md                               |  1 +
 custom_components/dreo/pydreo/models.py | 11 +++++++++++
 2 files changed, 12 insertions(+)

diff --git a/README.md b/README.md
index 85304e9..e12f982 100644
--- a/README.md
+++ b/README.md
@@ -82,6 +82,7 @@ Models that have been specifically tested can be found below.
 - DR-HSH0017S
 - WH719S
 - WH739S
+- WH517S
 
 ### Air Conditioners
 
diff --git a/custom_components/dreo/pydreo/models.py b/custom_components/dreo/pydreo/models.py
index 68cadca..85759c0 100644
--- a/custom_components/dreo/pydreo/models.py
+++ b/custom_components/dreo/pydreo/models.py
@@ -127,6 +127,17 @@ def __init__(
     "DR-HAP003S": DreoDeviceDetails(device_type=DreoDeviceType.AIR_PURIFIER),
 
     # Heaters
+    "DR-HSH017BS": DreoDeviceDetails(
+        device_type=DreoDeviceType.HEATER,
+        preset_modes=["H1", "H2", "H3"],
+        device_ranges={HEAT_RANGE: (1, 3), ECOLEVEL_RANGE: (41, 85)},
+        hvac_modes=[
+            HEATER_MODE_COOLAIR,
+            HEATER_MODE_HOTAIR,
+            HEATER_MODE_ECO,
+            HEATER_MODE_OFF,
+        ],
+    ),
     "DR-HSH004S": DreoDeviceDetails(
         device_type=DreoDeviceType.HEATER,
         preset_modes=["H1", "H2", "H3"],