mirror of
https://github.com/sususweet/midea-meiju-codec.git
synced 2026-02-17 09:08:08 +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.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)
|
||||
|
||||
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