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"
VS_SWITCHES = "switches"
VS_FAN = "fan"
VS_FANS = "fans"
VS_LIGHTS = "lights"
VS_SENSORS = "sensors"
@ -13,8 +14,12 @@ VS_NUMBERS = "numbers"
VS_BINARY_SENSORS = "binary_sensors"
VS_MANAGER = "manager"
VS_LEVELS = "levels"
VS_MODES = "modes"
VS_MODE_AUTO = "auto"
VS_MODE_MANUAL = "manual"
VS_MODE_SLEEP = "sleep"
VS_TO_HA_ATTRIBUTES = {"humidity": "current_humidity"}

46
custom_components/vesync/fan.py
View File

@ -2,7 +2,7 @@
import logging
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.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
@ -14,13 +14,22 @@ from homeassistant.util.percentage import (
)
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__)
FAN_MODE_AUTO = "auto"
FAN_MODE_SLEEP = "sleep"
async def async_setup_entry(
hass: HomeAssistant,
@ -49,7 +58,7 @@ def _setup_entities(devices, async_add_entities):
entities = []
for dev in devices:
_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))
else:
_LOGGER.warning(
@ -65,31 +74,32 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
def __init__(self, fan):
"""Initialize the VeSync fan device."""
_LOGGER.debug("Initializing fan")
super().__init__(fan)
self.smartfan = fan
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 = [
mode
for mode in ["auto", "sleep"]
if mode in self.smartfan.config_dict["modes"]
VS_MODE_MANUAL,
*[
mode
for mode in [VS_MODE_AUTO, VS_MODE_SLEEP]
if mode in self.smartfan.config_dict[VS_MODES]
],
]
else:
self._speed_range = (1, 1)
self._attr_preset_modes = []
self._attr_preset_modes = [FAN_MODE_AUTO, FAN_MODE_SLEEP]
self._attr_preset_modes = [VS_MODE_MANUAL, VS_MODE_AUTO, VS_MODE_SLEEP]
@property
def supported_features(self):
"""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
def percentage(self):
"""Return the current speed."""
if (
self.smartfan.mode == "manual"
self.smartfan.mode == VS_MODE_MANUAL
and (current_level := self.smartfan.fan_level) is not None
):
return ranged_value_to_percentage(self._speed_range, current_level)
@ -148,10 +158,12 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
if not self.smartfan.is_on:
self.smartfan.turn_on()
if preset_mode == FAN_MODE_AUTO:
if preset_mode == VS_MODE_AUTO:
self.smartfan.auto_mode()
elif preset_mode == FAN_MODE_SLEEP:
elif preset_mode == VS_MODE_SLEEP:
self.smartfan.sleep_mode()
elif preset_mode == VS_MODE_MANUAL:
self.smartfan.manual_mode()
self.schedule_update_ha_state()