trackdb: Calculate a track's decade

Signed-off-by: Anna Schumaker <Anna@NoWheyCreamery.com>
This commit is contained in:
Anna Schumaker 2020-10-18 13:38:27 -04:00
parent 89b40ab905
commit c98cec1b28
2 changed files with 9 additions and 0 deletions

View File

@ -25,6 +25,12 @@ class TestTrack(unittest.TestCase):
trak2 = track.Track(2, test_tracks / "02 - Test {Disc 2}.ogg")
self.assertEqual(trak2.tags["artist"], "Test Album Artist")
def test_track_decade(self):
trak = track.Track(1, test_tracks / "01 - Test Track.ogg")
self.assertEqual(trak.decade(), 2010)
trak.tags["year"] = 1974
self.assertEqual(trak.decade(), 1970)
def test_track_discnumber(self):
trak = track.Track(1, test_tracks / "01 - Test Track.ogg")
self.assertEqual(trak.tags["discnumber"], 1)

View File

@ -36,6 +36,9 @@ class Track:
return ", ".join(self.tags["genre"])
return None if tag == None else str(tag)
def decade(self):
return (self.tags["year"] // 10) * 10
def genres(self):
for g in self.tags["genre"]:
yield g