mirror/custom_vesync
1
0
Fork 0
mirror of https://github.com/micahqcade/custom_vesync.git synced 2025-03-12 15:50:33 +01:00
Code Issues Actions Packages Projects Releases Wiki Activity

Merge pull request #5 from HuffYk/main

Browse Source
This commit is contained in:
micahqcade 2024-06-30 09:26:26 -07:00 committed by GitHub
commit 06b06f3711
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
9 changed files with 248 additions and 31 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
custom_components/vesync/binary_sensor.py
View File

@ -58,6 +58,8 @@ def _setup_entities(devices, async_add_entities, coordinator):
entities.append(VeSyncOutOfWaterSensor(dev, coordinator))
if has_feature(dev, "details", "water_tank_lifted"):
entities.append(VeSyncWaterTankLiftedSensor(dev, coordinator))
if has_feature(dev, "details", "filter_open_state"):
entities.append(VeSyncFilterOpenStateSensor(dev, coordinator))
async_add_entities(entities, update_before_add=True)
@ -148,3 +150,22 @@ class VeSyncWaterTankLiftedSensor(VeSyncBinarySensorEntity):
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."""
@property
def unique_id(self):
"""Return unique ID for filter open state sensor on device."""
return f"{super().unique_id}-filter-open-state"
@property
def name(self):
"""Return sensor name."""
return f"{super().name} filter open state"
@property
def is_on(self) -> bool:
"""Return a value indicating whether the Humidifier's filter is open."""
return self.smarthumidifier.details["filter_open_state"]

2
custom_components/vesync/common.py
View File

@ -146,7 +146,7 @@ class VeSyncBaseEntity(CoordinatorEntity, Entity):
@property
def base_name(self):
"""Return the name of the device."""
return self.device.device_type
return self.device.device_name
@property
def name(self):

3
custom_components/vesync/const.py
View File

@ -25,10 +25,11 @@ VS_MODE_AUTO = "auto"
VS_MODE_HUMIDITY = "humidity"
VS_MODE_MANUAL = "manual"
VS_MODE_SLEEP = "sleep"
VS_MODE_TURBO = "turbo"
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"]

13
custom_components/vesync/fan.py
View File

@ -22,6 +22,7 @@ from .const import (
VS_MODE_AUTO,
VS_MODE_MANUAL,
VS_MODE_SLEEP,
VS_MODE_TURBO,
VS_MODES,
VS_TO_HA_ATTRIBUTES,
)
@ -69,15 +70,15 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
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]
if mode in self.smartfan.config_dict[VS_MODES]
for mode in [VS_MODE_AUTO, VS_MODE_SLEEP, VS_MODE_TURBO]
if mode in self.smartfan._config_dict[VS_MODES]
],
]
if self.smartfan.device_type == "LV-PUR131S":
@ -161,6 +162,8 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
self.smartfan.sleep_mode()
elif preset_mode == VS_MODE_MANUAL:
self.smartfan.manual_mode()
elif preset_mode == VS_MODE_TURBO:
self.smartfan.turbo_mode()
self.schedule_update_ha_state()

6
custom_components/vesync/light.py
View File

@ -15,7 +15,7 @@ from homeassistant.helpers.entity import EntityCategory
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
from .const import DEV_TYPE_TO_HA, DOMAIN, VS_DISCOVERY, VS_FAN_TYPES, VS_LIGHTS
_LOGGER = logging.getLogger(__name__)
@ -255,7 +255,7 @@ class VeSyncNightLightHA(VeSyncDimmableLightHA):
def turn_on(self, **kwargs):
"""Turn the night light on."""
if self.device.config_dict["module"] == "VeSyncAirBypass":
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:
@ -269,7 +269,7 @@ class VeSyncNightLightHA(VeSyncDimmableLightHA):
def turn_off(self, **kwargs):
"""Turn the night light off."""
if self.device.config_dict["module"] == "VeSyncAirBypass":
if self.device._config_dict["module"] in VS_FAN_TYPES:
self.device.set_night_light("off")
else:
self.device.set_night_light_brightness(0)

