From 05353f8e13ac7aeabbe7489c86c416c82d9a1d67 Mon Sep 17 00:00:00 2001
From: Kevin Hellemun <17928966+OGKevin@users.noreply.github.com>
Date: Thu, 28 Oct 2021 21:06:04 +0200
Subject: [PATCH] Add ROCKROBO_S5_MAX to xiaomi_miio vacuum models (#58591)

* Add ROCKROBO_S5_MAX to xiaomi_miio vacuum models.

https://github.com/home-assistant/core/issues/58550

Signed-off-by: Kevin Hellemun <17928966+OGKevin@users.noreply.github.com>

* disable pylint for todo

Signed-off-by: Kevin Hellemun <17928966+OGKevin@users.noreply.github.com>

* Minor refactor

Signed-off-by: Kevin Hellemun <17928966+OGKevin@users.noreply.github.com>
---
 homeassistant/components/xiaomi_miio/const.py | 21 +++++++++++++++++--
 1 file changed, 19 insertions(+), 2 deletions(-)

diff --git a/homeassistant/components/xiaomi_miio/const.py b/homeassistant/components/xiaomi_miio/const.py
index e27fb4d2110..a1f2414b713 100644
--- a/homeassistant/components/xiaomi_miio/const.py
+++ b/homeassistant/components/xiaomi_miio/const.py
@@ -195,8 +195,25 @@ MODELS_LIGHT = (
     + MODELS_LIGHT_BULB
     + MODELS_LIGHT_MONO
 )
-MODELS_VACUUM = [ROCKROBO_V1, ROCKROBO_S5, ROCKROBO_S6, ROCKROBO_S6_MAXV, ROCKROBO_S7]
-MODELS_VACUUM_WITH_MOP = [ROCKROBO_S5, ROCKROBO_S6, ROCKROBO_S6_MAXV, ROCKROBO_S7]
+
+# TODO: use const from pythonmiio once new release with the constant has been published. # pylint: disable=fixme
+ROCKROBO_S5_MAX = "roborock.vacuum.s5e"
+MODELS_VACUUM = [
+    ROCKROBO_V1,
+    ROCKROBO_S5,
+    ROCKROBO_S5_MAX,
+    ROCKROBO_S6,
+    ROCKROBO_S6_MAXV,
+    ROCKROBO_S7,
+]
+MODELS_VACUUM_WITH_MOP = [
+    ROCKROBO_S5,
+    ROCKROBO_S5_MAX,
+    ROCKROBO_S6,
+    ROCKROBO_S6_MAXV,
+    ROCKROBO_S7,
+]
+
 MODELS_AIR_MONITOR = [
     MODEL_AIRQUALITYMONITOR_V1,
     MODEL_AIRQUALITYMONITOR_B1,