From f2a99e83cd11fd67dcecdeac7fc81998de552ab0 Mon Sep 17 00:00:00 2001
From: huangyupeng <huangyupeng1111@163.com>
Date: Fri, 20 Jul 2018 17:23:09 +0800
Subject: [PATCH] Add Tuya fan support (#15525)
* Add Tuya fan platform
* Add Tuya fan platform
* fix as review required
---
homeassistant/components/fan/tuya.py | 99 ++++++++++++++++++++++++++++
homeassistant/components/tuya.py | 1 +
2 files changed, 100 insertions(+)
create mode 100644 homeassistant/components/fan/tuya.py
diff --git a/homeassistant/components/fan/tuya.py b/homeassistant/components/fan/tuya.py
new file mode 100644
index 00000000000..f19a9e5a5f7
--- /dev/null
+++ b/homeassistant/components/fan/tuya.py
@@ -0,0 +1,99 @@
+"""
+Support for Tuya fans.
+
+For more details about this platform, please refer to the documentation at
+https://home-assistant.io/components/fan.tuya/
+"""
+
+from homeassistant.components.fan import (
+ ENTITY_ID_FORMAT, FanEntity, SUPPORT_OSCILLATE, SUPPORT_SET_SPEED)
+from homeassistant.components.tuya import DATA_TUYA, TuyaDevice
+from homeassistant.const import STATE_OFF
+
+DEPENDENCIES = ['tuya']
+
+
+def setup_platform(hass, config, add_devices, discovery_info=None):
+ """Set up Tuya fan platform."""
+ if discovery_info is None:
+ return
+ tuya = hass.data[DATA_TUYA]
+ dev_ids = discovery_info.get('dev_ids')
+ devices = []
+ for dev_id in dev_ids:
+ device = tuya.get_device_by_id(dev_id)
+ if device is None:
+ continue
+ devices.append(TuyaFanDevice(device))
+ add_devices(devices)
+
+
+class TuyaFanDevice(TuyaDevice, FanEntity):
+ """Tuya fan devices."""
+
+ def __init__(self, tuya):
+ """Init Tuya fan device."""
+ super().__init__(tuya)
+ self.entity_id = ENTITY_ID_FORMAT.format(tuya.object_id())
+ self.speeds = [STATE_OFF]
+
+ 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:
+ self.turn_off()
+ else:
+ self.tuya.set_speed(speed)
+
+ def turn_on(self, speed: str = None, **kwargs) -> None:
+ """Turn on the fan."""
+ if speed is not None:
+ self.set_speed(speed)
+ else:
+ self.tuya.turn_on()
+
+ def turn_off(self, **kwargs) -> None:
+ """Turn the entity off."""
+ self.tuya.turn_off()
+
+ def oscillate(self, oscillating) -> None:
+ """Oscillate the fan."""
+ self.tuya.oscillate(oscillating)
+
+ @property
+ def oscillating(self):
+ """Return current oscillating status."""
+ if self.supported_features & SUPPORT_OSCILLATE == 0:
+ return None
+ if self.speed == STATE_OFF:
+ return False
+ return self.tuya.oscillating()
+
+ @property
+ def is_on(self):
+ """Return true if the entity is on."""
+ return self.tuya.state()
+
+ @property
+ def speed(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
+
+ @property
+ def supported_features(self) -> int:
+ """Flag supported features."""
+ supports = SUPPORT_SET_SPEED
+ if self.tuya.support_oscillate():
+ supports = supports | SUPPORT_OSCILLATE
+ return supports
diff --git a/homeassistant/components/tuya.py b/homeassistant/components/tuya.py
index ccb5227aead..337071baf69 100644
--- a/homeassistant/components/tuya.py
+++ b/homeassistant/components/tuya.py
@@ -34,6 +34,7 @@ SERVICE_PULL_DEVICES = 'pull_devices'
TUYA_TYPE_TO_HA = {
'climate': 'climate',
+ 'fan': 'fan',
'light': 'light',
'switch': 'switch',
}
--
GitLab