[pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
cdnninja · Jun 4, 2024 · f7cfd10 · f7cfd10
1 parent a920383
commit f7cfd10
Showing 1 changed file with 19 additions and 10 deletions.
diff --git a/custom_components/yoto/media_player.py b/custom_components/yoto/media_player.py
@@ -164,19 +164,28 @@ def media_title(self) -> str:
     def extra_state_attributes(self):
         """Return device specific state attributes."""
         if self.media_content_id in self.coordinator.yoto_manager.library:
-            if self.player.chapter_key in self.coordinator.yoto_manager.library[
-                self.media_content_id
-            ].chapters:
-                if self.player.track_key in self.coordinator.yoto_manager.library[
-                    self.media_content_id
-                ].chapters[self.player.chapter_key].tracks:
+            if (
+                self.player.chapter_key
+                in self.coordinator.yoto_manager.library[self.media_content_id].chapters
+            ):
+                if (
+                    self.player.track_key
+                    in self.coordinator.yoto_manager.library[self.media_content_id]
+                    .chapters[self.player.chapter_key]
+                    .tracks
+                ):
                     return {
                         "media_chapter_icon": self.coordinator.yoto_manager.library[
-                        self.media_content_id
-                    ].chapters[self.player.chapter_key].icon,
+                            self.media_content_id
+                        ]
+                        .chapters[self.player.chapter_key]
+                        .icon,
                         "media_track_icon": self.coordinator.yoto_manager.library[
-                        self.media_content_id
-                    ].chapters[self.player.chapter_key].tracks[self.player.track_key].icon,
+                            self.media_content_id
+                        ]
+                        .chapters[self.player.chapter_key]
+                        .tracks[self.player.track_key]
+                        .icon,
                     }
         else:
             return {}