Update tuya for new fan entity model (#45870)

pull/46522/head
J. Nick Koston 2021-02-13 16:23:19 -10:00 committed by GitHub
parent 17a4678906
commit 1845f69729
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 18 additions and 23 deletions

View File

@ -10,6 +10,10 @@ from homeassistant.components.fan import (
)
from homeassistant.const import CONF_PLATFORM, STATE_OFF
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.util.percentage import (
ordered_list_item_to_percentage,
percentage_to_ordered_list_item,
)
from . import TuyaDevice
from .const import DOMAIN, TUYA_DATA, TUYA_DISCOVERY_NEW
@ -61,27 +65,21 @@ class TuyaFanDevice(TuyaDevice, FanEntity):
"""Init Tuya fan device."""
super().__init__(tuya, platform)
self.entity_id = ENTITY_ID_FORMAT.format(tuya.object_id())
self.speeds = [STATE_OFF]
self.speeds = []
async def async_added_to_hass(self):
"""Create fan list when add to hass."""
await super().async_added_to_hass()
self.speeds.extend(self._tuya.speed_list())
def set_speed(self, speed: str) -> None:
"""Set the speed of the fan."""
if speed == STATE_OFF:
def set_percentage(self, percentage: int) -> None:
"""Set the speed percentage of the fan."""
if percentage == 0:
self.turn_off()
else:
self._tuya.set_speed(speed)
tuya_speed = percentage_to_ordered_list_item(self.speeds, percentage)
self._tuya.set_speed(tuya_speed)
#
# The fan entity model has changed to use percentages and preset_modes
# instead of speeds.
#
# Please review
# https://developers.home-assistant.io/docs/core/entity/fan/
#
def turn_on(
self,
speed: str = None,
@ -90,8 +88,8 @@ class TuyaFanDevice(TuyaDevice, FanEntity):
**kwargs,
) -> None:
"""Turn on the fan."""
if speed is not None:
self.set_speed(speed)
if percentage is not None:
self.set_percentage(percentage)
else:
self._tuya.turn_on()
@ -118,16 +116,13 @@ class TuyaFanDevice(TuyaDevice, FanEntity):
return self._tuya.state()
@property
def speed(self) -> str:
def percentage(self) -> str:
"""Return the current speed."""
if self.is_on:
return self._tuya.speed()
return STATE_OFF
@property
def speed_list(self) -> list:
"""Get the list of available speeds."""
return self.speeds
if not self.is_on:
return 0
if self.speeds is None:
return None
return ordered_list_item_to_percentage(self.speeds, self._tuya.speed())
@property
def supported_features(self) -> int: