Improved code.

This commit is contained in:
AndreaTomatis 2024-04-20 16:27:28 +02:00
parent f10450c8a3
commit f123f0bb77
5 changed files with 14 additions and 11 deletions

View File

@ -1,4 +1,5 @@
"""VeSync integration.""" """VeSync integration."""
from datetime import timedelta from datetime import timedelta
import logging import logging
@ -69,7 +70,7 @@ async def async_setup_entry(hass: HomeAssistant, config_entry: ConfigEntry) -> b
try: try:
await hass.async_add_executor_job(manager.update) await hass.async_add_executor_job(manager.update)
except Exception as err: except Exception as err:
raise UpdateFailed(f"Update failed: {err}") raise UpdateFailed(f"Update failed: {err}") from err
coordinator = DataUpdateCoordinator( coordinator = DataUpdateCoordinator(
hass, hass,

View File

@ -1,4 +1,5 @@
"""Support for power & energy sensors for VeSync outlets.""" """Support for power & energy sensors for VeSync outlets."""
import logging import logging
from homeassistant.components.binary_sensor import BinarySensorEntity from homeassistant.components.binary_sensor import BinarySensorEntity
@ -46,7 +47,7 @@ def _setup_entities(devices, async_add_entities, coordinator):
for dev in devices: for dev in devices:
if hasattr(dev, "fryer_status"): if hasattr(dev, "fryer_status"):
for stype in BINARY_SENSOR_TYPES_AIRFRYER.values(): for stype in BINARY_SENSOR_TYPES_AIRFRYER.values():
entities.append( entities.append( # noqa: PERF401
VeSyncairfryerSensor( VeSyncairfryerSensor(
dev, dev,
coordinator, coordinator,
@ -88,8 +89,7 @@ class VeSyncairfryerSensor(VeSyncBaseEntity, BinarySensorEntity):
@property @property
def is_on(self) -> bool: def is_on(self) -> bool:
"""Return a value indicating whether the Humidifier's water tank is lifted.""" """Return a value indicating whether the Humidifier's water tank is lifted."""
value = getattr(self.airfryer, self.stype[0], None) return getattr(self.airfryer, self.stype[0], None)
return value
# return self.smarthumidifier.details["water_tank_lifted"] # return self.smarthumidifier.details["water_tank_lifted"]
@property @property

View File

@ -1,4 +1,5 @@
"""Support for VeSync button.""" """Support for VeSync button."""
import logging import logging
from homeassistant.components.button import ButtonEntity from homeassistant.components.button import ButtonEntity
@ -55,7 +56,7 @@ def _setup_entities(devices, async_add_entities, coordinator):
for dev in devices: for dev in devices:
if hasattr(dev, "cook_set_temp"): if hasattr(dev, "cook_set_temp"):
for stype in SENSOR_TYPES_CS158.values(): for stype in SENSOR_TYPES_CS158.values():
entities.append( entities.append( # noqa: PERF401
VeSyncairfryerButton( VeSyncairfryerButton(
dev, dev,
coordinator, coordinator,

View File

@ -1,4 +1,5 @@
"""Support for VeSync fans.""" """Support for VeSync fans."""
import math import math
from homeassistant.components.fan import FanEntity, FanEntityFeature from homeassistant.components.fan import FanEntity, FanEntityFeature
@ -165,9 +166,9 @@ class VeSyncFanHA(VeSyncDevice, FanEntity):
def turn_on( def turn_on(
self, self,
speed: str = None, # speed: str | None = None,
percentage: int = None, percentage: int | None = None,
preset_mode: str = None, preset_mode: str | None = None,
**kwargs, **kwargs,
) -> None: ) -> None:
"""Turn the device on.""" """Turn the device on."""

View File

@ -1,4 +1,5 @@
"""Support for power & energy sensors for VeSync outlets.""" """Support for power & energy sensors for VeSync outlets."""
import logging import logging
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
@ -57,7 +58,7 @@ def _setup_entities(devices, async_add_entities, coordinator):
for dev in devices: for dev in devices:
if hasattr(dev, "fryer_status"): if hasattr(dev, "fryer_status"):
for stype in SENSOR_TYPES_AIRFRYER.values(): for stype in SENSOR_TYPES_AIRFRYER.values():
entities.append( entities.append( # noqa: PERF401
VeSyncairfryerSensor( VeSyncairfryerSensor(
dev, dev,
coordinator, coordinator,
@ -111,8 +112,7 @@ class VeSyncairfryerSensor(VeSyncBaseEntity, SensorEntity):
@property @property
def native_value(self): def native_value(self):
"""Return the value.""" """Return the value."""
value = getattr(self.airfryer, self.stype[5], None) return getattr(self.airfryer, self.stype[5], None)
return value
@property @property
def native_unit_of_measurement(self): def native_unit_of_measurement(self):