diff --git a/homeassistant/components/vizio/const.py b/homeassistant/components/vizio/const.py index bcfc38950d3..953d249207c 100644 --- a/homeassistant/components/vizio/const.py +++ b/homeassistant/components/vizio/const.py @@ -5,7 +5,7 @@ from pyvizio.const import ( ) import voluptuous as vol -from homeassistant.components.media_player import DEVICE_CLASS_SPEAKER, DEVICE_CLASS_TV +from homeassistant.components.media_player import MediaPlayerDeviceClass from homeassistant.components.media_player.const import ( SUPPORT_NEXT_TRACK, SUPPORT_PREVIOUS_TRACK, @@ -48,7 +48,7 @@ CONF_NAME_SPACE = "NAME_SPACE" CONF_MESSAGE = "MESSAGE" CONF_VOLUME_STEP = "volume_step" -DEFAULT_DEVICE_CLASS = DEVICE_CLASS_TV +DEFAULT_DEVICE_CLASS = MediaPlayerDeviceClass.TV DEFAULT_NAME = "Vizio SmartCast" DEFAULT_TIMEOUT = 8 DEFAULT_VOLUME_STEP = 1 @@ -56,7 +56,10 @@ DEFAULT_VOLUME_STEP = 1 DEVICE_ID = "pyvizio" DOMAIN = "vizio" -ICON = {DEVICE_CLASS_TV: "mdi:television", DEVICE_CLASS_SPEAKER: "mdi:speaker"} +ICON = { + MediaPlayerDeviceClass.TV: "mdi:television", + MediaPlayerDeviceClass.SPEAKER: "mdi:speaker", +} COMMON_SUPPORTED_COMMANDS = ( SUPPORT_SELECT_SOURCE @@ -68,8 +71,8 @@ COMMON_SUPPORTED_COMMANDS = ( ) SUPPORTED_COMMANDS = { - DEVICE_CLASS_SPEAKER: COMMON_SUPPORTED_COMMANDS, - DEVICE_CLASS_TV: ( + MediaPlayerDeviceClass.SPEAKER: COMMON_SUPPORTED_COMMANDS, + MediaPlayerDeviceClass.TV: ( COMMON_SUPPORTED_COMMANDS | SUPPORT_NEXT_TRACK | SUPPORT_PREVIOUS_TRACK ), } @@ -83,8 +86,8 @@ VIZIO_MUTE = "mute" # Since Vizio component relies on device class, this dict will ensure that changes to # the values of DEVICE_CLASS_SPEAKER or DEVICE_CLASS_TV don't require changes to pyvizio. VIZIO_DEVICE_CLASSES = { - DEVICE_CLASS_SPEAKER: VIZIO_DEVICE_CLASS_SPEAKER, - DEVICE_CLASS_TV: VIZIO_DEVICE_CLASS_TV, + MediaPlayerDeviceClass.SPEAKER: VIZIO_DEVICE_CLASS_SPEAKER, + MediaPlayerDeviceClass.TV: VIZIO_DEVICE_CLASS_TV, } VIZIO_SCHEMA = { @@ -92,7 +95,9 @@ VIZIO_SCHEMA = { vol.Optional(CONF_ACCESS_TOKEN): cv.string, vol.Optional(CONF_NAME, default=DEFAULT_NAME): cv.string, vol.Optional(CONF_DEVICE_CLASS, default=DEFAULT_DEVICE_CLASS): vol.All( - cv.string, vol.Lower, vol.In([DEVICE_CLASS_TV, DEVICE_CLASS_SPEAKER]) + cv.string, + vol.Lower, + vol.In([MediaPlayerDeviceClass.TV, MediaPlayerDeviceClass.SPEAKER]), ), vol.Optional(CONF_VOLUME_STEP, default=DEFAULT_VOLUME_STEP): vol.All( vol.Coerce(int), vol.Range(min=1, max=10)