fix sum_total_duration failing if duration_seconds is not an int (#598)

* fix sum_total_duration failing if duration_seconds is not an int Signed-off-by: Adwait Thattey <[email protected]> * format --------- Signed-off-by: Adwait Thattey <[email protected]> Co-authored-by: sigma67 <[email protected]>
sigma67 · Jul 3, 2024 · ab2aa28 · ab2aa28
1 parent d2cbe53
commit ab2aa28
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/ytmusicapi/helpers.py b/ytmusicapi/helpers.py
@@ -71,4 +71,11 @@ def to_int(string):
 def sum_total_duration(item):
     if "tracks" not in item:
         return 0
-    return sum([track["duration_seconds"] if "duration_seconds" in track else 0 for track in item["tracks"]])
+    return sum(
+        [
+            track["duration_seconds"]
+            if ("duration_seconds" in track and isinstance(track["duration_seconds"], int))
+            else 0
+            for track in item["tracks"]
+        ]
+    )