From 1b3e9f30674f7dc27a147bb76218c958f538595a Mon Sep 17 00:00:00 2001
From: Alone <hi@anlo.ng>
Date: Wed, 10 Apr 2024 23:49:56 +0800
Subject: [PATCH] =?UTF-8?q?=F0=9F=94=A5=20improve=20for=20climate=20(#1620?=
 =?UTF-8?q?)?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 custom_components/xiaomi_miot/climate.py                | 2 --
 custom_components/xiaomi_miot/core/device_customizes.py | 2 +-
 2 files changed, 1 insertion(+), 3 deletions(-)

diff --git a/custom_components/xiaomi_miot/climate.py b/custom_components/xiaomi_miot/climate.py
index d9f25403b..afdc6fba4 100644
--- a/custom_components/xiaomi_miot/climate.py
+++ b/custom_components/xiaomi_miot/climate.py
@@ -165,8 +165,6 @@ def __init__(self, config: dict, miot_service: MiotService):
             self._supported_features |= ClimateEntityFeature.FAN_MODE
         if self._prop_horizontal_swing or self._prop_vertical_swing:
             self._supported_features |= ClimateEntityFeature.SWING_MODE
-        if self._prop_heater and miot_service.name in ['air_conditioner', 'air_condition_outlet']:
-            self._supported_features |= ClimateEntityFeature.AUX_HEAT
 
         self._power_modes = []
         if miot_service.get_property('heat_level'):
diff --git a/custom_components/xiaomi_miot/core/device_customizes.py b/custom_components/xiaomi_miot/core/device_customizes.py
index c764261f2..7875ada02 100644
--- a/custom_components/xiaomi_miot/core/device_customizes.py
+++ b/custom_components/xiaomi_miot/core/device_customizes.py
@@ -1678,7 +1678,7 @@
 
     '*.aircondition.*': {
         'sensor_properties': 'electricity.electricity',
-        'switch_properties': 'air_conditioner.on,alarm.alarm',
+        'switch_properties': 'air_conditioner.on,alarm.alarm,heater',
         'fan_services': 'air_fresh',
     },
     '*.airer.*': {