mirror/custom_vesync
1
0
Fork 0
mirror of https://github.com/micahqcade/custom_vesync.git synced 2025-02-11 09:39:00 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Add constants to fan.py (#34)

Browse Source
This commit is contained in:
Vincent Le Bourlot 2022-07-30 10:00:37 +02:00 committed by GitHub
parent 84ad318e35
commit a6015cd12e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 34 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
custom_components/vesync/const.py
View File

@ -5,6 +5,7 @@ VS_DISCOVERY = "vesync_discovery_{}"
SERVICE_UPDATE_DEVS = "update_devices" SERVICE_UPDATE_DEVS = "update_devices"
VS_SWITCHES = "switches" VS_SWITCHES = "switches"
VS_FAN = "fan"
VS_FANS = "fans" VS_FANS = "fans"
VS_LIGHTS = "lights" VS_LIGHTS = "lights"
VS_SENSORS = "sensors" VS_SENSORS = "sensors"
@ -13,8 +14,12 @@ VS_NUMBERS = "numbers"
VS_BINARY_SENSORS = "binary_sensors" VS_BINARY_SENSORS = "binary_sensors"
VS_MANAGER = "manager" VS_MANAGER = "manager"
VS_LEVELS = "levels"
VS_MODES = "modes"
VS_MODE_AUTO = "auto" VS_MODE_AUTO = "auto"
VS_MODE_MANUAL = "manual" VS_MODE_MANUAL = "manual"
VS_MODE_SLEEP = "sleep"
VS_TO_HA_ATTRIBUTES = {"humidity": "current_humidity"} VS_TO_HA_ATTRIBUTES = {"humidity": "current_humidity"}

46
custom_components/vesync/fan.py
View File

@ -2,7 +2,7 @@
import logging import logging
import math import math
from homeassistant.components.fan import SUPPORT_SET_SPEED, FanEntity from homeassistant.components.fan import FanEntity, FanEntityFeature
from homeassistant.config_entries import ConfigEntry from homeassistant.config_entries import ConfigEntry
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect from homeassistant.helpers.dispatcher import async_dispatcher_connect
@ -14,13 +14,22 @@ from homeassistant.util.percentage import (
) )
from .common import VeSyncDevice from .common import VeSyncDevice
from .const import DEV_TYPE_TO_HA, DOMAIN, VS_DISCOVERY, VS_FANS, VS_TO_HA_ATTRIBUTES from .const import (
DEV_TYPE_TO_HA,
DOMAIN,
VS_DISCOVERY,
VS_FAN,
VS_FANS,
VS_LEVELS,
VS_MODE_AUTO,
VS_MODE_MANUAL,
VS_MODE_SLEEP,
VS_MODES,
VS_TO_HA_ATTRIBUTES,
)
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
FAN_MODE_AUTO = "auto"
FAN_MODE_SLEEP = "sleep"
async def async_setup_entry( async def async_setup_entry(
hass: HomeAssistant, hass: HomeAssistant,
@ -49,7 +58,7 @@ def _setup_entities(devices, async_add_entities):
entities = [] entities = []
for dev in devices: for dev in devices:
_LOGGER.debug("Adding device %s %s", dev.device_name, dev.device_type) _LOGGER.debug("Adding device %s %s", dev.device_name, dev.device_type)
if DEV_TYPE_TO_HA.get(dev.device_type) == "fan": if DEV_TYPE_TO_HA.get(dev.device_type) == VS_FAN:
entities.append(VeSyncFanHA(dev)) entities.append(VeSyncFanHA(dev))
else: else:
_LOGGER.warning( _LOGGER.warning(
@ -65,31 +74,32 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
def __init__(self, fan): def __init__(self, fan):
"""Initialize the VeSync fan device.""" """Initialize the VeSync fan device."""
_LOGGER.debug("Initializing fan")
super().__init__(fan) super().__init__(fan)
self.smartfan = fan self.smartfan = fan
if hasattr(self.smartfan, "config_dict"): if hasattr(self.smartfan, "config_dict"):
self._speed_range = (1, max(self.smartfan.config_dict["levels"])) self._speed_range = (1, max(self.smartfan.config_dict[VS_LEVELS]))
self._attr_preset_modes = [ self._attr_preset_modes = [
mode VS_MODE_MANUAL,
for mode in ["auto", "sleep"] *[
if mode in self.smartfan.config_dict["modes"] mode
for mode in [VS_MODE_AUTO, VS_MODE_SLEEP]
if mode in self.smartfan.config_dict[VS_MODES]
],
] ]
else: else:
self._speed_range = (1, 1) self._speed_range = (1, 1)
self._attr_preset_modes = [] self._attr_preset_modes = [VS_MODE_MANUAL, VS_MODE_AUTO, VS_MODE_SLEEP]
self._attr_preset_modes = [FAN_MODE_AUTO, FAN_MODE_SLEEP]
@property @property
def supported_features(self): def supported_features(self):
"""Flag supported features.""" """Flag supported features."""
return SUPPORT_SET_SPEED if self.speed_count > 1 else 0 return FanEntityFeature.PRESET_MODE if self.speed_count > 1 else 0
@property @property
def percentage(self): def percentage(self):
"""Return the current speed.""" """Return the current speed."""
if ( if (
self.smartfan.mode == "manual" self.smartfan.mode == VS_MODE_MANUAL
and (current_level := self.smartfan.fan_level) is not None and (current_level := self.smartfan.fan_level) is not None
): ):
return ranged_value_to_percentage(self._speed_range, current_level) return ranged_value_to_percentage(self._speed_range, current_level)
@ -148,10 +158,12 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
if not self.smartfan.is_on: if not self.smartfan.is_on:
self.smartfan.turn_on() self.smartfan.turn_on()
if preset_mode == FAN_MODE_AUTO: if preset_mode == VS_MODE_AUTO:
self.smartfan.auto_mode() self.smartfan.auto_mode()
elif preset_mode == FAN_MODE_SLEEP: elif preset_mode == VS_MODE_SLEEP:
self.smartfan.sleep_mode() self.smartfan.sleep_mode()
elif preset_mode == VS_MODE_MANUAL:
self.smartfan.manual_mode()
self.schedule_update_ha_state() self.schedule_update_ha_state()