Use SensorDeviceClass enum in beewi_smartclim (#61313)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-09 08:29:29 +01:00 committed by GitHub
parent c7eae8b0bc
commit cf440f2452
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -2,16 +2,12 @@
from beewi_smartclim import BeewiSmartClimPoller # pylint: disable=import-error from beewi_smartclim import BeewiSmartClimPoller # pylint: disable=import-error
import voluptuous as vol import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity from homeassistant.components.sensor import (
from homeassistant.const import ( PLATFORM_SCHEMA,
CONF_MAC, SensorDeviceClass,
CONF_NAME, SensorEntity,
DEVICE_CLASS_BATTERY,
DEVICE_CLASS_HUMIDITY,
DEVICE_CLASS_TEMPERATURE,
PERCENTAGE,
TEMP_CELSIUS,
) )
from homeassistant.const import CONF_MAC, CONF_NAME, PERCENTAGE, TEMP_CELSIUS
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
# Default values # Default values
@ -19,9 +15,9 @@ DEFAULT_NAME = "BeeWi SmartClim"
# Sensor config # Sensor config
SENSOR_TYPES = [ SENSOR_TYPES = [
[DEVICE_CLASS_TEMPERATURE, "Temperature", TEMP_CELSIUS], [SensorDeviceClass.TEMPERATURE, "Temperature", TEMP_CELSIUS],
[DEVICE_CLASS_HUMIDITY, "Humidity", PERCENTAGE], [SensorDeviceClass.HUMIDITY, "Humidity", PERCENTAGE],
[DEVICE_CLASS_BATTERY, "Battery", PERCENTAGE], [SensorDeviceClass.BATTERY, "Battery", PERCENTAGE],
] ]
PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend( PLATFORM_SCHEMA = PLATFORM_SCHEMA.extend(
@ -71,9 +67,9 @@ class BeewiSmartclimSensor(SensorEntity):
"""Fetch new state data from the poller.""" """Fetch new state data from the poller."""
self._poller.update_sensor() self._poller.update_sensor()
self._attr_native_value = None self._attr_native_value = None
if self._device == DEVICE_CLASS_TEMPERATURE: if self._device == SensorDeviceClass.TEMPERATURE:
self._attr_native_value = self._poller.get_temperature() self._attr_native_value = self._poller.get_temperature()
if self._device == DEVICE_CLASS_HUMIDITY: if self._device == SensorDeviceClass.HUMIDITY:
self._attr_native_value = self._poller.get_humidity() self._attr_native_value = self._poller.get_humidity()
if self._device == DEVICE_CLASS_BATTERY: if self._device == SensorDeviceClass.BATTERY:
self._attr_native_value = self._poller.get_battery() self._attr_native_value = self._poller.get_battery()