parse_seach_result: add playlist id to album search

sigma67 · May 21, 2024 · 96bf541 · 96bf541
1 parent 0cf0883
commit 96bf541
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/ytmusicapi/parsers/search.py b/ytmusicapi/parsers/search.py
@@ -159,6 +159,13 @@ def parse_search_result(data, search_result_types, result_type, category):
     if result_type in ["song", "album"]:
         search_result["isExplicit"] = nav(data, BADGE_LABEL, True) is not None
 
+    if result_type in ["album"]:
+        search_result["playlistId"] = nav(
+            data,
+            [*PLAY_BUTTON, "playNavigationEndpoint", "watchEndpoint", "playlistId"],
+            True,
+        )
+
     if result_type in ["episode"]:
         flex_item = get_flex_column_item(data, 1)
         has_date = int(len(nav(flex_item, TEXT_RUNS)) > 1)