fixup! handle libsndfile error

beetbox · Sep 24, 2023 · 9e8830a · 9e8830a
1 parent f685231
commit 9e8830a
Showing 1 changed file with 9 additions and 3 deletions.
diff --git a/beetsplug/autobpm.py b/beetsplug/autobpm.py
@@ -22,6 +22,7 @@
 from beets.plugins import BeetsPlugin
 
 from librosa import load, beat
+from soundfile import LibsndfileError
 
 class AutoBPMPlugin(BeetsPlugin):
 
@@ -54,10 +55,15 @@ def calculate_bpm(self, items, write=False):
             if item['bpm'] and not overwrite:
                 continue
 
-            y, sr = load(util.syspath(item.path), res_type="kaiser_fast")
-            tempo, _ = beat.beat_track(y=y, sr=sr)
-            bpm = round(tempo)
+            try:
+                y, sr = load(util.syspath(item.path), res_type="kaiser_fast")
+                tempo, _ = beat.beat_track(y=y, sr=sr)
+            except LibsndfileError:
+                self._log.error('failed to load track for {0}',
+                                util.displayable_path(item.path))
+                continue
 
+            bpm = round(tempo)
             item['bpm'] = bpm
             self._log.info('added computed bpm {0} for {1}',
                            bpm, util.displayable_path(item.path))