make parse_menu_playlists more error proof

sigma67 · May 27, 2024 · fdc2594 · fdc2594
1 parent bd51238
commit fdc2594
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/ytmusicapi/parsers/_utils.py b/ytmusicapi/parsers/_utils.py
@@ -4,7 +4,10 @@
 
 
 def parse_menu_playlists(data, result):
-    watch_menu = find_objects_by_key(nav(data, MENU_ITEMS), MNIR)
+    menu_items = nav(data, MENU_ITEMS, True)
+    if menu_items is None:
+        return
+    watch_menu = find_objects_by_key(menu_items, MNIR)
     for item in [_x[MNIR] for _x in watch_menu]:
         icon = nav(item, ICON_TYPE)
         if icon == "MUSIC_SHUFFLE":