evlli/hass-core
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Set roku media player device class in constructor (#100225)

Browse source
This commit is contained in:
J. Nick Koston 2023-09-12 12:58:20 -05:00 • committed by GitHub
parent 42c35da818
commit 9c775a8a24
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 8 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

16
homeassistant/components/roku/media_player.py
View file

@ -122,6 +122,14 @@ class RokuMediaPlayer(RokuEntity, MediaPlayerEntity):
| MediaPlayerEntityFeature.BROWSE_MEDIA
)
def __init__(self, coordinator: RokuDataUpdateCoordinator) -> None:
"""Initialize the Roku device."""
super().__init__(coordinator=coordinator)
if coordinator.data.info.device_type == "tv":
self._attr_device_class = MediaPlayerDeviceClass.TV
else:
self._attr_device_class = MediaPlayerDeviceClass.RECEIVER
def _media_playback_trackable(self) -> bool:
"""Detect if we have enough media data to track playback."""
if self.coordinator.data.media is None or self.coordinator.data.media.live:
@ -129,14 +137,6 @@ class RokuMediaPlayer(RokuEntity, MediaPlayerEntity):
return self.coordinator.data.media.duration > 0
@property
def device_class(self) -> MediaPlayerDeviceClass:
"""Return the class of this device."""
if self.coordinator.data.info.device_type == "tv":
return MediaPlayerDeviceClass.TV
return MediaPlayerDeviceClass.RECEIVER
@property
def state(self) -> MediaPlayerState | None:
"""Return the state of the device."""