Add fix in parsers

sigma67 · Jun 21, 2024 · ffc0eba · ffc0eba
1 parent 1d62e98
commit ffc0eba
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ytmusicapi/parsers/playlists.py b/ytmusicapi/parsers/playlists.py
@@ -38,8 +38,10 @@ def parse_playlist_header(response: Dict) -> Dict[str, Any]:
         playlist["duration"] = (
             None if not has_duration else second_subtitle_runs[has_views + has_duration]["text"]
         )
-        song_count = second_subtitle_runs[has_views + 0]["text"].split(" ")
-        song_count = to_int(song_count[0]) if len(song_count) > 1 else 0
+        song_count_text = second_subtitle_runs[has_views + 0]["text"]
+        song_count_search = re.search(r"\d+", song_count_text)
+        # extract the digits from the text, return 0 if no match
+        song_count = to_int(song_count_search.group()) if song_count_search is not None else 0
         playlist["trackCount"] = song_count
 
     return playlist