From 15caf2ac03a126a4f00fe97d44aa3ce997708176 Mon Sep 17 00:00:00 2001
From: Mike <7278201+mike391@users.noreply.github.com>
Date: Thu, 21 Sep 2023 04:53:18 -0400
Subject: [PATCH] Add support for Levoit Vital200s purifier (#100613)

---
 homeassistant/components/vesync/const.py | 7 ++++++-
 homeassistant/components/vesync/fan.py   | 4 ++++
 2 files changed, 10 insertions(+), 1 deletion(-)

diff --git a/homeassistant/components/vesync/const.py b/homeassistant/components/vesync/const.py
index b20a04b8a1c..f87f1cf3a8a 100644
--- a/homeassistant/components/vesync/const.py
+++ b/homeassistant/components/vesync/const.py
@@ -35,5 +35,10 @@ SKU_TO_BASE_DEVICE = {
     "Core600S": "Core600S",
     "LAP-C601S-WUS": "Core600S",  # Alt ID Model Core600S
     "LAP-C601S-WUSR": "Core600S",  # Alt ID Model Core600S
-    "LAP-C601S-WEU": "Core600S",  # Alt ID Model Core600S
+    "LAP-C601S-WEU": "Core600S",  # Alt ID Model Core600S,
+    "LAP-V201S-AASR": "Vital200S",
+    "LAP-V201S-WJP": "Vital200S",
+    "LAP-V201S-WEU": "Vital200S",
+    "LAP-V201S-WUS": "Vital200S",
+    "LAP-V201-AUSR": "Vital200S",
 }
diff --git a/homeassistant/components/vesync/fan.py b/homeassistant/components/vesync/fan.py
index e5347b204e6..87934ced81f 100644
--- a/homeassistant/components/vesync/fan.py
+++ b/homeassistant/components/vesync/fan.py
@@ -27,10 +27,12 @@ DEV_TYPE_TO_HA = {
     "Core300S": "fan",
     "Core400S": "fan",
     "Core600S": "fan",
+    "Vital200S": "fan",
 }
 
 FAN_MODE_AUTO = "auto"
 FAN_MODE_SLEEP = "sleep"
+FAN_MODE_PET = "pet"
 
 PRESET_MODES = {
     "LV-PUR131S": [FAN_MODE_AUTO, FAN_MODE_SLEEP],
@@ -38,6 +40,7 @@ PRESET_MODES = {
     "Core300S": [FAN_MODE_AUTO, FAN_MODE_SLEEP],
     "Core400S": [FAN_MODE_AUTO, FAN_MODE_SLEEP],
     "Core600S": [FAN_MODE_AUTO, FAN_MODE_SLEEP],
+    "Vital200S": [FAN_MODE_AUTO, FAN_MODE_SLEEP, FAN_MODE_PET],
 }
 SPEED_RANGE = {  # off is not included
     "LV-PUR131S": (1, 3),
@@ -45,6 +48,7 @@ SPEED_RANGE = {  # off is not included
     "Core300S": (1, 3),
     "Core400S": (1, 4),
     "Core600S": (1, 4),
+    "Vital200S": (1, 4),
 }