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

Use VacuumEntityFeature in neato (#70561)

Browse Source
This commit is contained in:
epenet 2022-04-23 23:42:32 +02:00 committed by GitHub
parent ca337b54a3
commit c1888ce559
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 14 additions and 22 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

36
homeassistant/components/neato/vacuum.py
View File

@ -15,16 +15,8 @@ from homeassistant.components.vacuum import (
STATE_DOCKED, STATE_DOCKED,
STATE_ERROR, STATE_ERROR,
STATE_RETURNING, STATE_RETURNING,
SUPPORT_BATTERY,
SUPPORT_CLEAN_SPOT,
SUPPORT_LOCATE,
SUPPORT_MAP,
SUPPORT_PAUSE,
SUPPORT_RETURN_HOME,
SUPPORT_START,
SUPPORT_STATE,
SUPPORT_STOP,
StateVacuumEntity, StateVacuumEntity,
VacuumEntityFeature,
) )
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.const import ATTR_MODE, STATE_IDLE, STATE_PAUSED from homeassistant.const import ATTR_MODE, STATE_IDLE, STATE_PAUSED
@ -51,18 +43,6 @@ _LOGGER = logging.getLogger(__name__)
SCAN_INTERVAL = timedelta(minutes=SCAN_INTERVAL_MINUTES) SCAN_INTERVAL = timedelta(minutes=SCAN_INTERVAL_MINUTES)
SUPPORT_NEATO = (
SUPPORT_BATTERY
| SUPPORT_PAUSE
| SUPPORT_RETURN_HOME
| SUPPORT_STOP
| SUPPORT_START
| SUPPORT_CLEAN_SPOT
| SUPPORT_STATE
| SUPPORT_MAP
| SUPPORT_LOCATE
)
ATTR_CLEAN_START = "clean_start" ATTR_CLEAN_START = "clean_start"
ATTR_CLEAN_STOP = "clean_stop" ATTR_CLEAN_STOP = "clean_stop"
ATTR_CLEAN_AREA = "clean_area" ATTR_CLEAN_AREA = "clean_area"
@ -114,6 +94,18 @@ async def async_setup_entry(
class NeatoConnectedVacuum(StateVacuumEntity): class NeatoConnectedVacuum(StateVacuumEntity):
"""Representation of a Neato Connected Vacuum.""" """Representation of a Neato Connected Vacuum."""
_attr_supported_features = (
VacuumEntityFeature.BATTERY
| VacuumEntityFeature.PAUSE
| VacuumEntityFeature.RETURN_HOME
| VacuumEntityFeature.STOP
| VacuumEntityFeature.START
| VacuumEntityFeature.CLEAN_SPOT
| VacuumEntityFeature.STATE
| VacuumEntityFeature.MAP
| VacuumEntityFeature.LOCATE
)
def __init__( def __init__(
self, self,
neato: NeatoHub, neato: NeatoHub,
@ -276,7 +268,7 @@ class NeatoConnectedVacuum(StateVacuumEntity):
@property @property
def supported_features(self) -> int: def supported_features(self) -> int:
"""Flag vacuum cleaner robot features that are supported.""" """Flag vacuum cleaner robot features that are supported."""
return SUPPORT_NEATO return self._attr_supported_features
@property @property
def battery_level(self) -> int | None: def battery_level(self) -> int | None: