Skip to content

Commit

Permalink
update to latest official pyvesync
Browse files Browse the repository at this point in the history
  • Loading branch information
HuffYk committed Jun 30, 2024
1 parent 0f2fd2f commit 077a7cf
Show file tree
Hide file tree
Showing 8 changed files with 47 additions and 33 deletions.
1 change: 1 addition & 0 deletions custom_components/vesync/binary_sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -151,6 +151,7 @@ def is_on(self) -> bool:
"""Return a value indicating whether the Humidifier's water tank is lifted."""
return self.smarthumidifier.details["water_tank_lifted"]


class VeSyncFilterOpenStateSensor(VeSyncBinarySensorEntity):
"""Filter Open Sensor."""

Expand Down
2 changes: 1 addition & 1 deletion custom_components/vesync/const.py
Original file line number Diff line number Diff line change
Expand Up @@ -29,7 +29,7 @@

VS_TO_HA_ATTRIBUTES = {"humidity": "current_humidity"}

VS_FAN_TYPES = ["VeSyncAirBypass", "VeSyncAir131", "VeSyncVital"]
VS_FAN_TYPES = ["VeSyncAirBypass", "VeSyncAir131", "VeSyncAirBaseV2"]
VS_HUMIDIFIERS_TYPES = ["VeSyncHumid200300S", "VeSyncHumid200S", "VeSyncHumid1000S"]
VS_AIRFRYER_TYPES = ["VeSyncAirFryer158"]

Expand Down
8 changes: 4 additions & 4 deletions custom_components/vesync/fan.py
Original file line number Diff line number Diff line change
Expand Up @@ -70,15 +70,15 @@ def __init__(self, fan, coordinator) -> None:
self.smartfan = fan
self._speed_range = (1, 1)
self._attr_preset_modes = [VS_MODE_MANUAL, VS_MODE_AUTO, VS_MODE_SLEEP]
if has_feature(self.smartfan, "config_dict", VS_LEVELS):
self._speed_range = (1, max(self.smartfan.config_dict[VS_LEVELS]))
if has_feature(self.smartfan, "config_dict", VS_MODES):
if has_feature(self.smartfan, "_config_dict", VS_LEVELS):
self._speed_range = (1, max(self.smartfan._config_dict[VS_LEVELS]))
if has_feature(self.smartfan, "_config_dict", VS_MODES):
self._attr_preset_modes = [
VS_MODE_MANUAL,
*[
mode
for mode in [VS_MODE_AUTO, VS_MODE_SLEEP, VS_MODE_TURBO]
if mode in self.smartfan.config_dict[VS_MODES]
if mode in self.smartfan._config_dict[VS_MODES]
],
]
if self.smartfan.device_type == "LV-PUR131S":
Expand Down
6 changes: 3 additions & 3 deletions custom_components/vesync/light.py
Original file line number Diff line number Diff line change
Expand Up @@ -15,7 +15,7 @@
from homeassistant.helpers.entity_platform import AddEntitiesCallback

from .common import VeSyncDevice, has_feature
from .const import DEV_TYPE_TO_HA, DOMAIN, VS_DISCOVERY, VS_LIGHTS, VS_FAN_TYPES
from .const import DEV_TYPE_TO_HA, DOMAIN, VS_DISCOVERY, VS_FAN_TYPES, VS_LIGHTS

_LOGGER = logging.getLogger(__name__)

Expand Down Expand Up @@ -255,7 +255,7 @@ def entity_category(self):

def turn_on(self, **kwargs):
"""Turn the night light on."""
if self.device.config_dict["module"] in VS_FAN_TYPES:
if self.device._config_dict["module"] in VS_FAN_TYPES:
if ATTR_BRIGHTNESS in kwargs and kwargs[ATTR_BRIGHTNESS] < 255:
self.device.set_night_light("dim")
else:
Expand All @@ -269,7 +269,7 @@ def turn_on(self, **kwargs):

