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

Use shorthand attributes in tuya vacuum (#70846)

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

37
homeassistant/components/tuya/vacuum.py
View File

@ -78,51 +78,54 @@ class TuyaVacuumEntity(TuyaEntity, StateVacuumEntity):
_fan_speed: EnumTypeData | None = None _fan_speed: EnumTypeData | None = None
_battery_level: IntegerTypeData | None = None _battery_level: IntegerTypeData | None = None
_supported_features = 0
def __init__(self, device: TuyaDevice, device_manager: TuyaDeviceManager) -> None: def __init__(self, device: TuyaDevice, device_manager: TuyaDeviceManager) -> None:
"""Init Tuya vacuum.""" """Init Tuya vacuum."""
super().__init__(device, device_manager) super().__init__(device, device_manager)
self._supported_features |= VacuumEntityFeature.SEND_COMMAND self._attr_supported_features = 0
self._attr_fan_speed_list = []
self._attr_supported_features |= VacuumEntityFeature.SEND_COMMAND
if self.find_dpcode(DPCode.PAUSE, prefer_function=True): if self.find_dpcode(DPCode.PAUSE, prefer_function=True):
self._supported_features |= VacuumEntityFeature.PAUSE self._attr_supported_features |= VacuumEntityFeature.PAUSE
if self.find_dpcode(DPCode.SWITCH_CHARGE, prefer_function=True): if self.find_dpcode(DPCode.SWITCH_CHARGE, prefer_function=True):
self._supported_features |= VacuumEntityFeature.RETURN_HOME self._attr_supported_features |= VacuumEntityFeature.RETURN_HOME
elif ( elif (
enum_type := self.find_dpcode( enum_type := self.find_dpcode(
DPCode.MODE, dptype=DPType.ENUM, prefer_function=True DPCode.MODE, dptype=DPType.ENUM, prefer_function=True
) )
) and TUYA_MODE_RETURN_HOME in enum_type.range: ) and TUYA_MODE_RETURN_HOME in enum_type.range:
self._supported_features |= VacuumEntityFeature.RETURN_HOME self._attr_supported_features |= VacuumEntityFeature.RETURN_HOME
if self.find_dpcode(DPCode.SEEK, prefer_function=True): if self.find_dpcode(DPCode.SEEK, prefer_function=True):
self._supported_features |= VacuumEntityFeature.LOCATE self._attr_supported_features |= VacuumEntityFeature.LOCATE
if self.find_dpcode(DPCode.STATUS, prefer_function=True): if self.find_dpcode(DPCode.STATUS, prefer_function=True):
self._supported_features |= ( self._attr_supported_features |= (
VacuumEntityFeature.STATE | VacuumEntityFeature.STATUS VacuumEntityFeature.STATE | VacuumEntityFeature.STATUS
) )
if self.find_dpcode(DPCode.POWER, prefer_function=True): if self.find_dpcode(DPCode.POWER, prefer_function=True):
self._supported_features |= ( self._attr_supported_features |= (
VacuumEntityFeature.TURN_ON | VacuumEntityFeature.TURN_OFF VacuumEntityFeature.TURN_ON | VacuumEntityFeature.TURN_OFF
) )
if self.find_dpcode(DPCode.POWER_GO, prefer_function=True): if self.find_dpcode(DPCode.POWER_GO, prefer_function=True):
self._supported_features |= ( self._attr_supported_features |= (
VacuumEntityFeature.STOP | VacuumEntityFeature.START VacuumEntityFeature.STOP | VacuumEntityFeature.START
) )
if enum_type := self.find_dpcode( if enum_type := self.find_dpcode(
DPCode.SUCTION, dptype=DPType.ENUM, prefer_function=True DPCode.SUCTION, dptype=DPType.ENUM, prefer_function=True
): ):
self._supported_features |= VacuumEntityFeature.FAN_SPEED
self._fan_speed = enum_type self._fan_speed = enum_type
self._attr_fan_speed_list = enum_type.range
self._attr_supported_features |= VacuumEntityFeature.FAN_SPEED
if int_type := self.find_dpcode(DPCode.ELECTRICITY_LEFT, dptype=DPType.INTEGER): if int_type := self.find_dpcode(DPCode.ELECTRICITY_LEFT, dptype=DPType.INTEGER):
self._supported_features |= VacuumEntityFeature.BATTERY self._attr_supported_features |= VacuumEntityFeature.BATTERY
self._battery_level = int_type self._battery_level = int_type
@property @property
@ -139,13 +142,6 @@ class TuyaVacuumEntity(TuyaEntity, StateVacuumEntity):
"""Return the fan speed of the vacuum cleaner.""" """Return the fan speed of the vacuum cleaner."""
return self.device.status.get(DPCode.SUCTION) return self.device.status.get(DPCode.SUCTION)
@property
def fan_speed_list(self) -> list[str]:
"""Get the list of available fan speed steps of the vacuum cleaner."""
if self._fan_speed is None:
return []
return self._fan_speed.range
@property @property
def state(self) -> str | None: def state(self) -> str | None:
"""Return Tuya vacuum device state.""" """Return Tuya vacuum device state."""
@ -157,11 +153,6 @@ class TuyaVacuumEntity(TuyaEntity, StateVacuumEntity):
return None return None
return TUYA_STATUS_TO_HA.get(status) return TUYA_STATUS_TO_HA.get(status)
@property
def supported_features(self) -> int:
"""Flag supported features."""
return self._supported_features
def turn_on(self, **kwargs: Any) -> None: def turn_on(self, **kwargs: Any) -> None:
"""Turn the device on.""" """Turn the device on."""
self._send_command([{"code": DPCode.POWER, "value": True}]) self._send_command([{"code": DPCode.POWER, "value": True}])