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

View File

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

View File

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

View File

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

View File

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