From e8a03ea8707f9e2bef92457b931397e716438829 Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Mon, 28 Nov 2022 12:26:24 +0100 Subject: [PATCH] Use attributes in xiaomi-tv media player (#82842) --- .../components/xiaomi_tv/media_player.py | 28 +++++-------------- 1 file changed, 7 insertions(+), 21 deletions(-) diff --git a/homeassistant/components/xiaomi_tv/media_player.py b/homeassistant/components/xiaomi_tv/media_player.py index 9b9971f9568..5e045d1404a 100644 --- a/homeassistant/components/xiaomi_tv/media_player.py +++ b/homeassistant/components/xiaomi_tv/media_player.py @@ -58,6 +58,7 @@ def setup_platform( class XiaomiTV(MediaPlayerEntity): """Represent the Xiaomi TV for Home Assistant.""" + _attr_assumed_state = True _attr_supported_features = ( MediaPlayerEntityFeature.VOLUME_STEP | MediaPlayerEntityFeature.TURN_ON @@ -70,23 +71,8 @@ class XiaomiTV(MediaPlayerEntity): # Initialize the Xiaomi TV. self._tv = pymitv.TV(ip) # Default name value, only to be overridden by user. - self._name = name - self._state = MediaPlayerState.OFF - - @property - def name(self): - """Return the display name of this TV.""" - return self._name - - @property - def state(self): - """Return _state variable, containing the appropriate constant.""" - return self._state - - @property - def assumed_state(self): - """Indicate that state is assumed.""" - return True + self._attr_name = name + self._attr_state = MediaPlayerState.OFF def turn_off(self) -> None: """ @@ -96,17 +82,17 @@ class XiaomiTV(MediaPlayerEntity): because the TV won't accept any input when turned off. Thus, the user would be unable to turn the TV back on, unless it's done manually. """ - if self._state != MediaPlayerState.OFF: + if self.state != MediaPlayerState.OFF: self._tv.sleep() - self._state = MediaPlayerState.OFF + self._attr_state = MediaPlayerState.OFF def turn_on(self) -> None: """Wake the TV back up from sleep.""" - if self._state != MediaPlayerState.ON: + if self.state != MediaPlayerState.ON: self._tv.wake() - self._state = MediaPlayerState.ON + self._attr_state = MediaPlayerState.ON def volume_up(self) -> None: """Increase volume by one."""