Update tuya vacuum supported features (#95832)

This commit is contained in:
Erik Montnemery 2023-07-04 13:19:16 +02:00 committed by GitHub
parent 91087392fe
commit c84dacf2fc
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -86,7 +86,9 @@ class TuyaVacuumEntity(TuyaEntity, StateVacuumEntity):
self._attr_fan_speed_list = [] self._attr_fan_speed_list = []
self._attr_supported_features |= VacuumEntityFeature.SEND_COMMAND self._attr_supported_features = (
VacuumEntityFeature.SEND_COMMAND | VacuumEntityFeature.STATE
)
if self.find_dpcode(DPCode.PAUSE, prefer_function=True): if self.find_dpcode(DPCode.PAUSE, prefer_function=True):
self._attr_supported_features |= VacuumEntityFeature.PAUSE self._attr_supported_features |= VacuumEntityFeature.PAUSE
@ -102,11 +104,6 @@ class TuyaVacuumEntity(TuyaEntity, StateVacuumEntity):
if self.find_dpcode(DPCode.SEEK, prefer_function=True): if self.find_dpcode(DPCode.SEEK, prefer_function=True):
self._attr_supported_features |= VacuumEntityFeature.LOCATE self._attr_supported_features |= VacuumEntityFeature.LOCATE
if self.find_dpcode(DPCode.STATUS, prefer_function=True):
self._attr_supported_features |= (
VacuumEntityFeature.STATE | VacuumEntityFeature.STATUS
)
if self.find_dpcode(DPCode.POWER, prefer_function=True): if self.find_dpcode(DPCode.POWER, prefer_function=True):
self._attr_supported_features |= ( self._attr_supported_features |= (
VacuumEntityFeature.TURN_ON | VacuumEntityFeature.TURN_OFF VacuumEntityFeature.TURN_ON | VacuumEntityFeature.TURN_OFF