Use EntityFeature enum in vizio (#69571)

This commit is contained in:
epenet 2022-04-07 14:02:17 +02:00 committed by GitHub
parent 23b84449e6
commit 889e1f4442
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -5,16 +5,9 @@ from pyvizio.const import (
) )
import voluptuous as vol import voluptuous as vol
from homeassistant.components.media_player import MediaPlayerDeviceClass from homeassistant.components.media_player import (
from homeassistant.components.media_player.const import ( MediaPlayerDeviceClass,
SUPPORT_NEXT_TRACK, MediaPlayerEntityFeature,
SUPPORT_PREVIOUS_TRACK,
SUPPORT_SELECT_SOURCE,
SUPPORT_TURN_OFF,
SUPPORT_TURN_ON,
SUPPORT_VOLUME_MUTE,
SUPPORT_VOLUME_SET,
SUPPORT_VOLUME_STEP,
) )
from homeassistant.const import ( from homeassistant.const import (
CONF_ACCESS_TOKEN, CONF_ACCESS_TOKEN,
@ -62,18 +55,20 @@ ICON = {
} }
COMMON_SUPPORTED_COMMANDS = ( COMMON_SUPPORTED_COMMANDS = (
SUPPORT_SELECT_SOURCE MediaPlayerEntityFeature.SELECT_SOURCE
| SUPPORT_TURN_ON | MediaPlayerEntityFeature.TURN_ON
| SUPPORT_TURN_OFF | MediaPlayerEntityFeature.TURN_OFF
| SUPPORT_VOLUME_MUTE | MediaPlayerEntityFeature.VOLUME_MUTE
| SUPPORT_VOLUME_SET | MediaPlayerEntityFeature.VOLUME_SET
| SUPPORT_VOLUME_STEP | MediaPlayerEntityFeature.VOLUME_STEP
) )
SUPPORTED_COMMANDS = { SUPPORTED_COMMANDS = {
MediaPlayerDeviceClass.SPEAKER: COMMON_SUPPORTED_COMMANDS, MediaPlayerDeviceClass.SPEAKER: COMMON_SUPPORTED_COMMANDS,
MediaPlayerDeviceClass.TV: ( MediaPlayerDeviceClass.TV: (
COMMON_SUPPORTED_COMMANDS | SUPPORT_NEXT_TRACK | SUPPORT_PREVIOUS_TRACK COMMON_SUPPORTED_COMMANDS
| MediaPlayerEntityFeature.NEXT_TRACK
| MediaPlayerEntityFeature.PREVIOUS_TRACK
), ),
} }