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
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

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: