From 7f4c5c04d39885ea31d316ebbd9306dc0cf1d1a8 Mon Sep 17 00:00:00 2001
From: McYars <70546784+McYars@users.noreply.github.com>
Date: Wed, 24 Aug 2022 16:09:01 +0300
Subject: [PATCH] Add Xiaomi Smartmi Fresh Air System XFXTDFR02ZM (#76637)
---
homeassistant/components/xiaomi_miio/const.py | 12 ++++++++++++
homeassistant/components/xiaomi_miio/number.py | 3 +++
homeassistant/components/xiaomi_miio/select.py | 4 ++++
homeassistant/components/xiaomi_miio/sensor.py | 2 ++
homeassistant/components/xiaomi_miio/switch.py | 3 +++
5 files changed, 24 insertions(+)
diff --git a/homeassistant/components/xiaomi_miio/const.py b/homeassistant/components/xiaomi_miio/const.py
index 54289bb8389..c0711a02a36 100644
--- a/homeassistant/components/xiaomi_miio/const.py
+++ b/homeassistant/components/xiaomi_miio/const.py
@@ -78,6 +78,7 @@ MODEL_AIRHUMIDIFIER_MJJSQ = "deerma.humidifier.mjjsq"
MODEL_AIRFRESH_A1 = "dmaker.airfresh.a1"
MODEL_AIRFRESH_VA2 = "zhimi.airfresh.va2"
+MODEL_AIRFRESH_VA4 = "zhimi.airfresh.va4"
MODEL_AIRFRESH_T2017 = "dmaker.airfresh.t2017"
MODEL_FAN_1C = "dmaker.fan.1c"
@@ -136,6 +137,7 @@ MODELS_PURIFIER_MIIO = [
MODEL_AIRPURIFIER_2H,
MODEL_AIRFRESH_A1,
MODEL_AIRFRESH_VA2,
+ MODEL_AIRFRESH_VA4,
MODEL_AIRFRESH_T2017,
]
MODELS_HUMIDIFIER_MIIO = [
@@ -415,6 +417,16 @@ FEATURE_FLAGS_AIRFRESH = (
| FEATURE_SET_EXTRA_FEATURES
)
+FEATURE_FLAGS_AIRFRESH_VA4 = (
+ FEATURE_SET_BUZZER
+ | FEATURE_SET_CHILD_LOCK
+ | FEATURE_SET_LED
+ | FEATURE_SET_LED_BRIGHTNESS
+ | FEATURE_RESET_FILTER
+ | FEATURE_SET_EXTRA_FEATURES
+ | FEATURE_SET_PTC
+)
+
FEATURE_FLAGS_AIRFRESH_T2017 = (
FEATURE_SET_BUZZER | FEATURE_SET_CHILD_LOCK | FEATURE_SET_DISPLAY | FEATURE_SET_PTC
)
diff --git a/homeassistant/components/xiaomi_miio/number.py b/homeassistant/components/xiaomi_miio/number.py
index 2d3106ddd4e..1bc843463e8 100644
--- a/homeassistant/components/xiaomi_miio/number.py
+++ b/homeassistant/components/xiaomi_miio/number.py
@@ -23,6 +23,7 @@ from .const import (
FEATURE_FLAGS_AIRFRESH,
FEATURE_FLAGS_AIRFRESH_A1,
FEATURE_FLAGS_AIRFRESH_T2017,
+ FEATURE_FLAGS_AIRFRESH_VA4,
FEATURE_FLAGS_AIRHUMIDIFIER_CA4,
FEATURE_FLAGS_AIRHUMIDIFIER_CA_AND_CB,
FEATURE_FLAGS_AIRPURIFIER_2S,
@@ -54,6 +55,7 @@ from .const import (
MODEL_AIRFRESH_A1,
MODEL_AIRFRESH_T2017,
MODEL_AIRFRESH_VA2,
+ MODEL_AIRFRESH_VA4,
MODEL_AIRHUMIDIFIER_CA1,
MODEL_AIRHUMIDIFIER_CA4,
MODEL_AIRHUMIDIFIER_CB1,
@@ -226,6 +228,7 @@ NUMBER_TYPES = {
MODEL_TO_FEATURES_MAP = {
MODEL_AIRFRESH_A1: FEATURE_FLAGS_AIRFRESH_A1,
MODEL_AIRFRESH_VA2: FEATURE_FLAGS_AIRFRESH,
+ MODEL_AIRFRESH_VA4: FEATURE_FLAGS_AIRFRESH_VA4,
MODEL_AIRFRESH_T2017: FEATURE_FLAGS_AIRFRESH_T2017,
MODEL_AIRHUMIDIFIER_CA1: FEATURE_FLAGS_AIRHUMIDIFIER_CA_AND_CB,
MODEL_AIRHUMIDIFIER_CA4: FEATURE_FLAGS_AIRHUMIDIFIER_CA4,
diff --git a/homeassistant/components/xiaomi_miio/select.py b/homeassistant/components/xiaomi_miio/select.py
index 14ceac7f2f4..69f8dbfad30 100644
--- a/homeassistant/components/xiaomi_miio/select.py
+++ b/homeassistant/components/xiaomi_miio/select.py
@@ -40,6 +40,7 @@ from .const import (
KEY_DEVICE,
MODEL_AIRFRESH_T2017,
MODEL_AIRFRESH_VA2,
+ MODEL_AIRFRESH_VA4,
MODEL_AIRHUMIDIFIER_CA1,
MODEL_AIRHUMIDIFIER_CA4,
MODEL_AIRHUMIDIFIER_CB1,
@@ -89,6 +90,9 @@ MODEL_TO_ATTR_MAP: dict[str, list] = {
MODEL_AIRFRESH_VA2: [
AttributeEnumMapping(ATTR_LED_BRIGHTNESS, AirfreshLedBrightness)
],
+ MODEL_AIRFRESH_VA4: [
+ AttributeEnumMapping(ATTR_LED_BRIGHTNESS, AirfreshLedBrightness)
+ ],
MODEL_AIRHUMIDIFIER_CA1: [
AttributeEnumMapping(ATTR_LED_BRIGHTNESS, AirhumidifierLedBrightness)
],
diff --git a/homeassistant/components/xiaomi_miio/sensor.py b/homeassistant/components/xiaomi_miio/sensor.py
index 5370357c58c..819d95f208c 100644
--- a/homeassistant/components/xiaomi_miio/sensor.py
+++ b/homeassistant/components/xiaomi_miio/sensor.py
@@ -57,6 +57,7 @@ from .const import (
MODEL_AIRFRESH_A1,
MODEL_AIRFRESH_T2017,
MODEL_AIRFRESH_VA2,
+ MODEL_AIRFRESH_VA4,
MODEL_AIRHUMIDIFIER_CA1,
MODEL_AIRHUMIDIFIER_CB1,
MODEL_AIRPURIFIER_3C,
@@ -516,6 +517,7 @@ FAN_ZA5_SENSORS = (ATTR_HUMIDITY, ATTR_TEMPERATURE)
MODEL_TO_SENSORS_MAP: dict[str, tuple[str, ...]] = {
MODEL_AIRFRESH_A1: AIRFRESH_SENSORS_A1,
MODEL_AIRFRESH_VA2: AIRFRESH_SENSORS,
+ MODEL_AIRFRESH_VA4: AIRFRESH_SENSORS,
MODEL_AIRFRESH_T2017: AIRFRESH_SENSORS_T2017,
MODEL_AIRHUMIDIFIER_CA1: HUMIDIFIER_CA1_CB1_SENSORS,
MODEL_AIRHUMIDIFIER_CB1: HUMIDIFIER_CA1_CB1_SENSORS,
diff --git a/homeassistant/components/xiaomi_miio/switch.py b/homeassistant/components/xiaomi_miio/switch.py
index 89cc80ce74f..577f9af837f 100644
--- a/homeassistant/components/xiaomi_miio/switch.py
+++ b/homeassistant/components/xiaomi_miio/switch.py
@@ -37,6 +37,7 @@ from .const import (
FEATURE_FLAGS_AIRFRESH,
FEATURE_FLAGS_AIRFRESH_A1,
FEATURE_FLAGS_AIRFRESH_T2017,
+ FEATURE_FLAGS_AIRFRESH_VA4,
FEATURE_FLAGS_AIRHUMIDIFIER,
FEATURE_FLAGS_AIRHUMIDIFIER_CA4,
FEATURE_FLAGS_AIRHUMIDIFIER_CA_AND_CB,
@@ -72,6 +73,7 @@ from .const import (
MODEL_AIRFRESH_A1,
MODEL_AIRFRESH_T2017,
MODEL_AIRFRESH_VA2,
+ MODEL_AIRFRESH_VA4,
MODEL_AIRHUMIDIFIER_CA1,
MODEL_AIRHUMIDIFIER_CA4,
MODEL_AIRHUMIDIFIER_CB1,
@@ -182,6 +184,7 @@ SERVICE_TO_METHOD = {
MODEL_TO_FEATURES_MAP = {
MODEL_AIRFRESH_A1: FEATURE_FLAGS_AIRFRESH_A1,
MODEL_AIRFRESH_VA2: FEATURE_FLAGS_AIRFRESH,
+ MODEL_AIRFRESH_VA4: FEATURE_FLAGS_AIRFRESH_VA4,
MODEL_AIRFRESH_T2017: FEATURE_FLAGS_AIRFRESH_T2017,
MODEL_AIRHUMIDIFIER_CA1: FEATURE_FLAGS_AIRHUMIDIFIER_CA_AND_CB,
MODEL_AIRHUMIDIFIER_CA4: FEATURE_FLAGS_AIRHUMIDIFIER_CA4,
--
GitLab