8
custom_components/vesync/manifest.json
View File

@ -15,11 +15,11 @@
"macaddress": "*"
}
],
"documentation": "https://github.com/AndreaTomatis/custom_vesync",
"documentation": "https://github.com/micahqcade/custom_vesync",
"iot_class": "cloud_polling",
"issue_tracker": "https://github.com/AndreaTomatis/custom_vesync",
"issue_tracker": "https://github.com/micahqcade/custom_vesync",
"requirements": [
"pyvesync==2.1.10"
"pyvesync==2.1.12"
],
"version": "1.3.1"
"version": "1.3.2"
}

20
custom_components/vesync/number.py
View File

@ -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)
@ -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
@ -99,7 +99,7 @@ class VeSyncFanSpeedLevelHA(VeSyncNumberEntity):
@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."""
@ -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
@ -134,7 +134,7 @@ class VeSyncHumidifierMistLevelHA(VeSyncNumberEntity):
@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."""
@ -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
@ -169,7 +169,7 @@ class VeSyncHumidifierWarmthLevelHA(VeSyncNumberEntity):
@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."""

204
custom_components/vesync/sensor.py
View File

@ -8,7 +8,13 @@ from homeassistant.components.sensor import (
SensorStateClass,
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import PERCENTAGE, UnitOfEnergy, UnitOfPower
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
@ -77,10 +83,18 @@ def _setup_entities(devices, async_add_entities, coordinator):
entities.append(VeSyncHumiditySensor(dev, coordinator))
if has_feature(dev, "details", "air_quality"):
entities.append(VeSyncAirQualitySensor(dev, coordinator))
if has_feature(dev, "details", "aq_percent"):
entities.append(VeSyncAirQualityPercSensor(dev, coordinator))
if has_feature(dev, "details", "air_quality_value"):
entities.append(VeSyncAirQualityValueSensor(dev, coordinator))
if has_feature(dev, "details", "pm1"):
entities.append(VeSyncPM1Sensor(dev, coordinator))
if has_feature(dev, "details", "pm10"):
entities.append(VeSyncPM10Sensor(dev, coordinator))
if has_feature(dev, "details", "filter_life"):
entities.append(VeSyncFilterLifeSensor(dev, coordinator))
if has_feature(dev, "details", "fan_rotate_angle"):
entities.append(VeSyncFanRotateAngleSensor(dev, coordinator))
async_add_entities(entities, update_before_add=True)
@ -238,14 +252,13 @@ class VeSyncHumidifierSensorEntity(VeSyncBaseEntity, SensorEntity):
@property
def entity_category(self):
"""Return the diagnostic entity category."""
return EntityCategory.DIAGNOSTIC
return None
class VeSyncAirQualitySensor(VeSyncHumidifierSensorEntity):
"""Representation of an air quality sensor."""
_attr_state_class = SensorStateClass.MEASUREMENT
_attr_native_unit_of_measurement = " "
def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
@ -285,12 +298,55 @@ class VeSyncAirQualitySensor(VeSyncHumidifierSensorEntity):
return None
class VeSyncAirQualityPercSensor(VeSyncHumidifierSensorEntity):
"""Representation of an air quality percentage sensor."""
_attr_state_class = SensorStateClass.MEASUREMENT
def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
super().__init__(device, coordinator)
self._numeric_quality = None
if self.native_value is not None:
self._numeric_quality = isinstance(self.native_value, (int, float))
@property
def unique_id(self):
"""Return unique ID for air quality sensor on device."""
return f"{super().unique_id}-air-quality-perc"
@property
def name(self):
"""Return sensor name."""
return f"{super().name} air quality percentage"
@property
def native_unit_of_measurement(self):
"""Return the % unit of measurement."""
return PERCENTAGE
@property
def native_value(self):
"""Return the air quality percentage."""
if has_feature(self.smarthumidifier, "details", "aq_percent"):
quality = self.smarthumidifier.details["aq_percent"]
if isinstance(quality, (int, float)):
return quality
_LOGGER.warning(
"Got non numeric value for AQI sensor from 'aq_percent' for %s: %s",
self.name,
quality,
)
_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.AQI
_attr_native_unit_of_measurement = " "
_attr_device_class = SensorDeviceClass.PM25
_attr_native_unit_of_measurement = CONCENTRATION_MICROGRAMS_PER_CUBIC_METER
def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
@ -322,13 +378,92 @@ class VeSyncAirQualityValueSensor(VeSyncHumidifierSensorEntity):
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
def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
super().__init__(device, coordinator)
@property
def unique_id(self):
"""Return unique ID for PM1 sensor on device."""
return f"{super().unique_id}-pm1"
@property
def name(self):
"""Return sensor name."""
return f"{super().name} PM1"
@property
def native_value(self):
"""Return the PM1."""
if has_feature(self.smarthumidifier, "details", "pm1"):
quality_value = self.smarthumidifier.details["pm1"]
if isinstance(quality_value, (int, float)):
return quality_value
_LOGGER.warning(
"Got non numeric value for PM1 sensor from 'pm1' for %s: %s",
self.name,
quality_value,
)
_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
def __init__(self, device, coordinator) -> None:
"""Initialize the VeSync device."""
super().__init__(device, coordinator)
@property
def unique_id(self):
"""Return unique ID for PM10 sensor on device."""
return f"{super().unique_id}-pm10"
@property
def name(self):
"""Return sensor name."""
return f"{super().name} PM10"
@property
def native_value(self):
"""Return the PM10."""
if has_feature(self.smarthumidifier, "details", "pm10"):
quality_value = self.smarthumidifier.details["pm10"]
if isinstance(quality_value, (int, float)):
return quality_value
_LOGGER.warning(
"Got non numeric value for PM10 sensor from 'pm10' for %s: %s",
self.name,
quality_value,
)
_LOGGER.warning("No PM10 value found in '%s'", self.name)
return None
class VeSyncFilterLifeSensor(VeSyncHumidifierSensorEntity):
"""Representation of a filter life sensor."""
def __init__(self, plug, coordinator) -> None:
"""Initialize the VeSync outlet device."""
"""Initialize the VeSync device."""
super().__init__(plug, coordinator)
@property
def entity_category(self):
"""Return the diagnostic entity category."""
return EntityCategory.DIAGNOSTIC
@property
def unique_id(self):
"""Return unique ID for filter life sensor on device."""
@ -372,6 +507,63 @@ class VeSyncFilterLifeSensor(VeSyncHumidifierSensorEntity):
else {}
)
@property
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."""
def __init__(self, plug, coordinator) -> None:
"""Initialize the VeSync device."""
super().__init__(plug, coordinator)
@property
def entity_category(self):
"""Return the diagnostic entity category."""
return EntityCategory.DIAGNOSTIC
@property
def unique_id(self):
"""Return unique ID for filter life sensor on device."""
return f"{super().unique_id}-fan-rotate-angle"
@property
def name(self):
"""Return sensor name."""
return f"{super().name} fan rotate angle"
@property
def device_class(self):
"""Return the fan rotate angle device class."""
return None
@property
def native_value(self):
"""Return the fan rotate angle index."""
return (
self.smarthumidifier.fan_rotate_angle
if hasattr(self.smarthumidifier, "fan_rotate_angle")
else self.smarthumidifier.details["fan_rotate_angle"]
)
@property
def native_unit_of_measurement(self):
"""Return the % unit of measurement."""
return DEGREE
@property
def state_class(self):
"""Return the measurement state class."""
return SensorStateClass.MEASUREMENT
@property
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."""

2
requirements.txt
View File

@ -1 +1 @@
pyvesync==2.1.10
pyvesync==2.1.12