Use SensorDeviceClass in greeneye-monitor (#61653)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-13 17:36:51 +01:00 committed by GitHub
parent 88a93d5d53
commit 00b90dff55
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,15 +5,12 @@ from typing import Any, Optional, Union, cast
import greeneye import greeneye
from homeassistant.components.sensor import SensorEntity from homeassistant.components.sensor import SensorDeviceClass, SensorEntity
from homeassistant.const import ( from homeassistant.const import (
CONF_NAME, CONF_NAME,
CONF_SENSOR_TYPE, CONF_SENSOR_TYPE,
CONF_SENSORS, CONF_SENSORS,
CONF_TEMPERATURE_UNIT, CONF_TEMPERATURE_UNIT,
DEVICE_CLASS_POWER,
DEVICE_CLASS_TEMPERATURE,
DEVICE_CLASS_VOLTAGE,
ELECTRIC_POTENTIAL_VOLT, ELECTRIC_POTENTIAL_VOLT,
POWER_WATT, POWER_WATT,
TIME_HOURS, TIME_HOURS,
@ -162,7 +159,7 @@ class CurrentSensor(GEMSensor):
"""Entity showing power usage on one channel of the monitor.""" """Entity showing power usage on one channel of the monitor."""
_attr_native_unit_of_measurement = UNIT_WATTS _attr_native_unit_of_measurement = UNIT_WATTS
_attr_device_class = DEVICE_CLASS_POWER _attr_device_class = SensorDeviceClass.POWER
def __init__( def __init__(
self, monitor_serial_number: int, number: int, name: str, net_metering: bool self, monitor_serial_number: int, number: int, name: str, net_metering: bool
@ -261,7 +258,7 @@ class PulseCounter(GEMSensor):
class TemperatureSensor(GEMSensor): class TemperatureSensor(GEMSensor):
"""Entity showing temperature from one temperature sensor.""" """Entity showing temperature from one temperature sensor."""
_attr_device_class = DEVICE_CLASS_TEMPERATURE _attr_device_class = SensorDeviceClass.TEMPERATURE
def __init__( def __init__(
self, monitor_serial_number: int, number: int, name: str, unit: str self, monitor_serial_number: int, number: int, name: str, unit: str
@ -291,7 +288,7 @@ class VoltageSensor(GEMSensor):
"""Entity showing voltage.""" """Entity showing voltage."""
_attr_native_unit_of_measurement = ELECTRIC_POTENTIAL_VOLT _attr_native_unit_of_measurement = ELECTRIC_POTENTIAL_VOLT
_attr_device_class = DEVICE_CLASS_VOLTAGE _attr_device_class = SensorDeviceClass.VOLTAGE
def __init__(self, monitor_serial_number: int, number: int, name: str) -> None: def __init__(self, monitor_serial_number: int, number: int, name: str) -> None:
"""Construct the entity.""" """Construct the entity."""