Use vol.Coerce for notify SourceType enum (#77930)

This commit is contained in:
Marc Mueller 2022-09-16 16:46:26 +02:00 committed by GitHub
parent 501b8b341f
commit 0b97dcf0bd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -71,12 +71,7 @@ from .const import (
SERVICE_SEE: Final = "see" SERVICE_SEE: Final = "see"
SOURCE_TYPES: Final[tuple[str, ...]] = ( SOURCE_TYPES = [cls.value for cls in SourceType]
SourceType.GPS,
SourceType.ROUTER,
SourceType.BLUETOOTH,
SourceType.BLUETOOTH_LE,
)
NEW_DEVICE_DEFAULTS_SCHEMA = vol.Any( NEW_DEVICE_DEFAULTS_SCHEMA = vol.Any(
None, None,
@ -108,7 +103,7 @@ SERVICE_SEE_PAYLOAD_SCHEMA: Final[vol.Schema] = vol.Schema(
ATTR_GPS_ACCURACY: cv.positive_int, ATTR_GPS_ACCURACY: cv.positive_int,
ATTR_BATTERY: cv.positive_int, ATTR_BATTERY: cv.positive_int,
ATTR_ATTRIBUTES: dict, ATTR_ATTRIBUTES: dict,
ATTR_SOURCE_TYPE: vol.In(SOURCE_TYPES), ATTR_SOURCE_TYPE: vol.Coerce(SourceType),
ATTR_CONSIDER_HOME: cv.time_period, ATTR_CONSIDER_HOME: cv.time_period,
# Temp workaround for iOS app introduced in 0.65 # Temp workaround for iOS app introduced in 0.65
vol.Optional("battery_status"): str, vol.Optional("battery_status"): str,