71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
from mutagen import mp3, id3
|
|
from yt_dlp.postprocessor import PostProcessor
|
|
|
|
import genius
|
|
|
|
|
|
ENC_UTF8 = 3
|
|
|
|
|
|
class InfoYouTubePP(PostProcessor):
|
|
'''Generates YT Music fields in info if necessary.
|
|
Must be run before downloading and post-processing with
|
|
FFmpegExtractAudioPP and ID3YouTubePP, so use only with
|
|
when<="before_dl" ("pre_process" also suits, see yt_dlp.utils.POSTPROCESS_WHEN)
|
|
for correct file path and ID3 tags'''
|
|
|
|
def run(self, information):
|
|
|
|
if not 'track' in information:
|
|
information['track'] = information['title']
|
|
if not 'artist' in information:
|
|
information['artist'] = information['channel'].removesuffix(' - Topic')
|
|
|
|
if not 'album' in information:
|
|
try:
|
|
information['album'] = information['playlist'].removeprefix('Album \u2013 ')
|
|
except KeyError:
|
|
pass
|
|
|
|
if not 'track_number' in information:
|
|
try:
|
|
information['track_number'] = information['playlist_index']
|
|
except KeyError:
|
|
pass
|
|
# here was some code for disc_number,
|
|
# but it is even not in mutagen ID3 fields list
|
|
|
|
return [], information
|
|
|
|
|
|
class ID3TagsPP(PostProcessor):
|
|
'''Inserts ID3 tags after all PPs (for YT: InfoYouTubePP and FFmpegExtractAudioPP),
|
|
triggers searching and parsing lyrics from Genius'''
|
|
|
|
def run(self, information):
|
|
|
|
file = mp3.MP3(information['filepath'])
|
|
|
|
title = information['track']
|
|
artist = information['artist']
|
|
|
|
file['TIT2'] = id3.TIT2(encoding=ENC_UTF8, text=title)
|
|
file['TPE1'] = id3.TPE1(encoding=ENC_UTF8, text=artist)
|
|
if 'album' in information:
|
|
file['TALB'] = id3.TALB(encoding=ENC_UTF8, text=information['album'])
|
|
if 'release_year' in information:
|
|
file['TDRC'] = id3.TDRC(encoding=ENC_UTF8, text=str(information['release_year']))
|
|
if 'track_number' in information:
|
|
file['TRCK'] = id3.TRCK(encoding=ENC_UTF8, text=str(information['track_number']))
|
|
if 'genre' in information:
|
|
file['TCON'] = id3.TCON(encoding=ENC_UTF8, text=information['genre'])
|
|
|
|
try:
|
|
lyr_url = genius.search(title, artist)
|
|
file['USLT'] = id3.USLT(encoding=ENC_UTF8, text=genius.parse(lyr_url))
|
|
except:
|
|
pass
|
|
|
|
file.save()
|
|
|
|
return [], information
|