Use new media player enums in soundtouch (#78110)

This commit is contained in:
epenet 2022-09-10 22:26:26 +02:00 committed by GitHub
parent 9e1cb914b1
commit 0d88567e0e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -13,12 +13,11 @@ import voluptuous as vol
from homeassistant.components import media_source from homeassistant.components import media_source
from homeassistant.components.media_player import ( from homeassistant.components.media_player import (
PLATFORM_SCHEMA, PLATFORM_SCHEMA,
BrowseMedia,
MediaPlayerDeviceClass, MediaPlayerDeviceClass,
MediaPlayerEntity, MediaPlayerEntity,
MediaPlayerEntityFeature, MediaPlayerEntityFeature,
) MediaPlayerState,
from homeassistant.components.media_player.browse_media import (
BrowseMedia,
async_process_play_media_url, async_process_play_media_url,
) )
from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry from homeassistant.config_entries import SOURCE_IMPORT, ConfigEntry
@ -27,11 +26,6 @@ from homeassistant.const import (
CONF_NAME, CONF_NAME,
CONF_PORT, CONF_PORT,
EVENT_HOMEASSISTANT_START, EVENT_HOMEASSISTANT_START,
STATE_OFF,
STATE_PAUSED,
STATE_PLAYING,
STATE_UNAVAILABLE,
STATE_UNKNOWN,
) )
from homeassistant.core import HomeAssistant, callback from homeassistant.core import HomeAssistant, callback
import homeassistant.helpers.config_validation as cv import homeassistant.helpers.config_validation as cv
@ -46,10 +40,10 @@ from .const import DOMAIN
_LOGGER = logging.getLogger(__name__) _LOGGER = logging.getLogger(__name__)
MAP_STATUS = { MAP_STATUS = {
"PLAY_STATE": STATE_PLAYING, "PLAY_STATE": MediaPlayerState.PLAYING,
"BUFFERING_STATE": STATE_PLAYING, "BUFFERING_STATE": MediaPlayerState.PLAYING,
"PAUSE_STATE": STATE_PAUSED, "PAUSE_STATE": MediaPlayerState.PAUSED,
"STOP_STATE": STATE_OFF, "STOP_STATE": MediaPlayerState.OFF,
} }
ATTR_SOUNDTOUCH_GROUP = "soundtouch_group" ATTR_SOUNDTOUCH_GROUP = "soundtouch_group"
@ -168,15 +162,15 @@ class SoundTouchMediaPlayer(MediaPlayerEntity):
return self._volume.actual / 100 return self._volume.actual / 100
@property @property
def state(self): def state(self) -> MediaPlayerState | None:
"""Return the state of the device.""" """Return the state of the device."""
if self._status is None or self._status.source == "STANDBY": if self._status is None or self._status.source == "STANDBY":
return STATE_OFF return MediaPlayerState.OFF
if self._status.source == "INVALID_SOURCE": if self._status.source == "INVALID_SOURCE":
return STATE_UNKNOWN return None
return MAP_STATUS.get(self._status.play_status, STATE_UNAVAILABLE) return MAP_STATUS.get(self._status.play_status)
@property @property
def source(self): def source(self):