Home-Assistant
/
core
mirror of https://github.com/home-assistant/core.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Use new enums in sma (#62439) 

Co-authored-by: epenet <epenet@users.noreply.github.com>
pull/62450/head
epenet 2021-12-20 22:30:22 +01:00 committed by GitHub
parent f913961d63
commit b483754ad3
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
homeassistant/components/sma/sensor.py
View File

@ -9,9 +9,9 @@ import voluptuous as vol
from homeassistant.components.sensor import ( from homeassistant.components.sensor import (
PLATFORM_SCHEMA, PLATFORM_SCHEMA,
STATE_CLASS_MEASUREMENT, SensorDeviceClass,
STATE_CLASS_TOTAL_INCREASING,
SensorEntity, SensorEntity,
SensorStateClass,
) )
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
from homeassistant.const import ( from homeassistant.const import (
@ -21,8 +21,6 @@ from homeassistant.const import (
CONF_SENSORS, CONF_SENSORS,
CONF_SSL, CONF_SSL,
CONF_VERIFY_SSL, CONF_VERIFY_SSL,
DEVICE_CLASS_ENERGY,
DEVICE_CLASS_POWER,
ENERGY_KILO_WATT_HOUR, ENERGY_KILO_WATT_HOUR,
POWER_WATT, POWER_WATT,
) )
@ -167,11 +165,11 @@ class SMAsensor(CoordinatorEntity, SensorEntity):
self._attr_device_info = device_info self._attr_device_info = device_info
if self.native_unit_of_measurement == ENERGY_KILO_WATT_HOUR: if self.native_unit_of_measurement == ENERGY_KILO_WATT_HOUR:
self._attr_state_class = STATE_CLASS_TOTAL_INCREASING self._attr_state_class = SensorStateClass.TOTAL_INCREASING
self._attr_device_class = DEVICE_CLASS_ENERGY self._attr_device_class = SensorDeviceClass.ENERGY
if self.native_unit_of_measurement == POWER_WATT: if self.native_unit_of_measurement == POWER_WATT:
self._attr_state_class = STATE_CLASS_MEASUREMENT self._attr_state_class = SensorStateClass.MEASUREMENT
self._attr_device_class = DEVICE_CLASS_POWER self._attr_device_class = SensorDeviceClass.POWER
# Set sensor enabled to False. # Set sensor enabled to False.
# Will be enabled by async_added_to_hass if actually used. # Will be enabled by async_added_to_hass if actually used.