52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
# Copyright 2021 (c) Anna Schumaker.
|
|
import mutagen
|
|
import re
|
|
|
|
class Metadata:
|
|
def __init__(self, filepath):
|
|
self.path = filepath
|
|
self.file = None
|
|
|
|
def __enter__(self):
|
|
self.file = mutagen.File(self.path)
|
|
return self
|
|
|
|
def __exit__(self, exp_type, exp_value, traceback):
|
|
self.file = None
|
|
return exp_type == None
|
|
|
|
def album(self):
|
|
return self.file.get("album", [ "Unknown Album" ])[0]
|
|
|
|
def artist(self):
|
|
artist = self.file.get("artist", [ "Unknown Artist" ])
|
|
return self.file.get("albumartist", artist)[0]
|
|
|
|
def artistsort(self):
|
|
sort = self.file.get("artistsort", [ None ])
|
|
return self.file.get("albumartistsort", sort)[0]
|
|
|
|
def decade(self):
|
|
return (self.year() // 10) * 10
|
|
|
|
def discnumber(self):
|
|
return int(self.file.get("discnumber", [ 1 ])[0])
|
|
|
|
def genres(self):
|
|
genre = self.file.get("genre", [ "" ])[0]
|
|
return [ g.strip() for g in re.split(",|;|/|:", genre) ]
|
|
|
|
def length(self):
|
|
return int(self.file.info.length)
|
|
|
|
def title(self):
|
|
return self.file.get("title", [ "" ])[0]
|
|
|
|
def tracknumber(self):
|
|
return int(self.file.get("tracknumber", [ 0 ])[0])
|
|
|
|
def year(self):
|
|
year = self.file.get("date", [ "0" ])
|
|
year = self.file.get("originalyear", year)[0]
|
|
return int(re.match("\d+", year).group(0))
|