Use device classes in sense (#83817)

This commit is contained in:
epenet 2022-12-12 11:42:57 +01:00 committed by GitHub
parent 3edb0969de
commit a48741d891
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 9 additions and 13 deletions

View file

@ -39,8 +39,6 @@ FROM_GRID_ID = "from_grid"
SOLAR_POWERED_NAME = "Solar Powered Percentage"
SOLAR_POWERED_ID = "solar_powered"
ICON = "mdi:flash"
SENSE_TIMEOUT_EXCEPTIONS = (asyncio.TimeoutError, SenseAPITimeoutException)
SENSE_EXCEPTIONS = (socket.gaierror, SenseWebsocketException)
SENSE_CONNECT_EXCEPTIONS = (

View file

@ -7,10 +7,10 @@ from homeassistant.components.sensor import (
)
from homeassistant.config_entries import ConfigEntry
from homeassistant.const import (
ELECTRIC_POTENTIAL_VOLT,
ENERGY_KILO_WATT_HOUR,
PERCENTAGE,
POWER_WATT,
UnitOfElectricPotential,
UnitOfEnergy,
UnitOfPower,
)
from homeassistant.core import HomeAssistant, callback
from homeassistant.helpers.dispatcher import async_dispatcher_connect
@ -27,7 +27,6 @@ from .const import (
DOMAIN,
FROM_GRID_ID,
FROM_GRID_NAME,
ICON,
MDI_ICONS,
NET_PRODUCTION_ID,
NET_PRODUCTION_NAME,
@ -156,8 +155,8 @@ async def async_setup_entry(
class SenseActiveSensor(SensorEntity):
"""Implementation of a Sense energy sensor."""
_attr_icon = ICON
_attr_native_unit_of_measurement = POWER_WATT
_attr_device_class = SensorDeviceClass.POWER
_attr_native_unit_of_measurement = UnitOfPower.WATT
_attr_attribution = ATTRIBUTION
_attr_should_poll = False
_attr_available = False
@ -210,9 +209,9 @@ class SenseActiveSensor(SensorEntity):
class SenseVoltageSensor(SensorEntity):
"""Implementation of a Sense energy voltage sensor."""
_attr_native_unit_of_measurement = ELECTRIC_POTENTIAL_VOLT
_attr_device_class = SensorDeviceClass.VOLTAGE
_attr_native_unit_of_measurement = UnitOfElectricPotential.VOLT
_attr_attribution = ATTRIBUTION
_attr_icon = ICON
_attr_should_poll = False
_attr_available = False
@ -256,9 +255,8 @@ class SenseTrendsSensor(CoordinatorEntity, SensorEntity):
_attr_device_class = SensorDeviceClass.ENERGY
_attr_state_class = SensorStateClass.TOTAL
_attr_native_unit_of_measurement = ENERGY_KILO_WATT_HOUR
_attr_native_unit_of_measurement = UnitOfEnergy.KILO_WATT_HOUR
_attr_attribution = ATTRIBUTION
_attr_icon = ICON
_attr_should_poll = False
def __init__(
@ -309,7 +307,7 @@ class SenseEnergyDevice(SensorEntity):
_attr_available = False
_attr_state_class = SensorStateClass.MEASUREMENT
_attr_native_unit_of_measurement = POWER_WATT
_attr_native_unit_of_measurement = UnitOfPower.WATT
_attr_attribution = ATTRIBUTION
_attr_device_class = SensorDeviceClass.POWER
_attr_should_poll = False