Use MediaPlayerDeviceClass in vizio (#69265)

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)