Midea/midea_ac_lan/midea/devices/ce/device.py

158 lines
5.4 KiB
Python
Raw Normal View History

2023-09-22 05:45:40 +00:00
import logging
import json
from .message import (
MessageQuery,
MessageCEResponse,
MessageSet
)
try:
from enum import StrEnum
except ImportError:
from ...backports.enum import StrEnum
from ...core.device import MiedaDevice
_LOGGER = logging.getLogger(__name__)
class DeviceAttributes(StrEnum):
power = "power"
mode = "mode"
child_lock = "child_lock"
scheduled = "scheduled"
fan_speed = "fan_speed"
pm25 = "pm25"
co2 = "co2"
current_humidity = "current_humidity"
current_temperature = "current_temperature"
hcho = "hcho"
link_to_ac = "link_to_ac"
sleep_mode = "sleep_mode"
eco_mode = "eco_mode"
aux_heating = "aux_heating'"
powerful_purify = "powerful_purify"
filter_cleaning_reminder = "filter_cleaning_reminder"
filter_change_reminder = "filter_change_reminder"
error_code = "error_code"
class MideaCEDevice(MiedaDevice):
_modes = [
"Normal", "Sleep mode", "ECO mode"
]
def __init__(
self,
name: str,
device_id: int,
ip_address: str,
port: int,
token: str,
key: str,
protocol: int,
model: str,
customize: str
):
super().__init__(
name=name,
device_id=device_id,
device_type=0xCE,
ip_address=ip_address,
port=port,
token=token,
key=key,
protocol=protocol,
model=model,
attributes={
DeviceAttributes.power: False,
DeviceAttributes.mode: None,
DeviceAttributes.child_lock: False,
DeviceAttributes.scheduled: False,
DeviceAttributes.fan_speed: 0,
DeviceAttributes.pm25: None,
DeviceAttributes.co2: None,
DeviceAttributes.current_humidity: None,
DeviceAttributes.current_temperature: None,
DeviceAttributes.hcho: None,
DeviceAttributes.link_to_ac: False,
DeviceAttributes.sleep_mode: False,
DeviceAttributes.eco_mode: False,
DeviceAttributes.aux_heating: None,
DeviceAttributes.powerful_purify: False,
DeviceAttributes.filter_cleaning_reminder: False,
DeviceAttributes.filter_change_reminder: False,
DeviceAttributes.error_code: 0
})
self._default_speed_count = 7
self._speed_count = self._default_speed_count
self.set_customize(customize)
@property
def speed_count(self):
return self._speed_count
@property
def preset_modes(self):
return self._modes
def build_query(self):
return [MessageQuery(self._device_protocol_version)]
def process_message(self, msg):
message = MessageCEResponse(msg)
_LOGGER.debug(f"[{self.device_id}] Received: {message}")
new_status = {}
for status in self._attributes.keys():
if hasattr(message, str(status)):
value = getattr(message, str(status))
self._attributes[status] = value
new_status[str(status)] = self._attributes[status]
if self._attributes[DeviceAttributes.sleep_mode]:
self._attributes[DeviceAttributes.mode] = "Sleep mode"
elif self._attributes[DeviceAttributes.eco_mode]:
self._attributes[DeviceAttributes.mode] = "ECO mode"
else:
self._attributes[DeviceAttributes.mode] = "None"
new_status[DeviceAttributes.mode.value] = self._attributes[DeviceAttributes.mode]
return new_status
def make_message_set(self):
message = MessageSet(self._device_protocol_version)
message.power = self._attributes[DeviceAttributes.power]
message.fan_speed = self._attributes[DeviceAttributes.fan_speed]
message.link_to_ac = self._attributes[DeviceAttributes.link_to_ac]
message.sleep_mode = self._attributes[DeviceAttributes.sleep_mode]
message.eco_mode = self._attributes[DeviceAttributes.eco_mode]
message.aux_heating = self._attributes[DeviceAttributes.aux_heating]
message.powerful_purify = self._attributes[DeviceAttributes.powerful_purify]
message.scheduled = self._attributes[DeviceAttributes.scheduled]
message.child_lock = self._attributes[DeviceAttributes.child_lock]
return message
def set_attribute(self, attr, value):
message = self.make_message_set()
if attr == DeviceAttributes.mode:
message.sleep_mode = False
message.eco_mode = False
if value == "Sleep mode":
message.sleep_mode = True
elif value == "ECO mode":
message.eco_mode = True
else:
setattr(message, str(attr), value)
self.build_send(message)
def set_customize(self, customize):
self._speed_count = self._default_speed_count
if customize and len(customize) > 0:
try:
params = json.loads(customize)
if params and "speed_count" in params:
self._speed_count = params.get("speed_count")
except Exception as e:
_LOGGER.error(f"[{self.device_id}] Set customize error: {repr(e)}")
self.update_all({"speed_count": self._speed_count})
class MideaAppliance(MideaCEDevice):
pass