Adjust type hints for MediaPlayerEntityFeature (#82258)

homeassistant/components/bluesound/media_player.py

@@ -775,7 +775,7 @@ def source(self):
         return None
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag of media commands that are supported."""
         if self._status is None:
             return 0

homeassistant/components/cast/media_player.py

@@ -899,7 +899,7 @@ def app_name(self):
         return self._chromecast.app_display_name if self._chromecast else None
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         support = (
             MediaPlayerEntityFeature.PLAY_MEDIA

homeassistant/components/denon/media_player.py

@@ -245,7 +245,7 @@ def media_title(self):
         return self._mediainfo
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self._mediasource in MEDIA_MODES.values():
             return SUPPORT_DENON | SUPPORT_MEDIA_MODES

homeassistant/components/denonavr/media_player.py

@@ -277,7 +277,7 @@ def sound_mode(self):
         return self._receiver.sound_mode
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self._receiver.input_func in self._receiver.netaudio_func_list:
             return self._supported_features_base | SUPPORT_MEDIA_MODES

homeassistant/components/directv/media_player.py

@@ -237,7 +237,7 @@ def source(self):
         return self._program.channel
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         return SUPPORT_DTV_CLIENT if self._is_client else SUPPORT_DTV
 

homeassistant/components/dlna_dmr/media_player.py

@@ -478,7 +478,7 @@ def state(self) -> MediaPlayerState | None:
         return MediaPlayerState.IDLE
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported at this moment.
 
         Supported features may change as the device enters different states.

homeassistant/components/dunehd/media_player.py

@@ -19,7 +19,7 @@
 
 CONF_SOURCES: Final = "sources"
 
-DUNEHD_PLAYER_SUPPORT: Final[int] = (
+DUNEHD_PLAYER_SUPPORT: Final[MediaPlayerEntityFeature] = (
     MediaPlayerEntityFeature.PAUSE
     | MediaPlayerEntityFeature.TURN_ON
     | MediaPlayerEntityFeature.TURN_OFF
@@ -105,7 +105,7 @@ def is_volume_muted(self) -> bool:
         return int(self._state.get("playback_mute", 0)) == 1
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         return DUNEHD_PLAYER_SUPPORT
 

homeassistant/components/emby/media_player.py

@@ -284,7 +284,7 @@ def media_album_artist(self):
         return self.device.media_album_artist
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         if self.supports_remote_control:
             return SUPPORT_EMBY

homeassistant/components/esphome/media_player.py

@@ -83,7 +83,7 @@ def volume_level(self) -> float | None:
         return self._state.volume
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag supported features."""
         flags = (
             MediaPlayerEntityFeature.PLAY_MEDIA

homeassistant/components/forked_daapd/media_player.py

@@ -17,6 +17,7 @@
     BrowseMedia,
     MediaPlayerEnqueue,
     MediaPlayerEntity,
+    MediaPlayerEntityFeature,
     MediaPlayerState,
     MediaType,
     async_process_play_media_url,
@@ -236,7 +237,7 @@ async def async_set_volume_level(self, volume: float) -> None:
         await self._api.set_volume(volume=volume * 100, output_id=self._output_id)
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         return SUPPORTED_FEATURES_ZONE
 
@@ -558,7 +559,7 @@ def shuffle(self):
         return self._player["shuffle"]
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         return SUPPORTED_FEATURES
 

homeassistant/components/hdmi_cec/media_player.py

@@ -163,7 +163,7 @@ def update(self) -> None:
             _LOGGER.warning("Unknown state: %s", device.status)
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self.type_id == TYPE_RECORDER or self.type == TYPE_PLAYBACK:
             return (

homeassistant/components/homekit_controller/media_player.py

@@ -80,7 +80,7 @@ def get_characteristic_types(self) -> list[str]:
         ]
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         features = 0
 

homeassistant/components/jellyfin/media_player.py

@@ -198,7 +198,7 @@ def media_image_url(self) -> str | None:
         return get_artwork_url(self.coordinator.api_client, self.now_playing, 150)
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         commands: list[str] = self.capabilities.get("SupportedCommands", [])
         controllable = self.capabilities.get("SupportsMediaControl", False)

homeassistant/components/lg_netcast/media_player.py

@@ -197,7 +197,7 @@ def media_title(self):
         return self._program_name
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self._on_action_script:
             return SUPPORT_LGTV | MediaPlayerEntityFeature.TURN_ON

homeassistant/components/lookin/media_player.py

@@ -82,7 +82,6 @@ def __init__(
     ) -> None:
         """Init the lookin media player."""
         self._attr_device_class = device_class
-        self._attr_supported_features: int = 0
         super().__init__(coordinator, uuid, device, lookin_data)
         for function_name, feature in _FUNCTION_NAME_TO_FEATURE.items():
             if function_name in self._function_names:

homeassistant/components/media_player/__init__.py

@@ -492,7 +492,7 @@ class MediaPlayerEntity(Entity):
     _attr_source_list: list[str] | None = None
     _attr_source: str | None = None
     _attr_state: MediaPlayerState | str | None = None
-    _attr_supported_features: int = 0
+    _attr_supported_features: MediaPlayerEntityFeature | int = 0
     _attr_volume_level: float | None = None
 
     # Implement these for your media player
@@ -692,7 +692,7 @@ def group_members(self) -> list[str] | None:
         return self._attr_group_members
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         return self._attr_supported_features
 

homeassistant/components/mpd/media_player.py

@@ -338,7 +338,7 @@ def volume_level(self):
         return None
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         if self._status is None:
             return 0

homeassistant/components/onkyo/media_player.py

@@ -553,7 +553,7 @@ def update(self) -> None:
             )
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Return media player features that are supported."""
         if self._supports_volume:
             return SUPPORT_ONKYO

homeassistant/components/philips_js/media_player.py

@@ -104,7 +104,7 @@ async def _async_update_soon(self):
         await self.coordinator.async_request_refresh()
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         supports = self._supports
         if self.coordinator.turn_on or (

homeassistant/components/plex/media_player.py

@@ -389,7 +389,7 @@ def media_episode(self):
         return self.session.media_episode
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self.device and "playback" in self._device_protocol_capabilities:
             return (

homeassistant/components/universal/media_player.py

@@ -453,9 +453,11 @@ def shuffle(self):
         return self._override_or_child_attr(ATTR_MEDIA_SHUFFLE)
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
-        flags = self._child_attr(ATTR_SUPPORTED_FEATURES) or 0
+        flags: MediaPlayerEntityFeature | int = (
+            self._child_attr(ATTR_SUPPORTED_FEATURES) or 0
+        )
 
         if SERVICE_TURN_ON in self._cmds:
             flags |= MediaPlayerEntityFeature.TURN_ON

homeassistant/components/webostv/media_player.py

@@ -137,7 +137,7 @@ def __init__(
         self._current_source = None
         self._source_list: dict = {}
 
-        self._supported_features: int = 0
+        self._supported_features: MediaPlayerEntityFeature | int = 0
         self._update_states()
 
     async def async_added_to_hass(self) -> None:
@@ -314,7 +314,7 @@ async def async_update(self) -> None:
             await self._client.connect()
 
     @property
-    def supported_features(self) -> int:
+    def supported_features(self) -> MediaPlayerEntityFeature | int:
         """Flag media player features that are supported."""
         if self._wrapper.turn_on:
             return self._supported_features | MediaPlayerEntityFeature.TURN_ON

homeassistant/components/xbox/media_player.py

@@ -107,7 +107,7 @@ def state(self):
         return XBOX_STATE_MAP[status.power_state]
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         if self.state not in [MediaPlayerState.PLAYING, MediaPlayerState.PAUSED]:
             return (

homeassistant/components/yamaha/media_player.py

@@ -311,7 +311,7 @@ def zone_id(self):
         return f"{self.receiver.ctrl_url}:{self._zone}"
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         supported_features = SUPPORT_YAMAHA
 

homeassistant/components/yamaha_musiccast/media_player.py

@@ -424,7 +424,7 @@ def repeat(self):
         )
 
     @property
-    def supported_features(self):
+    def supported_features(self) -> MediaPlayerEntityFeature:
         """Flag media player features that are supported."""
         supported_features = MUSIC_PLAYER_BASE_SUPPORT
         zone = self.coordinator.data.zones[self._zone_id]

pylint/plugins/hass_enforce_type_hints.py

@@ -1788,6 +1788,10 @@ class ClassTypeHintMatch:
                     function_name="group_members",
                     return_type=["list[str]", None],
                 ),
+                TypeHintMatch(
+                    function_name="supported_features",
+                    return_type=["MediaPlayerEntityFeature", "int"],
+                ),
                 TypeHintMatch(
                     function_name="turn_on",
                     return_type=None,