fix to pass tests

[wolnelektury.git] / src / catalogue / models / book.py

diff --git a/src/catalogue/models/book.py b/src/catalogue/models/book.py
index 004c27e..7cd6386 100644 (file)
--- a/src/catalogue/models/book.py
+++ b/src/catalogue/models/book.py
@@ -233,7 +233,7 @@ class Book(models.Model):
     def get_audio_length(self):
         from mutagen.mp3 import MP3
         total = 0
-        for media in self.get_mp3():
+        for media in self.get_mp3() or ():
             audio = MP3(media.file.path)
             total += audio.info.length
         return int(total)