Make song result dependent on the song having an album (some songs ha…

…ve no album)
sigma67 · Feb 23, 2020 · 1ac614e · 1ac614e
1 parent fcea964
commit 1ac614e
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ytmusicapi/helpers.py b/ytmusicapi/helpers.py
@@ -61,8 +61,10 @@ def parse_search_result(data, resultType = None):
         search_result['itemCount'] = get_item_text(data, 2 + default).split(' ')[0]
 
     elif resultType in ['song']:
-        search_result['album'] = get_item_text(data, 2 + default)
-        search_result['duration'] = get_item_text(data, 3 + default)
+        hasAlbum = len(data['flexColumns'] == 4)
+        if hasAlbum:
+            search_result['album'] = get_item_text(data, 2 + default)
+        search_result['duration'] = get_item_text(data, 2 + hasAlbum + default)
 
     elif resultType in ['video']:
         search_result['views'] = get_item_text(data, 2 + default).split(' ')[0]