deemix-py/deemix/utils/taggers.py

172 lines
6.4 KiB
Python
Raw Normal View History

2020-02-17 15:46:18 +00:00
#!/usr/bin/env python3
from mutagen.flac import FLAC, Picture
from mutagen.id3 import ID3, ID3NoHeaderError, TXXX, TIT2, TPE1, TALB, TPE2, TRCK, TPOS, TCON, TYER, TDAT, TLEN, TBPM, \
TPUB, TSRC, USLT, SYLT, APIC, IPLS, TCOM, TCOP, TCMP
2020-08-14 18:31:37 +00:00
# Adds tags to a MP3 file
def tagID3(stream, track, save):
2020-08-14 18:31:37 +00:00
# Delete exsisting tags
try:
tag = ID3(stream)
tag.delete()
except ID3NoHeaderError:
tag = ID3()
2020-02-17 15:46:18 +00:00
if save['title']:
2020-08-15 19:34:10 +00:00
tag.add(TIT2(text=track.title))
if save['artist'] and len(track.artists):
if save['multiArtistSeparator'] != "default":
if save['multiArtistSeparator'] == "nothing":
2020-08-15 19:34:10 +00:00
tag.add(TPE1(text=track.mainArtist['name']))
else:
2020-08-15 19:34:10 +00:00
tag.add(TPE1(text=track.artistsString))
tag.add(TXXX(desc="ARTISTS", text=track.artists))
else:
2020-08-15 19:34:10 +00:00
tag.add(TPE1(text=track.artists))
2020-08-14 18:31:37 +00:00
if save['album']:
2020-08-15 19:34:10 +00:00
tag.add(TALB(text=track.album['title']))
2020-08-14 18:31:37 +00:00
2020-08-15 19:34:10 +00:00
if save['albumArtist'] and len(track.album['artists']):
if save['singleAlbumArtist'] and track.album['mainArtist']['save']:
tag.add(TPE2(text=track.album['mainArtist']['name']))
else:
2020-08-15 19:34:10 +00:00
tag.add(TPE2(text=track.album['artists']))
2020-08-14 18:31:37 +00:00
if save['trackNumber']:
tag.add(TRCK(
2020-08-15 19:34:10 +00:00
text=str(track.trackNumber) + ("/" + str(track.album['trackTotal']) if save['trackTotal'] else "")))
if save['discNumber']:
tag.add(
2020-08-15 19:34:10 +00:00
TPOS(text=str(track.discNumber) + ("/" + str(track.album['discTotal']) if save['discTotal'] else "")))
if save['genre']:
2020-08-15 19:34:10 +00:00
tag.add(TCON(text=track.album['genre']))
if save['year']:
2020-08-15 19:34:10 +00:00
tag.add(TYER(text=str(track.date['year'])))
if save['date']:
2020-08-15 19:34:10 +00:00
tag.add(TDAT(text=str(track.date['month']) + str(track.date['day'])))
if save['length']:
2020-08-15 19:34:10 +00:00
tag.add(TLEN(text=str(int(track.duration)*1000)))
if save['bpm']:
2020-08-15 19:34:10 +00:00
tag.add(TBPM(text=str(track.bpm)))
if save['label']:
2020-08-15 19:34:10 +00:00
tag.add(TPUB(text=track.album['label']))
if save['isrc']:
2020-08-15 19:34:10 +00:00
tag.add(TSRC(text=track.ISRC))
if save['barcode']:
2020-08-15 19:34:10 +00:00
tag.add(TXXX(desc="BARCODE", text=track.album['barcode']))
if save['explicit']:
2020-08-15 19:34:10 +00:00
tag.add(TXXX(desc="ITUNESADVISORY", text="1" if track.explicit else "0"))
if save['replayGain']:
2020-08-15 19:34:10 +00:00
tag.add(TXXX(desc="REPLAYGAIN_TRACK_GAIN", text=track.replayGain))
2020-08-16 11:25:25 +00:00
if track.lyrics['unsync'] and save['lyrics']:
2020-08-15 19:34:10 +00:00
tag.add(USLT(text=track.lyrics['unsync']))
if track.lyrics['syncID3'] and save['syncedLyrics']:
tag.add(SYLT(3, format=2, type=1, text=track.lyrics['syncID3']))
2020-08-14 18:31:37 +00:00
involved_people = []
2020-08-15 19:34:10 +00:00
for role in track.contributors:
if role in ['author', 'engineer', 'mixer', 'producer', 'writer']:
2020-08-15 19:34:10 +00:00
for person in track.contributors[role]:
involved_people.append([role, person])
elif role == 'composer' and save['composer']:
2020-08-15 19:34:10 +00:00
tag.add(TCOM(text=track.contributors['composer']))
if len(involved_people) > 0 and save['involvedPeople']:
tag.add(IPLS(people=involved_people))
2020-08-14 18:31:37 +00:00
if save['copyright']:
2020-08-15 19:34:10 +00:00
tag.add(TCOP(text=track.copyright))
2020-08-15 21:03:05 +00:00
if save['savePlaylistAsCompilation'] and track.playlist:
tag.add(TCMP(text="1"))
2020-08-14 18:31:37 +00:00
2020-08-15 19:34:10 +00:00
if save['cover'] and track.album['picPath']:
with open(track.album['picPath'], 'rb') as f:
tag.add(
2020-09-10 11:13:58 +00:00
APIC(0, 'image/jpeg' if track.album['picPath'].endswith('jpg') else 'image/png', 3, desc='cover', data=f.read()))
2020-02-17 15:46:18 +00:00
tag.save(stream, v1=2 if save['saveID3v1'] else 0, v2_version=3,
v23_sep=None if save['useNullSeparator'] else '/')
2020-02-17 15:46:18 +00:00
2020-08-14 18:31:37 +00:00
# Adds tags to a FLAC file
def tagFLAC(stream, track, save):
2020-08-14 18:31:37 +00:00
# Delete exsisting tags
tag = FLAC(stream)
tag.delete()
tag.clear_pictures()
2020-08-14 18:31:37 +00:00
if save['title']:
2020-08-15 19:34:10 +00:00
tag["TITLE"] = track.title
2020-08-14 18:31:37 +00:00
2020-08-15 19:34:10 +00:00
if save['artist'] and len(track.artists):
if save['multiArtistSeparator'] != "default":
if save['multiArtistSeparator'] == "nothing":
2020-08-15 19:34:10 +00:00
tag["ARTIST"] = track.mainArtist['name']
else:
2020-08-15 19:34:10 +00:00
tag["ARTIST"] = track.artistsString
tag["ARTISTS"] = track.artists
else:
2020-08-15 19:34:10 +00:00
tag["ARTIST"] = track.artists
2020-08-14 18:31:37 +00:00
if save['album']:
2020-08-15 19:34:10 +00:00
tag["ALBUM"] = track.album['title']
2020-08-14 18:31:37 +00:00
2020-08-15 19:34:10 +00:00
if save['albumArtist'] and len(track.album['artists']):
if save['singleAlbumArtist']:
2020-08-15 19:34:10 +00:00
tag["ALBUMARTIST"] = track.album['mainArtist']['name']
else:
2020-08-15 19:34:10 +00:00
tag["ALBUMARTIST"] = track.album['artists']
2020-08-14 18:31:37 +00:00
if save['trackNumber']:
2020-08-15 19:34:10 +00:00
tag["TRACKNUMBER"] = str(track.trackNumber)
if save['trackTotal']:
2020-08-15 19:34:10 +00:00
tag["TRACKTOTAL"] = str(track.album['trackTotal'])
if save['discNumber']:
2020-08-15 19:34:10 +00:00
tag["DISCNUMBER"] = str(track.discNumber)
if save['discTotal']:
2020-08-15 19:34:10 +00:00
tag["DISCTOTAL"] = str(track.album['discTotal'])
if save['genre']:
2020-08-15 19:34:10 +00:00
tag["GENRE"] = track.album['genre']
if save['date']:
2020-08-15 19:34:10 +00:00
tag["DATE"] = track.dateString
elif save['year']:
2020-08-15 19:34:10 +00:00
tag["YEAR"] = str(track.date['year'])
if save['length']:
2020-08-15 19:34:10 +00:00
tag["LENGTH"] = str(track.duration)
if save['bpm']:
2020-08-15 19:34:10 +00:00
tag["BPM"] = str(track.bpm)
if save['label']:
2020-08-15 19:34:10 +00:00
tag["PUBLISHER"] = track.album['label']
if save['isrc']:
2020-08-15 19:34:10 +00:00
tag["ISRC"] = track.ISRC
if save['barcode']:
2020-08-15 19:34:10 +00:00
tag["BARCODE"] = track.album['barcode']
if save['explicit']:
2020-08-15 19:34:10 +00:00
tag["ITUNESADVISORY"] = "1" if track.explicit else "0"
if save['replayGain']:
2020-08-15 19:34:10 +00:00
tag["REPLAYGAIN_TRACK_GAIN"] = track.replayGain
2020-08-16 11:25:25 +00:00
if track.lyrics['unsync'] and save['lyrics']:
2020-08-15 19:34:10 +00:00
tag["LYRICS"] = track.lyrics['unsync']
2020-08-14 18:31:37 +00:00
2020-08-15 19:34:10 +00:00
for role in track.contributors:
if role in ['author', 'engineer', 'mixer', 'producer', 'writer', 'composer']:
if save['involvedPeople'] and role != 'composer' or role == 'composer' and save['composer']:
2020-08-15 19:34:10 +00:00
tag[role] = track.contributors[role]
elif role == 'musicpublisher' and save['involvedPeople']:
2020-08-15 19:34:10 +00:00
tag["ORGANIZATION"] = track.contributors['musicpublisher']
2020-08-14 18:31:37 +00:00
if save['copyright']:
2020-08-15 19:34:10 +00:00
tag["COPYRIGHT"] = track.copyright
2020-08-15 21:03:05 +00:00
if save['savePlaylistAsCompilation'] and track.playlist:
tag["COMPILATION"] = "1"
2020-02-17 15:46:18 +00:00
2020-08-15 19:34:10 +00:00
if save['cover'] and track.album['picPath']:
image = Picture()
image.type = 3
2020-08-15 19:34:10 +00:00
image.mime = 'image/jpeg' if track.album['picPath'].endswith('jpg') else 'image/png'
with open(track.album['picPath'], 'rb') as f:
image.data = f.read()
tag.add_picture(image)
2020-02-17 15:46:18 +00:00
tag.save(deleteid3=True)