jeans/core
jeans/core
1
0
Fork 0
mirror of https://github.com/home-assistant/core.git synced 2025-07-23 21:27:38 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix VeSync device to match pyvesync type (#73034)

Browse Source 
* vesync: change device to match pyvesync type

* MartinHjelmare's suggestion for derived classes

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* MartinHjelmare's suggestion for derived classes

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* MartinHjelmare's suggestion for derived classes

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>

* MartinHjelmare's suggestion for annotations

* vesync: fix imports

Co-authored-by: Martin Hjelmare <marhje52@gmail.com>
This commit is contained in:
b3nj1 2022-06-08 07:43:24 -07:00 committed by GitHub
parent e74c711ef3
commit 0dc1e7d1e6
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

22
homeassistant/components/vesync/sensor.py
View File

@ -1,8 +1,14 @@
"""Support for power & energy sensors for VeSync outlets.""" """Support for power & energy sensors for VeSync outlets."""
from __future__ import annotations
from collections.abc import Callable from collections.abc import Callable
from dataclasses import dataclass from dataclasses import dataclass
import logging import logging
from pyvesync.vesyncfan import VeSyncAirBypass
from pyvesync.vesyncoutlet import VeSyncOutlet
from pyvesync.vesyncswitch import VeSyncSwitch
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
SensorDeviceClass, SensorDeviceClass,
SensorEntity, SensorEntity,
@ -22,7 +28,7 @@ from homeassistant.helpers.entity import EntityCategory
from homeassistant.helpers.entity_platform import AddEntitiesCallback from homeassistant.helpers.entity_platform import AddEntitiesCallback
from homeassistant.helpers.typing import StateType from homeassistant.helpers.typing import StateType
from .common import VeSyncBaseEntity, VeSyncDevice from .common import VeSyncBaseEntity
from .const import DEV_TYPE_TO_HA, DOMAIN, SKU_TO_BASE_DEVICE, VS_DISCOVERY, VS_SENSORS from .const import DEV_TYPE_TO_HA, DOMAIN, SKU_TO_BASE_DEVICE, VS_DISCOVERY, VS_SENSORS
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -32,7 +38,7 @@ _LOGGER = logging.getLogger(__name__)
class VeSyncSensorEntityDescriptionMixin: class VeSyncSensorEntityDescriptionMixin:
"""Mixin for required keys.""" """Mixin for required keys."""
value_fn: Callable[[VeSyncDevice], StateType] value_fn: Callable[[VeSyncAirBypass | VeSyncOutlet | VeSyncSwitch], StateType]
@dataclass @dataclass
@ -41,8 +47,12 @@ class VeSyncSensorEntityDescription(
): ):
"""Describe VeSync sensor entity.""" """Describe VeSync sensor entity."""
exists_fn: Callable[[VeSyncDevice], bool] = lambda _: True exists_fn: Callable[
update_fn: Callable[[VeSyncDevice], None] = lambda _: None [VeSyncAirBypass | VeSyncOutlet | VeSyncSwitch], bool
] = lambda _: True
update_fn: Callable[
[VeSyncAirBypass | VeSyncOutlet | VeSyncSwitch], None
] = lambda _: None
def update_energy(device): def update_energy(device):
@ -107,7 +117,7 @@ SENSORS: tuple[VeSyncSensorEntityDescription, ...] = (
device_class=SensorDeviceClass.ENERGY, device_class=SensorDeviceClass.ENERGY,
native_unit_of_measurement=ENERGY_KILO_WATT_HOUR, native_unit_of_measurement=ENERGY_KILO_WATT_HOUR,
state_class=SensorStateClass.TOTAL_INCREASING, state_class=SensorStateClass.TOTAL_INCREASING,
value_fn=lambda device: device.details["energy"], value_fn=lambda device: device.energy_today,
update_fn=update_energy, update_fn=update_energy,
exists_fn=lambda device: ha_dev_type(device) == "outlet", exists_fn=lambda device: ha_dev_type(device) == "outlet",
), ),
@ -151,7 +161,7 @@ class VeSyncSensorEntity(VeSyncBaseEntity, SensorEntity):
def __init__( def __init__(
self, self,
device: VeSyncDevice, device: VeSyncAirBypass | VeSyncOutlet | VeSyncSwitch,
description: VeSyncSensorEntityDescription, description: VeSyncSensorEntityDescription,
) -> None: ) -> None:
"""Initialize the VeSync outlet device.""" """Initialize the VeSync outlet device."""