Remove str from media player device class (#83398)
* Remove str from media player device class * Adjust pylint plugin
This commit is contained in:
parent
cb69364ad2
commit
647469f472
5 changed files with 8 additions and 7 deletions
|
@ -79,7 +79,7 @@ class LookinMedia(LookinPowerPushRemoteEntity, MediaPlayerEntity):
|
|||
uuid: str,
|
||||
device: Remote,
|
||||
lookin_data: LookinData,
|
||||
device_class: str,
|
||||
device_class: MediaPlayerDeviceClass,
|
||||
) -> None:
|
||||
"""Init the lookin media player."""
|
||||
self._attr_device_class = device_class
|
||||
|
|
|
@ -453,7 +453,7 @@ async def async_unload_entry(hass: HomeAssistant, entry: ConfigEntry) -> bool:
|
|||
class MediaPlayerEntityDescription(EntityDescription):
|
||||
"""A class that describes media player entities."""
|
||||
|
||||
device_class: MediaPlayerDeviceClass | str | None = None
|
||||
device_class: MediaPlayerDeviceClass | None = None
|
||||
|
||||
|
||||
class MediaPlayerEntity(Entity):
|
||||
|
@ -464,7 +464,7 @@ class MediaPlayerEntity(Entity):
|
|||
|
||||
_attr_app_id: str | None = None
|
||||
_attr_app_name: str | None = None
|
||||
_attr_device_class: MediaPlayerDeviceClass | str | None
|
||||
_attr_device_class: MediaPlayerDeviceClass | None
|
||||
_attr_group_members: list[str] | None = None
|
||||
_attr_is_volume_muted: bool | None = None
|
||||
_attr_media_album_artist: str | None = None
|
||||
|
@ -497,7 +497,7 @@ class MediaPlayerEntity(Entity):
|
|||
|
||||
# Implement these for your media player
|
||||
@property
|
||||
def device_class(self) -> MediaPlayerDeviceClass | str | None:
|
||||
def device_class(self) -> MediaPlayerDeviceClass | None:
|
||||
"""Return the class of this entity."""
|
||||
if hasattr(self, "_attr_device_class"):
|
||||
return self._attr_device_class
|
||||
|
|
|
@ -130,7 +130,7 @@ class RokuMediaPlayer(RokuEntity, MediaPlayerEntity):
|
|||
return self.coordinator.data.media.duration > 0
|
||||
|
||||
@property
|
||||
def device_class(self) -> str | None:
|
||||
def device_class(self) -> MediaPlayerDeviceClass:
|
||||
"""Return the class of this device."""
|
||||
if self.coordinator.data.info.device_type == "tv":
|
||||
return MediaPlayerDeviceClass.TV
|
||||
|
|
|
@ -40,6 +40,7 @@ from homeassistant.components.media_player import (
|
|||
SERVICE_PLAY_MEDIA,
|
||||
SERVICE_SELECT_SOUND_MODE,
|
||||
SERVICE_SELECT_SOURCE,
|
||||
MediaPlayerDeviceClass,
|
||||
MediaPlayerEntity,
|
||||
MediaPlayerEntityFeature,
|
||||
MediaPlayerState,
|
||||
|
@ -279,7 +280,7 @@ class UniversalMediaPlayer(MediaPlayerEntity):
|
|||
)
|
||||
|
||||
@property
|
||||
def device_class(self) -> str | None:
|
||||
def device_class(self) -> MediaPlayerDeviceClass | None:
|
||||
"""Return the class of this device."""
|
||||
return self._device_class
|
||||
|
||||
|
|
|
@ -1649,7 +1649,7 @@ _INHERITANCE_MATCH: dict[str, list[ClassTypeHintMatch]] = {
|
|||
matches=[
|
||||
TypeHintMatch(
|
||||
function_name="device_class",
|
||||
return_type=["MediaPlayerDeviceClass", "str", None],
|
||||
return_type=["MediaPlayerDeviceClass", None],
|
||||
),
|
||||
TypeHintMatch(
|
||||
function_name="state",
|
||||
|
|
Loading…
Add table
Reference in a new issue