bandcamp-dl/bandcamp_dl/bandcamp.py

from bs4 import BeautifulSoup
import requests
from .jsobj import read_js_object


class Bandcamp:
    def parse(self, url):
        try:
            r = requests.get(url)
        except requests.exceptions.MissingSchema:
            return None

        if r.status_code is not 200:
            return None

        try:
            self.soup = BeautifulSoup(r.text, "lxml")
        except:
            self.soup = BeautifulSoup(r.text, "html.parser")

        album = {
            "tracks": [],
            "title": "",
            "artist": "",
            "full": False,
            "art": "",
            "date": ""
        }

        album_meta = self.extract_album_meta_data(r)

        album['artist'] = album_meta['artist']
        album['title'] = album_meta['title']
        album['date'] = album_meta['date']

        for track in album_meta['tracks']:
            track = self.get_track_meta_data(track)
            album['tracks'].append(track)

        album['full'] = self.all_tracks_available(album)
        album['art'] = self.get_album_art()

        return album

    def all_tracks_available(self, album):
        for track in album['tracks']:
            if track['url'] is None:
                return False

        return True

    def is_basestring(self, obj):
        if isinstance(obj, str) or isinstance(obj, bytes) or isinstance(obj, bytearray):
            return True
        return False

    def get_track_meta_data(self, track):
        new_track = {}
        if not self.is_basestring(track['file']):
            if 'mp3-128' in track['file']:
                new_track['url'] = track['file']['mp3-128']
        else:
            new_track['url'] = None

        new_track['duration'] = track['duration']
        new_track['track'] = track['track_num']
        new_track['title'] = track['title']
        if track['lyrics'] != 'null':
            track['lyrics'] = track['lyrics'].encode(encoding='UTF-8', errors='ignore')
            new_track['lyrics'] = track['lyrics'].replace('\\r\\n', '\n')

        return new_track

    def extract_album_meta_data(self, request):
        album = {}

        embedData = self.get_embed_string_block(request)

        block = request.text.split("var TralbumData = ")

        stringBlock = block[1]

        stringBlock = stringBlock.split("};")[0] + "};"
        stringBlock = read_js_object("var TralbumData = {}".format(stringBlock))

        if 'album_title' not in embedData['EmbedData']:
            album['title'] = "Unknown Album"
        else:
            album['title'] = embedData['EmbedData']['album_title']

        album['artist'] = stringBlock['TralbumData']['artist']
        album['tracks'] = stringBlock['TralbumData']['trackinfo']

        if stringBlock['TralbumData']['album_release_date'] == "null":
            album['date'] = ""
        else:
            album['date'] = stringBlock['TralbumData']['album_release_date'].split()[2]

        return album

    @staticmethod
    def generate_album_url(artist, album):
        return "http://{0}.bandcamp.com/album/{1}".format(artist, album)

    def get_album_art(self):
        try:
            url = self.soup.find(id='tralbumArt').find_all('img')[0]['src']
            return url
        except:
            pass

    def get_embed_string_block(self, request):
        embedBlock = request.text.split("var EmbedData = ")

        embedStringBlock = embedBlock[1]
        embedStringBlock = embedStringBlock.split("};")[0] + "};"
        embedStringBlock = read_js_object("var EmbedData = {}".format(embedStringBlock))

        return embedStringBlock