Use enums in obihai (#62078)

This commit is contained in:
Robert Hillis 2021-12-16 16:18:28 -05:00 committed by GitHub
parent dab1a786a5
commit 06f670272f
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,13 +5,12 @@ import logging
from pyobihai import PyObihai from pyobihai import PyObihai
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_HOST, SensorDeviceClass,
CONF_PASSWORD, SensorEntity,
CONF_USERNAME,
DEVICE_CLASS_TIMESTAMP,
) )
from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
@ -105,7 +104,7 @@ class ObihaiServiceSensors(SensorEntity):
def device_class(self): def device_class(self):
"""Return the device class for uptime sensor.""" """Return the device class for uptime sensor."""
if self._service_name == "Last Reboot": if self._service_name == "Last Reboot":
return DEVICE_CLASS_TIMESTAMP return SensorDeviceClass.TIMESTAMP
return None return None
@property @property