From b0e2681c4e084da82c59412abd7deb6e6d47f824 Mon Sep 17 00:00:00 2001 From: Cyborg2017 Date: Mon, 9 Feb 2026 14:27:36 +0800 Subject: [PATCH] feat: add vacuum platform support with vacuum.py implementation - Add Platform.VACUUM to supported platforms list - Implement vacuum.py with core functionality - Update device registry to handle vacuum devices --- .../midea_auto_cloud/__init__.py | 4 +- custom_components/midea_auto_cloud/vacuum.py | 195 ++++++++++++++++++ 2 files changed, 198 insertions(+), 1 deletion(-) create mode 100644 custom_components/midea_auto_cloud/vacuum.py diff --git a/custom_components/midea_auto_cloud/__init__.py b/custom_components/midea_auto_cloud/__init__.py index 3f671c3..ae8e85d 100644 --- a/custom_components/midea_auto_cloud/__init__.py +++ b/custom_components/midea_auto_cloud/__init__.py @@ -61,7 +61,8 @@ PLATFORMS: list[Platform] = [ Platform.LIGHT, Platform.HUMIDIFIER, Platform.NUMBER, - Platform.BUTTON + Platform.BUTTON, + Platform.VACUUM ] async def import_module_async(module_name): @@ -381,6 +382,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry): str(Platform.SWITCH), str(Platform.FAN), str(Platform.SELECT), + str(Platform.VACUUM), ]: for entity_key in platform_cfg.keys(): preset_keys.add(entity_key) diff --git a/custom_components/midea_auto_cloud/vacuum.py b/custom_components/midea_auto_cloud/vacuum.py new file mode 100644 index 0000000..a744526 --- /dev/null +++ b/custom_components/midea_auto_cloud/vacuum.py @@ -0,0 +1,195 @@ +from homeassistant.components.vacuum import ( + StateVacuumEntity, + VacuumEntityFeature, + VacuumActivity, +) +from homeassistant.config_entries import ConfigEntry +from homeassistant.const import Platform +from homeassistant.core import HomeAssistant +from homeassistant.helpers.entity_platform import AddEntitiesCallback + +from .const import DOMAIN +from .core.logger import MideaLogger +from .midea_entity import MideaEntity +from . import load_device_config + +async def async_setup_entry( + hass: HomeAssistant, + config_entry: ConfigEntry, + async_add_entities: AddEntitiesCallback, +) -> None: + """Set up vacuum entities for Midea devices.""" + # 账号型 entry：从 __init__ 写入的 accounts 桶加载设备和协调器 + account_bucket = hass.data.get(DOMAIN, {}).get("accounts", {}).get(config_entry.entry_id) + if not account_bucket: + async_add_entities([]) + return + device_list = account_bucket.get("device_list", {}) + coordinator_map = account_bucket.get("coordinator_map", {}) + + devs = [] + for device_id, info in device_list.items(): + device_type = info.get("type") + sn8 = info.get("sn8") + config = await load_device_config(hass, device_type, sn8) or {} + entities_cfg = (config.get("entities") or {}).get(Platform.VACUUM, {}) + manufacturer = config.get("manufacturer") + rationale = config.get("rationale") + coordinator = coordinator_map.get(device_id) + device = coordinator.device if coordinator else None + + for entity_key, ecfg in entities_cfg.items(): + devs.append(MideaVacuumEntity( + coordinator, device, manufacturer, rationale, entity_key, ecfg + )) + async_add_entities(devs) + +class MideaVacuumEntity(MideaEntity, StateVacuumEntity): + def __init__(self, coordinator, device, manufacturer, rationale, entity_key, config): + super().__init__( + coordinator, + device.device_id, + device.device_name, + f"T0x{device.device_type:02X}", + device.sn, + device.sn8, + device.model, + entity_key, + device=device, + manufacturer=manufacturer, + rationale=rationale, + config=config, + ) + self._key_battery_level = self._config.get("battery_level") + self._key_status = self._config.get("status") + self._key_fan_speeds = self._config.get("fan_speeds") + #self._key_locate = self._config.get("locate") + #self._key_clean_spot = self._config.get("clean_spot") + #self._key_map = self._config.get("map") + + @property + def supported_features(self): + features = VacuumEntityFeature(0) + features |= VacuumEntityFeature.STOP + features |= VacuumEntityFeature.PAUSE + features |= VacuumEntityFeature.START + features |= VacuumEntityFeature.RETURN_HOME + features |= VacuumEntityFeature.FAN_SPEED + features |= VacuumEntityFeature.STATUS + features |= VacuumEntityFeature.BATTERY + #features |= VacuumEntityFeature.LOCATE + #features |= VacuumEntityFeature.CLEAN_SPOT + #features |= VacuumEntityFeature.MAP + return features + + @property + def battery_level(self): + """Return the battery level of the vacuum cleaner.""" + battery = self._get_nested_value(self._key_battery_level) + if battery is not None: + try: + return int(battery) + except (ValueError, TypeError): + return None + return None + + @property + def status(self): + """Return the status of the vacuum cleaner.""" + status = self._get_nested_value(self._key_status) + if status is not None: + return status + return None + + @property + def state(self): + """Return the state of the vacuum cleaner.""" + status = self.status + if not status: + return None + + # Map Midea status to Home Assistant states + status_mapping = { + # === 清洁中状态 (CLEANING) === + "work": VacuumActivity.CLEANING, # 清扫中 + "auto_clean": VacuumActivity.CLEANING, # 自动清扫中 + + # === 已停靠状态 (DOCKED) === + "charging_on_dock": VacuumActivity.DOCKED, # 座充中 + "on_base": VacuumActivity.DOCKED, # 在基站上 + "charge_finish": VacuumActivity.DOCKED, # 充电完成 + + # === 空闲状态 (IDLE) === + "stop": VacuumActivity.IDLE, # 已停止 + "sleep": VacuumActivity.IDLE, # 休眠中 + + # === 暂停状态 (PAUSED) === + "clean_pause": VacuumActivity.PAUSED, # 清扫暂停 + "charge_pause": VacuumActivity.PAUSED, # 充电暂停 + + # === 返回中状态 (RETURNING) === + "charging": VacuumActivity.RETURNING, # 返回基站中 + + # === 错误状态 (ERROR) === + "error": VacuumActivity.ERROR, # 错误 + } + + return status_mapping.get(status, status) + + @property + def fan_speed(self): + """Return the current fan speed.""" + return self._dict_get_selected(self._key_fan_speeds) + + @property + def fan_speed_list(self): + """Return the list of available fan speeds.""" + return list(self._key_fan_speeds.keys()) + + async def async_start(self): + """Start or resume the cleaning task.""" + # 设置为工作状态 + if self._key_status: + await self.async_set_attribute(self._key_status, "work") + else: + await self._async_set_status_on_off(self._key_power, True) + + async def async_stop(self): + """Stop the vacuum cleaner.""" + # 设置为停止状态 + if self._key_status: + await self.async_set_attribute(self._key_status, "stop") + else: + await self._async_set_status_on_off(self._key_power, False) + + async def async_pause(self): + """Pause the cleaning task.""" + # 设置为暂停状态 + if self._key_status: + await self.async_set_attribute(self._key_status, "pause") + + async def async_return_to_base(self): + """Return the vacuum cleaner to its base.""" + # 设置为回基站状态 + if self._key_status: + await self.async_set_attribute(self._key_status, "charge") + + async def async_set_fan_speed(self, fan_speed: str): + """Set the fan speed.""" + new_status = self._key_fan_speeds.get(fan_speed) + if new_status is not None: + await self.async_set_attributes(new_status) + + #async def async_locate(self): + #"""Locate the vacuum cleaner.""" + # 定位设备 + # 具体实现取决于设备的控制方式 + #if hasattr(self, "_key_locate"): + #await self.async_set_attribute(self._key_locate, True) + + #async def async_clean_spot(self): + #"""Perform a clean spot.""" + # 执行定点清扫 + # 具体实现取决于设备的控制方式 + #if hasattr(self, "_key_clean_spot"): + #await self.async_set_attribute(self._key_clean_spot, True)