From 0482d50d135e304f0a9719e010dfc45f3700b09c Mon Sep 17 00:00:00 2001 From: epenet <6771947+epenet@users.noreply.github.com> Date: Fri, 26 Aug 2022 10:54:22 +0200 Subject: [PATCH] Improve type hint in frontier silicon media player (#77167) --- .../frontier_silicon/media_player.py | 18 +++++++++++++----- 1 file changed, 13 insertions(+), 5 deletions(-) diff --git a/homeassistant/components/frontier_silicon/media_player.py b/homeassistant/components/frontier_silicon/media_player.py index 4c1e4390e61..87c7c4036f2 100644 --- a/homeassistant/components/frontier_silicon/media_player.py +++ b/homeassistant/components/frontier_silicon/media_player.py @@ -125,7 +125,7 @@ class AFSAPIDevice(MediaPlayerEntity): self._supports_sound_mode: bool = True - async def async_update(self): + async def async_update(self) -> None: """Get the latest date and update device state.""" afsapi = self.fs_device try: @@ -291,11 +291,19 @@ class AFSAPIDevice(MediaPlayerEntity): volume = int(volume * self._max_volume) await self.fs_device.set_volume(volume) - async def async_select_source(self, source): + async def async_select_source(self, source: str) -> None: """Select input source.""" await self.fs_device.set_power(True) - await self.fs_device.set_mode(self.__modes_by_label.get(source)) + if ( + self.__modes_by_label + and (mode := self.__modes_by_label.get(source)) is not None + ): + await self.fs_device.set_mode(mode) - async def async_select_sound_mode(self, sound_mode): + async def async_select_sound_mode(self, sound_mode: str) -> None: """Select EQ Preset.""" - await self.fs_device.set_eq_preset(self.__sound_modes_by_label[sound_mode]) + if ( + self.__sound_modes_by_label + and (mode := self.__sound_modes_by_label.get(sound_mode)) is not None + ): + await self.fs_device.set_eq_preset(mode)