def turn_off(self, **kwargs):
"""Turn the night light off."""
if self.device.config_dict["module"] in VS_FAN_TYPES:
if self.device._config_dict["module"] in VS_FAN_TYPES:
self.device.set_night_light("off")
else:
self.device.set_night_light_brightness(0)
4 changes: 2 additions & 2 deletions custom_components/vesync/manifest.json
Original file line number Diff line number Diff line change
Expand Up @@ -19,7 +19,7 @@
"iot_class": "cloud_polling",
"issue_tracker": "https://github.com/AndreaTomatis/custom_vesync",
"requirements": [
"pyvesync==2.1.10"
"pyvesync==2.1.12"
],
"version": "1.3.1"
"version": "1.3.2"
}
20 changes: 10 additions & 10 deletions custom_components/vesync/number.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def _setup_entities(devices, async_add_entities, coordinator):
entities.append(VeSyncHumidifierTargetLevelHA(dev, coordinator))
if has_feature(dev, "details", "warm_mist_level"):
entities.append(VeSyncHumidifierWarmthLevelHA(dev, coordinator))
if has_feature(dev, "config_dict", "levels"):
if has_feature(dev, "_config_dict", "levels"):
entities.append(VeSyncFanSpeedLevelHA(dev, coordinator))

async_add_entities(entities, update_before_add=True)
Expand All @@ -77,8 +77,8 @@ class VeSyncFanSpeedLevelHA(VeSyncNumberEntity):
def __init__(self, device, coordinator) -> None:
"""Initialize the number entity."""
super().__init__(device, coordinator)
self._attr_native_min_value = device.config_dict["levels"][0]
self._attr_native_max_value = device.config_dict["levels"][-1]
self._attr_native_min_value = device._config_dict["levels"][0]
self._attr_native_max_value = device._config_dict["levels"][-1]
self._attr_native_step = 1

@property
Expand All @@ -99,7 +99,7 @@ def native_value(self):
@property
def extra_state_attributes(self):
"""Return the state attributes of the humidifier."""
return {"fan speed levels": self.device.config_dict["levels"]}
return {"fan speed levels": self.device._config_dict["levels"]}

def set_native_value(self, value):
"""Set the fan speed level."""
Expand All @@ -112,8 +112,8 @@ class VeSyncHumidifierMistLevelHA(VeSyncNumberEntity):
def __init__(self, device, coordinator) -> None:
"""Initialize the number entity."""
super().__init__(device, coordinator)
self._attr_native_min_value = device.config_dict["mist_levels"][0]
self._attr_native_max_value = device.config_dict["mist_levels"][-1]
self._attr_native_min_value = device._config_dict["mist_levels"][0]
self._attr_native_max_value = device._config_dict["mist_levels"][-1]
self._attr_native_step = 1

@property
Expand All @@ -134,7 +134,7 @@ def native_value(self):
@property
def extra_state_attributes(self):
"""Return the state attributes of the humidifier."""
return {"mist levels": self.device.config_dict["mist_levels"]}
return {"mist levels": self.device._config_dict["mist_levels"]}

def set_native_value(self, value):
"""Set the mist level."""
Expand All @@ -147,8 +147,8 @@ class VeSyncHumidifierWarmthLevelHA(VeSyncNumberEntity):
def __init__(self, device, coordinator) -> None:
"""Initialize the number entity."""
super().__init__(device, coordinator)
self._attr_native_min_value = device.config_dict["warm_mist_levels"][0]
self._attr_native_max_value = device.config_dict["warm_mist_levels"][-1]
self._attr_native_min_value = device._config_dict["warm_mist_levels"][0]
self._attr_native_max_value = device._config_dict["warm_mist_levels"][-1]
self._attr_native_step = 1

@property
Expand All @@ -169,7 +169,7 @@ def native_value(self):
@property
def extra_state_attributes(self):
"""Return the state attributes of the humidifier."""
return {"warm mist levels": self.device.config_dict["warm_mist_levels"]}
return {"warm mist levels": self.device._config_dict["warm_mist_levels"]}

