From 06f670272fb266d91a7c9b36b17eac85a2194c69 Mon Sep 17 00:00:00 2001
From: Robert Hillis <tkdrob4390@yahoo.com>
Date: Thu, 16 Dec 2021 16:18:28 -0500
Subject: [PATCH] Use enums in obihai (#62078)

---
 homeassistant/components/obihai/sensor.py | 13 ++++++-------
 1 file changed, 6 insertions(+), 7 deletions(-)

diff --git a/homeassistant/components/obihai/sensor.py b/homeassistant/components/obihai/sensor.py
index 7ba28ee0741..f54611283a6 100644
--- a/homeassistant/components/obihai/sensor.py
+++ b/homeassistant/components/obihai/sensor.py
@@ -5,13 +5,12 @@ import logging
 from pyobihai import PyObihai
 import voluptuous as vol
 
-from homeassistant.components.sensor import PLATFORM_SCHEMA, SensorEntity
-from homeassistant.const import (
-    CONF_HOST,
-    CONF_PASSWORD,
-    CONF_USERNAME,
-    DEVICE_CLASS_TIMESTAMP,
+from homeassistant.components.sensor import (
+    PLATFORM_SCHEMA,
+    SensorDeviceClass,
+    SensorEntity,
 )
+from homeassistant.const import CONF_HOST, CONF_PASSWORD, CONF_USERNAME
 import homeassistant.helpers.config_validation as cv
 
 _LOGGER = logging.getLogger(__name__)
@@ -105,7 +104,7 @@ class ObihaiServiceSensors(SensorEntity):
     def device_class(self):
         """Return the device class for uptime sensor."""
         if self._service_name == "Last Reboot":
-            return DEVICE_CLASS_TIMESTAMP
+            return SensorDeviceClass.TIMESTAMP
         return None
 
     @property