jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Use VacuumEntityFeature in xiaomi_miio (#70564)

Browse Source
This commit is contained in:
epenet 2022-04-23 23:45:10 +02:00 committed by GitHub
parent 7c3fcdee95
commit 4247318634
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 25 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

40
homeassistant/components/xiaomi_miio/vacuum.py
View File

@ -14,17 +14,8 @@ from homeassistant.components.vacuum import (
STATE_IDLE, STATE_IDLE,
STATE_PAUSED, STATE_PAUSED,
STATE_RETURNING, STATE_RETURNING,
SUPPORT_BATTERY,
SUPPORT_CLEAN_SPOT,
SUPPORT_FAN_SPEED,
SUPPORT_LOCATE,
SUPPORT_PAUSE,
SUPPORT_RETURN_HOME,
SUPPORT_SEND_COMMAND,
SUPPORT_START,
SUPPORT_STATE,
SUPPORT_STOP,
StateVacuumEntity, StateVacuumEntity,
VacuumEntityFeature,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
@ -61,20 +52,6 @@ ATTR_ZONE_ARRAY = "zone"
ATTR_ZONE_REPEATER = "repeats" ATTR_ZONE_REPEATER = "repeats"
ATTR_TIMERS = "timers" ATTR_TIMERS = "timers"
SUPPORT_XIAOMI = (
SUPPORT_STATE
| SUPPORT_PAUSE
| SUPPORT_STOP
| SUPPORT_RETURN_HOME
| SUPPORT_FAN_SPEED
| SUPPORT_SEND_COMMAND
| SUPPORT_LOCATE
| SUPPORT_BATTERY
| SUPPORT_CLEAN_SPOT
| SUPPORT_START
)
STATE_CODE_TO_STATE = { STATE_CODE_TO_STATE = {
1: STATE_IDLE, # "Starting" 1: STATE_IDLE, # "Starting"
2: STATE_IDLE, # "Charger disconnected" 2: STATE_IDLE, # "Charger disconnected"
@ -209,6 +186,19 @@ class MiroboVacuum(
): ):
"""Representation of a Xiaomi Vacuum cleaner robot.""" """Representation of a Xiaomi Vacuum cleaner robot."""
_attr_supported_features = (
VacuumEntityFeature.STATE
| VacuumEntityFeature.PAUSE
| VacuumEntityFeature.STOP
| VacuumEntityFeature.RETURN_HOME
| VacuumEntityFeature.FAN_SPEED
| VacuumEntityFeature.SEND_COMMAND
| VacuumEntityFeature.LOCATE
| VacuumEntityFeature.BATTERY
| VacuumEntityFeature.CLEAN_SPOT
| VacuumEntityFeature.START
)
def __init__( def __init__(
self, self,
name, name,
@ -289,7 +279,7 @@ class MiroboVacuum(
@property @property
def supported_features(self): def supported_features(self):
"""Flag vacuum cleaner robot features that are supported.""" """Flag vacuum cleaner robot features that are supported."""
return SUPPORT_XIAOMI return self._attr_supported_features
async def _try_command(self, mask_error, func, *args, **kwargs): async def _try_command(self, mask_error, func, *args, **kwargs):
"""Call a vacuum command handling error messages.""" """Call a vacuum command handling error messages."""