86 lines
2.4 KiB
Python
86 lines
2.4 KiB
Python
"""Support for INSTEON fans via PowerLinc Modem."""
|
|
import logging
|
|
|
|
from homeassistant.components.fan import (
|
|
SPEED_HIGH, SPEED_LOW, SPEED_MEDIUM, SPEED_OFF, SUPPORT_SET_SPEED,
|
|
FanEntity)
|
|
from homeassistant.const import STATE_OFF
|
|
|
|
from . import InsteonEntity
|
|
|
|
_LOGGER = logging.getLogger(__name__)
|
|
|
|
SPEED_TO_HEX = {
|
|
SPEED_OFF: 0x00,
|
|
SPEED_LOW: 0x3f,
|
|
SPEED_MEDIUM: 0xbe,
|
|
SPEED_HIGH: 0xff,
|
|
}
|
|
|
|
FAN_SPEEDS = [STATE_OFF, SPEED_LOW, SPEED_MEDIUM, SPEED_HIGH]
|
|
|
|
|
|
async def async_setup_platform(
|
|
hass, config, async_add_entities, discovery_info=None):
|
|
"""Set up the INSTEON device class for the hass platform."""
|
|
insteon_modem = hass.data['insteon'].get('modem')
|
|
|
|
address = discovery_info['address']
|
|
device = insteon_modem.devices[address]
|
|
state_key = discovery_info['state_key']
|
|
|
|
_LOGGER.debug('Adding device %s entity %s to Fan platform',
|
|
device.address.hex, device.states[state_key].name)
|
|
|
|
new_entity = InsteonFan(device, state_key)
|
|
|
|
async_add_entities([new_entity])
|
|
|
|
|
|
class InsteonFan(InsteonEntity, FanEntity):
|
|
"""An INSTEON fan component."""
|
|
|
|
@property
|
|
def speed(self) -> str:
|
|
"""Return the current speed."""
|
|
return self._hex_to_speed(self._insteon_device_state.value)
|
|
|
|
@property
|
|
def speed_list(self) -> list:
|
|
"""Get the list of available speeds."""
|
|
return FAN_SPEEDS
|
|
|
|
@property
|
|
def supported_features(self) -> int:
|
|
"""Flag supported features."""
|
|
return SUPPORT_SET_SPEED
|
|
|
|
async def async_turn_on(self, speed: str = None, **kwargs) -> None:
|
|
"""Turn on the entity."""
|
|
if speed is None:
|
|
speed = SPEED_MEDIUM
|
|
await self.async_set_speed(speed)
|
|
|
|
async def async_turn_off(self, **kwargs) -> None:
|
|
"""Turn off the entity."""
|
|
await self.async_set_speed(SPEED_OFF)
|
|
|
|
async def async_set_speed(self, speed: str) -> None:
|
|
"""Set the speed of the fan."""
|
|
fan_speed = SPEED_TO_HEX[speed]
|
|
if fan_speed == 0x00:
|
|
self._insteon_device_state.off()
|
|
else:
|
|
self._insteon_device_state.set_level(fan_speed)
|
|
|
|
@staticmethod
|
|
def _hex_to_speed(speed: int):
|
|
hex_speed = SPEED_OFF
|
|
if speed > 0xfe:
|
|
hex_speed = SPEED_HIGH
|
|
elif speed > 0x7f:
|
|
hex_speed = SPEED_MEDIUM
|
|
elif speed > 0:
|
|
hex_speed = SPEED_LOW
|
|
return hex_speed
|