From 31f0a95d8a4638dbdd3d5a58fdb4959f8e379e13 Mon Sep 17 00:00:00 2001
From: Marvin Schenkel <marvinschenkel@gmail.com>
Date: Wed, 6 Jul 2022 16:04:35 +0200
Subject: [PATCH] Add library playlist, library songs and artist albums

---
 examples/ytmusic.py                        |  4 ++--
 music_assistant/music_providers/ytmusic.py | 27 +++++++++++-----------
 2 files changed, 16 insertions(+), 15 deletions(-)

diff --git a/examples/ytmusic.py b/examples/ytmusic.py
index e6247f48..1701c8d0 100644
--- a/examples/ytmusic.py
+++ b/examples/ytmusic.py
@@ -132,8 +132,8 @@ async def main():
         await mass.music.start_sync(schedule=3)
         artists = await mass.music.artists.count()
         print(f"Got {artists} artists in library")
-        albums = await mass.music.albums.count()
-        print(f"Got {albums} albums in library")
+        # albums = await mass.music.albums.count()
+        # print(f"Got {albums} albums in library")
         tracks = await mass.music.tracks.count()
         print(f"Got {tracks} tracks in library")
         playlists = await mass.music.playlists.library()
diff --git a/music_assistant/music_providers/ytmusic.py b/music_assistant/music_providers/ytmusic.py
index 2f7c2cb7..574f88a6 100644
--- a/music_assistant/music_providers/ytmusic.py
+++ b/music_assistant/music_providers/ytmusic.py
@@ -254,7 +254,7 @@ class YTMusic(MusicProvider):
                 item_id=str(item["videoId"]),
                 provider=self.type,
                 name=item["title"],
-                duration=item["duration"],
+                duration=item["duration_seconds"],
             )
             artists = []
             for artist in item["artists"]:
@@ -272,18 +272,19 @@ class YTMusic(MusicProvider):
                 MediaItemImage(ImageType.THUMB, thumb["url"])
                 for thumb in item["thumbnails"]
             ]
-            album = Album(
-                item_id=str(item["album"]["id"]),
-                name=item["album"]["name"],
-                provider=self.type,
-            )
-            album.add_provider_id(
-                MediaItemProviderId(
-                    item_id=str(item["album"]["id"]),
-                    prov_type=self.type,
-                    prov_id=self.id,
-                )
-            )
+            album = await self.get_album(item["album"]["id"])
+            # album = Album(
+            #     item_id=str(item["album"]["id"]),
+            #     name=item["album"]["name"],
+            #     provider=self.type,
+            # )
+            # album.add_provider_id(
+            #     MediaItemProviderId(
+            #         item_id=str(item["album"]["id"]),
+            #         prov_type=self.type,
+            #         prov_id=self.id,
+            #     )
+            # )
             track.album = album
             track.metadata.explicit = item["isExplicit"]
             track.add_provider_id(
-- 
2.34.1