Add media player to System Bridge integration (#97532)
Co-authored-by: Joost Lekkerkerker <joostlek@outlook.com>
This commit is contained in:
Aidan Timson
GitHub
parent
96aba1c1a6
commit
9982483395
6 changed files with 274 additions and 0 deletions
|
|
@ -1294,6 +1294,7 @@ omit =
|
|||
homeassistant/components/system_bridge/__init__.py
|
||||
homeassistant/components/system_bridge/binary_sensor.py
|
||||
homeassistant/components/system_bridge/coordinator.py
|
||||
homeassistant/components/system_bridge/media_player.py
|
||||
homeassistant/components/system_bridge/notify.py
|
||||
homeassistant/components/system_bridge/sensor.py
|
||||
homeassistant/components/systemmonitor/sensor.py
|
||||
|
|
|
|||
|
|
@ -46,6 +46,7 @@ _LOGGER = logging.getLogger(__name__)
|
|||
|
||||
PLATFORMS = [
|
||||
Platform.BINARY_SENSOR,
|
||||
Platform.MEDIA_PLAYER,
|
||||
Platform.NOTIFY,
|
||||
Platform.SENSOR,
|
||||
]
|
||||
|
|
|
|||
|
|
@ -8,6 +8,7 @@ MODULES = [
|
|||
"disk",
|
||||
"display",
|
||||
"gpu",
|
||||
"media",
|
||||
"memory",
|
||||
"system",
|
||||
]
|
||||
|
|
|
|||
|
|
@ -19,6 +19,7 @@ from systembridgeconnector.models.disk import Disk
|
|||
from systembridgeconnector.models.display import Display
|
||||
from systembridgeconnector.models.get_data import GetData
|
||||
from systembridgeconnector.models.gpu import Gpu
|
||||
from systembridgeconnector.models.media import Media
|
||||
from systembridgeconnector.models.media_directories import MediaDirectories
|
||||
from systembridgeconnector.models.media_files import File as MediaFile, MediaFiles
|
||||
from systembridgeconnector.models.media_get_file import MediaGetFile
|
||||
|
|
@ -50,6 +51,7 @@ class SystemBridgeCoordinatorData(BaseModel):
|
|||
disk: Disk = None
|
||||
display: Display = None
|
||||
gpu: Gpu = None
|
||||
media: Media = None
|
||||
memory: Memory = None
|
||||
system: System = None
|
||||
|
||||
|
|
|
|||
264
homeassistant/components/system_bridge/media_player.py
Normal file
264
homeassistant/components/system_bridge/media_player.py
Normal file
|
|
@ -0,0 +1,264 @@
|
|||
"""Support for System Bridge media players."""
|
||||
from __future__ import annotations
|
||||
|
||||
import datetime as dt
|
||||
from typing import Final
|
||||
|
||||
from systembridgeconnector.models.media_control import (
|
||||
Action as MediaAction,
|
||||
MediaControl,
|
||||
)
|
||||
|
||||
from homeassistant.components.media_player import (
|
||||
MediaPlayerDeviceClass,
|
||||
MediaPlayerEntity,
|
||||
MediaPlayerEntityDescription,
|
||||
MediaPlayerEntityFeature,
|
||||
MediaPlayerState,
|
||||
RepeatMode,
|
||||
)
|
||||
from homeassistant.config_entries import ConfigEntry
|
||||
from homeassistant.const import CONF_PORT
|
||||
from homeassistant.core import HomeAssistant
|
||||
from homeassistant.helpers.entity_platform import AddEntitiesCallback
|
||||
|
||||
from . import SystemBridgeEntity
|
||||
from .const import DOMAIN
|
||||
from .coordinator import SystemBridgeCoordinatorData, SystemBridgeDataUpdateCoordinator
|
||||
|
||||
STATUS_CHANGING: Final[str] = "CHANGING"
|
||||
STATUS_STOPPED: Final[str] = "STOPPED"
|
||||
STATUS_PLAYING: Final[str] = "PLAYING"
|
||||
STATUS_PAUSED: Final[str] = "PAUSED"
|
||||
|
||||
REPEAT_NONE: Final[str] = "NONE"
|
||||
REPEAT_TRACK: Final[str] = "TRACK"
|
||||
REPEAT_LIST: Final[str] = "LIST"
|
||||
|
||||
MEDIA_STATUS_MAP: Final[dict[str, MediaPlayerState]] = {
|
||||
STATUS_CHANGING: MediaPlayerState.IDLE,
|
||||
STATUS_STOPPED: MediaPlayerState.IDLE,
|
||||
STATUS_PLAYING: MediaPlayerState.PLAYING,
|
||||
STATUS_PAUSED: MediaPlayerState.PAUSED,
|
||||
}
|
||||
|
||||
MEDIA_REPEAT_MAP: Final[dict[str, RepeatMode]] = {
|
||||
REPEAT_NONE: RepeatMode.OFF,
|
||||
REPEAT_TRACK: RepeatMode.ONE,
|
||||
REPEAT_LIST: RepeatMode.ALL,
|
||||
}
|
||||
|
||||
MEDIA_SET_REPEAT_MAP: Final[dict[RepeatMode, int]] = {
|
||||
RepeatMode.OFF: 0,
|
||||
RepeatMode.ONE: 1,
|
||||
RepeatMode.ALL: 2,
|
||||
}
|
||||
|
||||
|
||||
async def async_setup_entry(
|
||||
hass: HomeAssistant,
|
||||
entry: ConfigEntry,
|
||||
async_add_entities: AddEntitiesCallback,
|
||||
) -> None:
|
||||
"""Set up System Bridge media players based on a config entry."""
|
||||
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
|
||||
data: SystemBridgeCoordinatorData = coordinator.data
|
||||
|
||||
if data.media is not None:
|
||||
async_add_entities(
|
||||
[
|
||||
SystemBridgeMediaPlayer(
|
||||
coordinator,
|
||||
MediaPlayerEntityDescription(
|
||||
key="media",
|
||||
translation_key="media",
|
||||
icon="mdi:volume-high",
|
||||
device_class=MediaPlayerDeviceClass.RECEIVER,
|
||||
),
|
||||
entry.data[CONF_PORT],
|
||||
)
|
||||
]
|
||||
)
|
||||
|
||||
|
||||
class SystemBridgeMediaPlayer(SystemBridgeEntity, MediaPlayerEntity):
|
||||
"""Define a System Bridge media player."""
|
||||
|
||||
entity_description: MediaPlayerEntityDescription
|
||||
|
||||
def __init__(
|
||||
self,
|
||||
coordinator: SystemBridgeDataUpdateCoordinator,
|
||||
description: MediaPlayerEntityDescription,
|
||||
api_port: int,
|
||||
) -> None:
|
||||
"""Initialize."""
|
||||
super().__init__(
|
||||
coordinator,
|
||||
api_port,
|
||||
description.key,
|
||||
)
|
||||
self.entity_description = description
|
||||
|
||||
@property
|
||||
def available(self) -> bool:
|
||||
"""Return True if entity is available."""
|
||||
return self.coordinator.data.media is not None
|
||||
|
||||
@property
|
||||
def supported_features(self) -> MediaPlayerEntityFeature:
|
||||
"""Flag media player features that are supported."""
|
||||
features = (
|
||||
MediaPlayerEntityFeature.REPEAT_SET | MediaPlayerEntityFeature.SHUFFLE_SET
|
||||
)
|
||||
|
||||
data = self._systembridge_data
|
||||
if data.media.is_previous_enabled:
|
||||
features |= MediaPlayerEntityFeature.PREVIOUS_TRACK
|
||||
if data.media.is_next_enabled:
|
||||
features |= MediaPlayerEntityFeature.NEXT_TRACK
|
||||
if data.media.is_pause_enabled:
|
||||
features |= MediaPlayerEntityFeature.PAUSE
|
||||
if data.media.is_play_enabled:
|
||||
features |= MediaPlayerEntityFeature.PLAY
|
||||
if data.media.is_stop_enabled:
|
||||
features |= MediaPlayerEntityFeature.STOP
|
||||
|
||||
return features
|
||||
|
||||
@property
|
||||
def _systembridge_data(self) -> SystemBridgeCoordinatorData:
|
||||
"""Return data for the entity."""
|
||||
return self.coordinator.data
|
||||
|
||||
@property
|
||||
def state(self) -> MediaPlayerState | None:
|
||||
"""State of the player."""
|
||||
if self._systembridge_data.media.status is None:
|
||||
return None
|
||||
return MEDIA_STATUS_MAP.get(
|
||||
self._systembridge_data.media.status,
|
||||
MediaPlayerState.IDLE,
|
||||
)
|
||||
|
||||
@property
|
||||
def media_duration(self) -> int | None:
|
||||
"""Duration of current playing media in seconds."""
|
||||
if self._systembridge_data.media.duration is None:
|
||||
return None
|
||||
return int(self._systembridge_data.media.duration)
|
||||
|
||||
@property
|
||||
def media_position(self) -> int | None:
|
||||
"""Position of current playing media in seconds."""
|
||||
if self._systembridge_data.media.position is None:
|
||||
return None
|
||||
return int(self._systembridge_data.media.position)
|
||||
|
||||
@property
|
||||
def media_position_updated_at(self) -> dt.datetime | None:
|
||||
"""When was the position of the current playing media valid."""
|
||||
if self._systembridge_data.media.updated_at is None:
|
||||
return None
|
||||
return dt.datetime.fromtimestamp(self._systembridge_data.media.updated_at)
|
||||
|
||||
@property
|
||||
def media_title(self) -> str | None:
|
||||
"""Title of current playing media."""
|
||||
return self._systembridge_data.media.title
|
||||
|
||||
@property
|
||||
def media_artist(self) -> str | None:
|
||||
"""Artist of current playing media, music track only."""
|
||||
return self._systembridge_data.media.artist
|
||||
|
||||
@property
|
||||
def media_album_name(self) -> str | None:
|
||||
"""Album name of current playing media, music track only."""
|
||||
return self._systembridge_data.media.album_title
|
||||
|
||||
@property
|
||||
def media_album_artist(self) -> str | None:
|
||||
"""Album artist of current playing media, music track only."""
|
||||
return self._systembridge_data.media.album_artist
|
||||
|
||||
@property
|
||||
def media_track(self) -> int | None:
|
||||
"""Track number of current playing media, music track only."""
|
||||
return self._systembridge_data.media.track_number
|
||||
|
||||
@property
|
||||
def shuffle(self) -> bool | None:
|
||||
"""Boolean if shuffle is enabled."""
|
||||
return self._systembridge_data.media.shuffle
|
||||
|
||||
@property
|
||||
def repeat(self) -> RepeatMode | None:
|
||||
"""Return current repeat mode."""
|
||||
if self._systembridge_data.media.repeat is None:
|
||||
return RepeatMode.OFF
|
||||
return MEDIA_REPEAT_MAP.get(self._systembridge_data.media.repeat)
|
||||
|
||||
async def async_media_play(self) -> None:
|
||||
"""Send play command."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.play,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_media_pause(self) -> None:
|
||||
"""Send pause command."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.pause,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_media_stop(self) -> None:
|
||||
"""Send stop command."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.stop,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_media_previous_track(self) -> None:
|
||||
"""Send previous track command."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.previous,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_media_next_track(self) -> None:
|
||||
"""Send next track command."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.next,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_set_shuffle(
|
||||
self,
|
||||
shuffle: bool,
|
||||
) -> None:
|
||||
"""Enable/disable shuffle mode."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.shuffle,
|
||||
value=shuffle,
|
||||
)
|
||||
)
|
||||
|
||||
async def async_set_repeat(
|
||||
self,
|
||||
repeat: RepeatMode,
|
||||
) -> None:
|
||||
"""Set repeat mode."""
|
||||
await self.coordinator.websocket_client.media_control(
|
||||
MediaControl(
|
||||
action=MediaAction.repeat,
|
||||
value=MEDIA_SET_REPEAT_MAP.get(repeat),
|
||||
)
|
||||
)
|
||||
|
|
@ -29,6 +29,11 @@
|
|||
}
|
||||
},
|
||||
"entity": {
|
||||
"media_player": {
|
||||
"media": {
|
||||
"name": "Media"
|
||||
}
|
||||
},
|
||||
"sensor": {
|
||||
"boot_time": {
|
||||
"name": "Boot time"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue