diff --git a/tiddl/tests.py b/tiddl/tests.py index 80c1f30..e67ab97 100644 --- a/tiddl/tests.py +++ b/tiddl/tests.py @@ -134,6 +134,7 @@ def test_formatFilename(self): "videoCover": None, }, "mixes": {"TRACK_MIX": "001ec78dae0d4a470999adefffd570"}, + "playlistNumber": None } self.assertEqual(formatFilename("{title}", track), ("", "HAUTE COUTURE")) diff --git a/tiddl/utils.py b/tiddl/utils.py index 00ba514..50c1734 100644 --- a/tiddl/utils.py +++ b/tiddl/utils.py @@ -91,7 +91,7 @@ def formatFilename(template: str, track: Track, playlist=""): "playlist": playlist.strip(), "released": release_date.strftime("%m-%d-%Y"), "year": release_date.strftime("%Y"), - "playlist_number": str(track["playlistNumber"]) or "" + "playlist_number": str(track.get("playlistNumber", "")) } dirs = template.split("/")