From f7cfd10d4b7ea84370343bc18aeb2a3da46281e5 Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Tue, 4 Jun 2024 23:42:02 +0000
Subject: [PATCH] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 custom_components/yoto/media_player.py | 29 +++++++++++++++++---------
 1 file changed, 19 insertions(+), 10 deletions(-)

diff --git a/custom_components/yoto/media_player.py b/custom_components/yoto/media_player.py
index e979b1c..1abe0da 100644
--- 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 {}