Use DeviceClass Enum in ness-alarm schema (#61969)

This commit is contained in:
epenet 2021-12-16 15:20:40 +01:00 committed by GitHub
parent b16d779280
commit c8607b1a4c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -5,7 +5,10 @@ import datetime
from nessclient import ArmingState, Client from nessclient import ArmingState, Client
import voluptuous as vol 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 ( from homeassistant.const import (
ATTR_CODE, ATTR_CODE,
ATTR_STATE, ATTR_STATE,
@ -36,12 +39,14 @@ SIGNAL_ARMING_STATE_CHANGED = "ness_alarm.arming_state_changed"
ZoneChangedData = namedtuple("ZoneChangedData", ["zone_id", "state"]) ZoneChangedData = namedtuple("ZoneChangedData", ["zone_id", "state"])
DEFAULT_ZONE_TYPE = "motion" DEFAULT_ZONE_TYPE = BinarySensorDeviceClass.MOTION
ZONE_SCHEMA = vol.Schema( ZONE_SCHEMA = vol.Schema(
{ {
vol.Required(CONF_ZONE_NAME): cv.string, vol.Required(CONF_ZONE_NAME): cv.string,
vol.Required(CONF_ZONE_ID): cv.positive_int, 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,
} }
) )