metallum.py

#!/usr/bin/env python

import argparse, json, re, sys
try:
    import urllib2 as urlreq
    from urllib import urlencode
except:
    import urllib.request as urlreq
    from urllib.parse import urlencode

base_url = 'https://www.metal-archives.com/'
url_search_songs = 'search/ajax-advanced/searching/songs?'
url_lyrics = 'release/ajax-view-lyrics/id/'
lyrics_not_available = '(lyrics not available)'
lyric_id_re = re.compile(r'id=.+[a-z]+.(?P<id>\d+)')
band_name_re = re.compile(r'title="(?P<name>.*)\"')
tags_re = re.compile(r'<[^>]+>')
headers = {'User-Agent': 'Mozilla/5.0 (X11; Linux x86_64; rv:95.0) Gecko/20100101 Firefox/95.0'}

def get_songs_data(band_name, song_title):
    """Search on metal-archives for song coincidences"""
    params = dict(bandName = band_name, songTitle = song_title)
    url = base_url + url_search_songs + urlencode(params)
    req = urlreq.Request(url, headers=headers)
    return json.load(urlreq.urlopen(req))['aaData']

def get_lyrics_by_song_id(song_id):
    """Search on metal-archives for lyrics based on song_id"""
    url = base_url + url_lyrics + song_id
    req = urlreq.Request(url, headers=headers)
    return tags_re.sub('', urlreq.urlopen(req).read().strip().decode('utf-8'))

def iterate_songs_and_print(songs, args):
    '''Iterate over returned song matches. If the lyrics are different than\
    "(lyrics not available)" then break the loop and print them out.\
    Otherwise the last song of the list will be printed.'''
    for song in songs:
        band_name = band_name_re.search(song[0]).group("name")
        song_title = song[3]
        song_id = lyric_id_re.search(song[4]).group("id")
        lyrics = get_lyrics_by_song_id(song_id)
        if lyrics != lyrics_not_available:
            break

    title = band_name + " - " + song_title + "\n"

    if args.notitle:
        sys.exit("\033[0m" + lyrics + "\n")
    else:
        underline = "\033[4m"
        sys.exit(underline + title + "\n\033[0m" + lyrics + "\n")

def main():
    """Runs the program and handles command line options"""
    parser = argparse.ArgumentParser(description='Fetch lyrics from https://metal-archives.com')
    parser.add_argument('-t', '--notitle', action="store_true", help="don't show the artist and song names before the lyrics")
    parser.add_argument('band_name', type=str, help='The name of the band. e.g.: "Dark Reality"')
    parser.add_argument('song_title', type=str, help='The title of the song. e.g.: "Mopin Carol"')
    args = parser.parse_args()

    try:
        songs_data = get_songs_data(args.band_name, args.song_title)
        if len(songs_data):
            iterate_songs_and_print(songs_data, args)
    except Exception as e:
        sys.exit(e)

    sys.exit("\n\033[031m Lyrics not found\n")

if __name__ == '__main__':
    main()