From 0b97dcf0bd64b538f7dabc5de8203048b8645d1b Mon Sep 17 00:00:00 2001 From: Marc Mueller <30130371+cdce8p@users.noreply.github.com> Date: Fri, 16 Sep 2022 16:46:26 +0200 Subject: [PATCH] Use vol.Coerce for notify SourceType enum (#77930) --- homeassistant/components/device_tracker/legacy.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) diff --git a/homeassistant/components/device_tracker/legacy.py b/homeassistant/components/device_tracker/legacy.py index 8216c5fba27..09fd5dce432 100644 --- a/homeassistant/components/device_tracker/legacy.py +++ b/homeassistant/components/device_tracker/legacy.py @@ -71,12 +71,7 @@ from .const import ( SERVICE_SEE: Final = "see" -SOURCE_TYPES: Final[tuple[str, ...]] = ( - SourceType.GPS, - SourceType.ROUTER, - SourceType.BLUETOOTH, - SourceType.BLUETOOTH_LE, -) +SOURCE_TYPES = [cls.value for cls in SourceType] NEW_DEVICE_DEFAULTS_SCHEMA = vol.Any( None, @@ -108,7 +103,7 @@ SERVICE_SEE_PAYLOAD_SCHEMA: Final[vol.Schema] = vol.Schema( ATTR_GPS_ACCURACY: cv.positive_int, ATTR_BATTERY: cv.positive_int, ATTR_ATTRIBUTES: dict, - ATTR_SOURCE_TYPE: vol.In(SOURCE_TYPES), + ATTR_SOURCE_TYPE: vol.Coerce(SourceType), ATTR_CONSIDER_HOME: cv.time_period, # Temp workaround for iOS app introduced in 0.65 vol.Optional("battery_status"): str,