From c8607b1a4c3ffd8e842f016813ff23cf28733ad2 Mon Sep 17 00:00:00 2001
From: epenet <6771947+epenet@users.noreply.github.com>
Date: Thu, 16 Dec 2021 15:20:40 +0100
Subject: [PATCH] Use DeviceClass Enum in ness-alarm schema (#61969)

---
 homeassistant/components/ness_alarm/__init__.py | 11 ++++++++---
 1 file changed, 8 insertions(+), 3 deletions(-)

diff --git a/homeassistant/components/ness_alarm/__init__.py b/homeassistant/components/ness_alarm/__init__.py
index e17e63ee46c..e15e0fb2e20 100644
--- a/homeassistant/components/ness_alarm/__init__.py
+++ b/homeassistant/components/ness_alarm/__init__.py
@@ -5,7 +5,10 @@ import datetime
 from nessclient import ArmingState, Client
 import voluptuous as vol
 
-from homeassistant.components.binary_sensor import DEVICE_CLASSES
+from homeassistant.components.binary_sensor import (
+    DEVICE_CLASSES_SCHEMA as BINARY_SENSOR_DEVICE_CLASSES_SCHEMA,
+    BinarySensorDeviceClass,
+)
 from homeassistant.const import (
     ATTR_CODE,
     ATTR_STATE,
@@ -36,12 +39,14 @@ SIGNAL_ARMING_STATE_CHANGED = "ness_alarm.arming_state_changed"
 
 ZoneChangedData = namedtuple("ZoneChangedData", ["zone_id", "state"])
 
-DEFAULT_ZONE_TYPE = "motion"
+DEFAULT_ZONE_TYPE = BinarySensorDeviceClass.MOTION
 ZONE_SCHEMA = vol.Schema(
     {
         vol.Required(CONF_ZONE_NAME): cv.string,
         vol.Required(CONF_ZONE_ID): cv.positive_int,
-        vol.Optional(CONF_ZONE_TYPE, default=DEFAULT_ZONE_TYPE): vol.In(DEVICE_CLASSES),
+        vol.Optional(
+            CONF_ZONE_TYPE, default=DEFAULT_ZONE_TYPE
+        ): BINARY_SENSOR_DEVICE_CLASSES_SCHEMA,
     }
 )