102 lines
2.7 KiB
Python
102 lines
2.7 KiB
Python
"""Support for Lutron Caseta fans."""
|
|
import logging
|
|
|
|
from pylutron_caseta import FAN_HIGH, FAN_LOW, FAN_MEDIUM, FAN_MEDIUM_HIGH, FAN_OFF
|
|
|
|
from homeassistant.components.fan import (
|
|
DOMAIN,
|
|
SPEED_HIGH,
|
|
SPEED_LOW,
|
|
SPEED_MEDIUM,
|
|
SPEED_OFF,
|
|
SUPPORT_SET_SPEED,
|
|
FanEntity,
|
|
)
|
|
|
|
from . import DOMAIN as CASETA_DOMAIN, LutronCasetaDevice
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
VALUE_TO_SPEED = {
|
|
None: SPEED_OFF,
|
|
FAN_OFF: SPEED_OFF,
|
|
FAN_LOW: SPEED_LOW,
|
|
FAN_MEDIUM: SPEED_MEDIUM,
|
|
FAN_MEDIUM_HIGH: SPEED_MEDIUM,
|
|
FAN_HIGH: SPEED_HIGH,
|
|
}
|
|
|
|
SPEED_TO_VALUE = {
|
|
SPEED_OFF: FAN_OFF,
|
|
SPEED_LOW: FAN_LOW,
|
|
SPEED_MEDIUM: FAN_MEDIUM,
|
|
SPEED_HIGH: FAN_HIGH,
|
|
}
|
|
|
|
FAN_SPEEDS = [SPEED_OFF, SPEED_LOW, SPEED_MEDIUM, SPEED_HIGH]
|
|
|
|
|
|
async def async_setup_entry(hass, config_entry, async_add_entities):
|
|
"""Set up the Lutron Caseta fan platform.
|
|
|
|
Adds fan controllers from the Caseta bridge associated with the config_entry
|
|
as fan entities.
|
|
"""
|
|
|
|
entities = []
|
|
bridge = hass.data[CASETA_DOMAIN][config_entry.entry_id]
|
|
fan_devices = bridge.get_devices_by_domain(DOMAIN)
|
|
|
|
for fan_device in fan_devices:
|
|
entity = LutronCasetaFan(fan_device, bridge)
|
|
entities.append(entity)
|
|
|
|
async_add_entities(entities, True)
|
|
|
|
|
|
class LutronCasetaFan(LutronCasetaDevice, FanEntity):
|
|
"""Representation of a Lutron Caseta fan. Including Fan Speed."""
|
|
|
|
@property
|
|
def speed(self) -> str:
|
|
"""Return the current speed."""
|
|
return VALUE_TO_SPEED[self._device["fan_speed"]]
|
|
|
|
@property
|
|
def speed_list(self) -> list:
|
|
"""Get the list of available speeds."""
|
|
return FAN_SPEEDS
|
|
|
|
@property
|
|
def supported_features(self) -> int:
|
|
"""Flag supported features. Speed Only."""
|
|
return SUPPORT_SET_SPEED
|
|
|
|
async def async_turn_on(self, speed: str = None, **kwargs):
|
|
"""Turn the fan on."""
|
|
if speed is None:
|
|
speed = SPEED_MEDIUM
|
|
await self.async_set_speed(speed)
|
|
|
|
async def async_turn_off(self, **kwargs):
|
|
"""Turn the fan off."""
|
|
await self.async_set_speed(SPEED_OFF)
|
|
|
|
async def async_set_speed(self, speed: str) -> None:
|
|
"""Set the speed of the fan."""
|
|
self._smartbridge.set_fan(self.device_id, SPEED_TO_VALUE[speed])
|
|
|
|
@property
|
|
def is_on(self):
|
|
"""Return true if device is on."""
|
|
return VALUE_TO_SPEED[self._device["fan_speed"]] in [
|
|
SPEED_LOW,
|
|
SPEED_MEDIUM,
|
|
SPEED_HIGH,
|
|
]
|
|
|
|
async def async_update(self):
|
|
"""Update when forcing a refresh of the device."""
|
|
self._device = self._smartbridge.get_device_by_id(self.device_id)
|
|
_LOGGER.debug("State of this lutron fan device is %s", self._device)
|