mirror-bandcamp-dl/bandcamp_dl/__main__.py

import argparse
import logging
import pathlib
import sys

from bandcamp_dl import __version__
from bandcamp_dl.bandcamp import Bandcamp
from bandcamp_dl.bandcampdownloader import BandcampDownloader
from bandcamp_dl import config

from urllib.parse import urlparse


def main():
    # parse config if found, else create it
    conf = config.Config()

    parser = argparse.ArgumentParser()
    parser.add_argument('URL', help="Bandcamp album/track URL", nargs="*")
    parser.add_argument('-v', '--version', action='store_true', help='Show version')
    parser.add_argument('-d', '--debug', action='store_true', help='Verbose logging', default=conf.debug)
    parser.add_argument('--artist', help="Specify an artist's slug to download their full discography.")
    parser.add_argument('--track', help="Specify a track's slug to download a single track. Must be used with --artist.")
    parser.add_argument('--album', help="Specify an album's slug to download a single album. Must be used with --artist.")
    parser.add_argument('--template', help=f"Output filename template, default: "
                        f"{conf.template.replace('%', '%%')}", default=conf.template)
    parser.add_argument('--base-dir', help='Base location of which all files are downloaded',
                        default=conf.base_dir)
    parser.add_argument('-f', '--full-album', help='Download only if all tracks are available',
                        action='store_true')
    parser.add_argument('-o', '--overwrite', action='store_true',
                        help=f'Overwrite tracks that already exist. Default is {conf.overwrite}.',
                        default=conf.overwrite)
    parser.add_argument('-n', '--no-art', help='Skip grabbing album art', action='store_true',
                        default=conf.no_art)
    parser.add_argument('-e', '--embed-lyrics', help='Embed track lyrics (If available)',
                        action='store_true', default=conf.embed_lyrics)
    parser.add_argument('-g', '--group', help='Use album/track Label as iTunes grouping',
                        action='store_true', default=conf.group)
    parser.add_argument('-r', '--embed-art', help='Embed album art (If available)',
                        action='store_true', default=conf.embed_art)
    parser.add_argument('--cover-quality', help='Set the cover art quality. 0 is source, 10 is album page (1200x1200), 16 is default embed (700x700).',
                        default=conf.cover_quality, type=int, choices=[0, 10, 16])
    parser.add_argument('--untitled-path-from-slug', help='For albums titled untitled, use the URL slug to generate the folder path.',
                        action='store_true', default=conf.untitled_path_from_slug)
    parser.add_argument('-y', '--no-slugify', action='store_true', default=conf.no_slugify,
                        help='Disable slugification of track, album, and artist names')
    parser.add_argument('-c', '--ok-chars', default=conf.ok_chars,
                        help=f'Specify allowed chars in slugify, default: {conf.ok_chars}')
    parser.add_argument('-s', '--space-char', help=f'Specify the char to use in place of spaces, '
                        f'default: {conf.space_char}', default=conf.space_char)
    parser.add_argument('-a', '--ascii-only', help='Only allow ASCII chars (北京 (capital of '
                        'china) -> bei-jing-capital-of-china)', action='store_true',
                        default=conf.ascii_only)
    parser.add_argument('-k', '--keep-spaces', help='Retain whitespace in filenames',
                        action='store_true', default=conf.keep_spaces)
    parser.add_argument('-x', '--case-convert', help=f'Specify the char case conversion logic, '
                        f'default: {conf.case_mode}', default=conf.case_mode, dest='case_mode',
                        choices=[config.CASE_LOWER, config.CASE_UPPER, config.CASE_CAMEL,
                        config.CASE_NONE]) 
    parser.add_argument('--no-confirm', help='Override confirmation prompts. Use with caution',
                        action='store_true', default=conf.no_confirm)
    parser.add_argument('--embed-genres', help='Embed album/track genres',
                        action='store_true', default=conf.embed_genres)
    parser.add_argument('--truncate-album', metavar='LENGTH', type=int, default=0,
                        help='Truncate album title to a maximum length. 0 for no limit.')
    parser.add_argument('--truncate-track', metavar='LENGTH', type=int, default=0,
                        help='Truncate track title to a maximum length. 0 for no limit.')


    arguments = parser.parse_args()
    if arguments.version:
        sys.stdout.write(f"bandcamp-dl {__version__}\n")
        return

    if arguments.debug:
        logging.basicConfig(level=logging.DEBUG)
    else:
        logging.basicConfig()
    logging_handle = "bandcamp-dl"
    logger = logging.getLogger(logging_handle)

    # TODO: Its possible to break bandcamp-dl temporarily by simply erasing a line in the config, catch this and warn.
    logger.debug(f"Config/Args: {arguments}")
    if not arguments.URL and not arguments.artist:
        parser.print_usage()
        sys.stderr.write(f"{pathlib.Path(sys.argv[0]).name}: error: the following arguments are "
                         f"required: URL or --artist\n")
        sys.exit(2)

    for arg, val in [('base_dir', config.USER_HOME), ('template', config.TEMPLATE),
                     ('ok_chars', config.OK_CHARS), ('space_char', config.SPACE_CHAR)]:
        if not getattr(arguments, arg):
            setattr(arguments, arg, val)
    bandcamp = Bandcamp()

    if arguments.artist and arguments.album:
        urls = Bandcamp.generate_album_url(arguments.artist, arguments.album, "album")
    elif arguments.artist and arguments.track:
        urls = Bandcamp.generate_album_url(arguments.artist, arguments.track, "track")
    elif arguments.artist:
        urls = Bandcamp.get_full_discography(bandcamp, arguments.artist, "music")
    else:
        urls = []
        for url in arguments.URL:
            parsed_url = urlparse(url)
            if parsed_url.netloc.endswith('.bandcamp.com') and (parsed_url.path == '/music' or parsed_url.path == '/' or parsed_url.path == ''):
                artist = parsed_url.netloc.split('.')[0]
                print(f"Found artist page, fetching full discography for: {artist}")
                urls.extend(bandcamp.get_full_discography(artist, "music"))
            else:
                urls.append(url)


    album_list = []

    for url in urls:
        if "/album/" not in url and "/track/" not in url:
            continue
        logger.debug("\n\tURL: %s", url)
        album_list.append(bandcamp.parse(url, not arguments.no_art, arguments.embed_lyrics, arguments.embed_genres,
                                         arguments.debug, arguments.cover_quality))

    for album in album_list:
        logger.debug(f" Album data:\n\t{album}")
        if arguments.full_album and not album['full']:
            print("Full album not available. Skipping ", album['title'], " ...")
            # Remove not-full albums BUT continue with the rest of the albums.
            album_list.remove(album)

    if arguments.URL or arguments.artist:
        logger.debug("Preparing download process..")
        for album in album_list:
            bandcamp_downloader = BandcampDownloader(arguments, album['url'])
            logger.debug("Initiating download process..")
            bandcamp_downloader.start(album)
            # Add a newline to stop prompt mangling
            print("")
    else:
        logger.debug(r" /!\ Something went horribly wrong /!\ ")


if __name__ == '__main__':
    main()