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, } )