mirror of
https://github.com/sususweet/midea-meiju-codec.git
synced 2026-02-27 14:16:35 +00:00
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
This commit is contained in:
@@ -61,7 +61,8 @@ PLATFORMS: list[Platform] = [
|
|||||||
Platform.LIGHT,
|
Platform.LIGHT,
|
||||||
Platform.HUMIDIFIER,
|
Platform.HUMIDIFIER,
|
||||||
Platform.NUMBER,
|
Platform.NUMBER,
|
||||||
Platform.BUTTON
|
Platform.BUTTON,
|
||||||
|
Platform.VACUUM
|
||||||
]
|
]
|
||||||
|
|
||||||
async def import_module_async(module_name):
|
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.SWITCH),
|
||||||
str(Platform.FAN),
|
str(Platform.FAN),
|
||||||
str(Platform.SELECT),
|
str(Platform.SELECT),
|
||||||
|
str(Platform.VACUUM),
|
||||||
]:
|
]:
|
||||||
for entity_key in platform_cfg.keys():
|
for entity_key in platform_cfg.keys():
|
||||||
preset_keys.add(entity_key)
|
preset_keys.add(entity_key)
|
||||||
|
|||||||
195
custom_components/midea_auto_cloud/vacuum.py
Normal file
195
custom_components/midea_auto_cloud/vacuum.py
Normal file
@@ -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)
|
||||||
Reference in New Issue
Block a user