williamp/localtuya-modded
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Convert fan platform to asyncio

Browse Source
This commit is contained in:
Pierre Ståhl
2020-10-14 13:45:41 +02:00
committed by rospogrigio
parent 136e44c549
commit 5d64cb4e18
1 changed files with 111 additions and 113 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
custom_components/localtuya/fan.py
View File

@@ -36,8 +36,8 @@ class LocaltuyaFan(LocalTuyaEntity, FanEntity):
"""Initialize the entity.""" """Initialize the entity."""
super().__init__(device, config_entry, fanid, **kwargs) super().__init__(device, config_entry, fanid, **kwargs)
self._is_on = False self._is_on = False
self._speed = SPEED_OFF self._speed = None
self._oscillating = False self._oscillating = None
@property @property
def oscillating(self): def oscillating(self):
@@ -59,36 +59,34 @@ class LocaltuyaFan(LocalTuyaEntity, FanEntity):
"""Get the list of available speeds.""" """Get the list of available speeds."""
return [SPEED_OFF, SPEED_LOW, SPEED_MEDIUM, SPEED_HIGH] return [SPEED_OFF, SPEED_LOW, SPEED_MEDIUM, SPEED_HIGH]
def turn_on(self, speed: str = None, **kwargs) -> None: async def async_turn_on(self, speed: str = None, **kwargs) -> None:
"""Turn on the entity.""" """Turn on the entity."""
self._device.set_dps(True, "1") await self._device.set_dps(True, "1")
if speed is not None: if speed is not None:
self.set_speed(speed) await self.async_set_speed(speed)
else: else:
self.schedule_update_ha_state() self.schedule_update_ha_state()
def turn_off(self, **kwargs) -> None: async def async_turn_off(self, **kwargs) -> None:
"""Turn off the entity.""" """Turn off the entity."""
self._device.set_dps(False, "1") await self._device.set_dps(False, "1")
self.schedule_update_ha_state() self.schedule_update_ha_state()
def set_speed(self, speed: str) -> None: async def async_set_speed(self, speed: str) -> None:
"""Set the speed of the fan.""" """Set the speed of the fan."""
self._speed = speed
if speed == SPEED_OFF: if speed == SPEED_OFF:
self._device.set_dps(False, "1") await self._device.set_dps(False, "1")
elif speed == SPEED_LOW: elif speed == SPEED_LOW:
self._device.set_dps("1", "2") await self._device.set_dps("1", "2")
elif speed == SPEED_MEDIUM: elif speed == SPEED_MEDIUM:
self._device.set_dps("2", "2") await self._device.set_dps("2", "2")
elif speed == SPEED_HIGH: elif speed == SPEED_HIGH:
self._device.set_dps("3", "2") await self._device.set_dps("3", "2")
self.schedule_update_ha_state() self.schedule_update_ha_state()
def oscillate(self, oscillating: bool) -> None: async def async_oscillate(self, oscillating: bool) -> None:
"""Set oscillation.""" """Set oscillation."""
self._oscillating = oscillating await self._device.set_dps(oscillating, "8")
self._device.set_value("8", oscillating)
self.schedule_update_ha_state() self.schedule_update_ha_state()
@property @property