Address System Bridge post merge review (#101614)

Address Post Merged PR Review for #97532
This commit is contained in:
Aidan Timson
2023-10-08 09:42:08 +01:00
committed by GitHub
parent b8b28e3ba0
commit e406b8d1e3

View File

@@ -54,6 +54,15 @@ MEDIA_SET_REPEAT_MAP: Final[dict[RepeatMode, int]] = {
RepeatMode.ALL: 2,
}
MEDIA_PLAYER_DESCRIPTION: Final[
MediaPlayerEntityDescription
] = MediaPlayerEntityDescription(
key="media",
translation_key="media",
icon="mdi:volume-high",
device_class=MediaPlayerDeviceClass.RECEIVER,
)
async def async_setup_entry(
hass: HomeAssistant,
@@ -62,19 +71,14 @@ async def async_setup_entry(
) -> None:
"""Set up System Bridge media players based on a config entry."""
coordinator: SystemBridgeDataUpdateCoordinator = hass.data[DOMAIN][entry.entry_id]
data: SystemBridgeCoordinatorData = coordinator.data
data = 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,
),
MEDIA_PLAYER_DESCRIPTION,
entry.data[CONF_PORT],
)
]
@@ -103,7 +107,7 @@ class SystemBridgeMediaPlayer(SystemBridgeEntity, MediaPlayerEntity):
@property
def available(self) -> bool:
"""Return True if entity is available."""
return self.coordinator.data.media is not None
return super().available and self.coordinator.data.media is not None
@property
def supported_features(self) -> MediaPlayerEntityFeature: