Use SensorDeviceClass in mfi (#61900)

Co-authored-by: epenet <epenet@users.noreply.github.com>
This commit is contained in:
epenet 2021-12-16 15:51:36 +01:00 committed by GitHub
parent 599c8f4757
commit 7c8d235356
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -5,7 +5,11 @@ from mficlient.client import FailedToLogin, MFiClient
import requests
import voluptuous as vol
from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
from homeassistant.components.sensor import (
PLATFORM_SCHEMA,
SensorDeviceClass,
SensorEntity,
)
from homeassistant.const import (
CONF_HOST,
CONF_PASSWORD,
@ -13,7 +17,6 @@ from homeassistant.const import (
CONF_SSL,
CONF_USERNAME,
CONF_VERIFY_SSL,
DEVICE_CLASS_TEMPERATURE,
STATE_OFF,
STATE_ON,
TEMP_CELSIUS,
@ -110,7 +113,7 @@ class MfiSensor(SensorEntity):
return None
if tag == "temperature":
return DEVICE_CLASS_TEMPERATURE
return SensorDeviceClass.TEMPERATURE
return None