def set_native_value(self, value):
"""Set the mist level."""
Expand Down
37 changes: 25 additions & 12 deletions custom_components/vesync/sensor.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,13 @@
SensorStateClass,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfEnergy, UnitOfPower, CONCENTRATION_MICROGRAMS_PER_CUBIC_METER, DEGREE
from homeassistant.const import (
CONCENTRATION_MICROGRAMS_PER_CUBIC_METER,
DEGREE,
PERCENTAGE,
UnitOfEnergy,
UnitOfPower,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
from homeassistant.helpers.entity import EntityCategory
Expand Down Expand Up @@ -269,12 +275,12 @@ def device_class(self):
@property
def unique_id(self):
"""Return unique ID for air quality sensor on device."""
return f"{super().unique_id}-air-quality-index"
return f"{super().unique_id}-air-quality"

@property
def name(self):
"""Return sensor name."""
return f"{super().name} air quality index"
return f"{super().name} air quality"

@property
def native_value(self):
Expand All @@ -291,6 +297,7 @@ def native_value(self):
_LOGGER.warning("No air quality index found in '%s'", self.name)
return None


class VeSyncAirQualityPercSensor(VeSyncHumidifierSensorEntity):
"""Representation of an air quality percentage sensor."""

Expand All @@ -306,12 +313,12 @@ def __init__(self, device, coordinator) -> None:
@property
def unique_id(self):
"""Return unique ID for air quality sensor on device."""
return f"{super().unique_id}-air-quality"
return f"{super().unique_id}-air-quality-perc"

@property
def name(self):
"""Return sensor name."""
return f"{super().name} air quality"
return f"{super().name} air quality percentage"

@property
def native_unit_of_measurement(self):
Expand All @@ -333,12 +340,13 @@ def native_value(self):
_LOGGER.warning("No air quality percentage found in '%s'", self.name)
return None


class VeSyncAirQualityValueSensor(VeSyncHumidifierSensorEntity):
"""Representation of an air quality sensor."""

_attr_state_class = SensorStateClass.MEASUREMENT
_attr_device_class = SensorDeviceClass.PM25
_attr_native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
_attr_native_unit_of_measurement = CONCENTRATION_MICROGRAMS_PER_CUBIC_METER

def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
Expand All @@ -347,12 +355,12 @@ def __init__(self, device, coordinator) -> None:
@property
def unique_id(self):
"""Return unique ID for air quality sensor on device."""
return f"{super().unique_id}-pm25"
return f"{super().unique_id}-air-quality-value"

@property
def name(self):
"""Return sensor name."""
return f"{super().name} PM2.5"
return f"{super().name} air quality value"

@property
def native_value(self):
Expand All @@ -369,12 +377,13 @@ def native_value(self):
_LOGGER.warning("No air quality value found in '%s'", self.name)
return None


class VeSyncPM1Sensor(VeSyncHumidifierSensorEntity):
"""Representation of a PM1 sensor."""

_attr_state_class = SensorStateClass.MEASUREMENT
_attr_device_class = SensorDeviceClass.PM1
_attr_native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
_attr_native_unit_of_measurement = CONCENTRATION_MICROGRAMS_PER_CUBIC_METER

def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
Expand All @@ -392,7 +401,7 @@ def name(self):

@property
def native_value(self):
"""Return the PM1"""
"""Return the PM1."""
if has_feature(self.smarthumidifier, "details", "pm1"):
quality_value = self.smarthumidifier.details["pm1"]
if isinstance(quality_value, (int, float)):
Expand All @@ -405,12 +414,13 @@ def native_value(self):
_LOGGER.warning("No PM1 value found in '%s'", self.name)
return None


class VeSyncPM10Sensor(VeSyncHumidifierSensorEntity):
"""Representation of a PM10 sensor."""

_attr_state_class = SensorStateClass.MEASUREMENT
_attr_device_class = SensorDeviceClass.PM10
_attr_native_unit_of_measurement=CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
_attr_native_unit_of_measurement = CONCENTRATION_MICROGRAMS_PER_CUBIC_METER

def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
Expand All @@ -428,7 +438,7 @@ def name(self):

@property
def native_value(self):
"""Return the PM10"""
"""Return the PM10."""
if has_feature(self.smarthumidifier, "details", "pm10"):
quality_value = self.smarthumidifier.details["pm10"]
if isinstance(quality_value, (int, float)):
Expand All @@ -441,6 +451,7 @@ def native_value(self):
_LOGGER.warning("No PM10 value found in '%s'", self.name)
return None


class VeSyncFilterLifeSensor(VeSyncHumidifierSensorEntity):
"""Representation of a filter life sensor."""

Expand Down Expand Up @@ -501,6 +512,7 @@ def icon(self):
"""Return the icon to use in the frontend, if any."""
return "mdi:air-filter"


class VeSyncFanRotateAngleSensor(VeSyncHumidifierSensorEntity):
"""Representation of a fan rotate angle sensor."""

Expand Down Expand Up @@ -552,6 +564,7 @@ def icon(self):
"""Return the icon to use in the frontend, if any."""
return "mdi:rotate-3d-variant"


class VeSyncHumiditySensor(VeSyncHumidifierSensorEntity):
"""Representation of current humidity for a VeSync humidifier."""

Expand Down
2 changes: 1 addition & 1 deletion requirements.txt
Original file line number Diff line number Diff line change
@@ -1 +1 @@
pyvesync==2.1.10
pyvesync==2.1.12

0 comments on commit 077a7cf

Please sign in to comment.