From f4e4a647ef4ac011a38c42f5c91bc5ecde122dbb Mon Sep 17 00:00:00 2001
From: Luke Bonaccorsi <luke@lukeb.co.uk>
Date: Tue, 15 Aug 2023 12:29:27 +0100
Subject: [PATCH] Add LR20 and X9 Pro

---
 custom_components/robovac/robovac.py | 21 ++++++++++-----------
 1 file changed, 10 insertions(+), 11 deletions(-)

diff --git a/custom_components/robovac/robovac.py b/custom_components/robovac/robovac.py
index fb3b383..9b12139 100644
--- a/custom_components/robovac/robovac.py
+++ b/custom_components/robovac/robovac.py
@@ -18,9 +18,6 @@ class RoboVacEntityFeature(IntEnum):
     MAP = 512
     BOOST_IQ = 1024
 
-
-HAS_MAP_FEATURE = ["T2182","T2253","T2261","T2262"]
-
 ROBOVAC_SERIES = {
     "C": [
         "T2103",
@@ -41,10 +38,12 @@ ROBOVAC_SERIES = {
         "T2150",
         "T2255",
     ],
-	"L": ["T2182"],
-    "X": ["T2261", "T2262"],
+    "L": ["T2182", "T2192"],
+    "X": ["T2261", "T2262", "T2320"],
 }
 
+HAS_MAP_FEATURE = ["T2253", *ROBOVAC_SERIES["L"], *ROBOVAC_SERIES["X"]]
+
 ROBOVAC_SERIES_FEATURES = {
     "C": RoboVacEntityFeature.EDGE | RoboVacEntityFeature.SMALL_ROOM,
     "G": RoboVacEntityFeature.CLEANING_TIME
@@ -52,7 +51,7 @@ ROBOVAC_SERIES_FEATURES = {
     | RoboVacEntityFeature.DO_NOT_DISTURB
     | RoboVacEntityFeature.AUTO_RETURN
     | RoboVacEntityFeature.CONSUMABLES,
-	"L": RoboVacEntityFeature.CLEANING_TIME
+    "L": RoboVacEntityFeature.CLEANING_TIME
     | RoboVacEntityFeature.CLEANING_AREA
     | RoboVacEntityFeature.DO_NOT_DISTURB
     | RoboVacEntityFeature.AUTO_RETURN
@@ -73,10 +72,10 @@ ROBOVAC_SERIES_FEATURES = {
 }
 
 ROBOVAC_SERIES_FAN_SPEEDS = {
-"C": ["No Suction", "Standard", "Boost IQ", "Max"],
-"G": ["Standard", "Turbo", "Max", "Boost IQ"],
-"L": ["Pure", "Standard", "Turbo", "Max"],
-"X": ["Pure", "Standard", "Turbo", "Max"],
+    "C": ["No Suction", "Standard", "Boost IQ", "Max"],
+    "G": ["Standard", "Turbo", "Max", "Boost IQ"],
+    "L": ["Quiet", "Standard", "Turbo", "Max"],
+    "X": ["Pure", "Standard", "Turbo", "Max"],
 }
 
 
@@ -84,6 +83,7 @@ SUPPORTED_ROBOVAC_MODELS = list(
     set([item for sublist in ROBOVAC_SERIES.values() for item in sublist])
 )
 
+
 class ModelNotSupportedException(Exception):
     """This model is not supported"""
 
@@ -127,6 +127,5 @@ class RoboVac(TuyaDevice):
             if self.model_code in models:
                 return series
 
-
     def getFanSpeeds(self):
         return ROBOVAC_SERIES_FAN_SPEEDS[self.getRoboVacSeries()]