X-Git-Url: https://diplodocus.org/git/flac-archive/blobdiff_plain/bbb47ac8a49e5cfa41d7fd3b5c584b76ff07ac4f..003b5f1c71fd698ad5f66478c170e9d90ea7a4ec:/fa-rip diff --git a/fa-rip b/fa-rip index eabf52e..4b7f76d 100755 --- a/fa-rip +++ b/fa-rip @@ -1,36 +1,273 @@ -#! /usr/bin/env zsh +#! /usr/bin/env python2.4 -# $Id$ +""" +=head1 NAME -set -e -setopt NULL_GLOB +B - rip a CD for B -get_cddev () { - local raw - raw=$(sysctl -n kern.rawpartition > /dev/null | awk '{printf "%c",97+$0}') +=head1 SYNOPSIS - for CDDEV in /dev/{cdroms/cdrom*,cdrom*,rcd*${raw},{a,}cd*c}; do - [[ -e ${CDDEV} ]] && return 0 - done +B [B<-d> I] [B<-p> I [B<-t> I] - return 1 -} +=head1 DESCRIPTION -if [[ -z ${CDDEV} ]]; then - if ! get_cddev; then - echo 'CDDEV environment variable not set, defaults did not work' - exit 2 - fi -fi +B creates a temporary directory for storage of its +intermediate files, uses MusicBrainz to create the "cue" file and +candidate tags files, and runs C to rip the CD to the +"wav" file. -dir=$(mktemp -d flac-archive.XXXXXXXXXX) -cd ${dir} +In order for this CD to be processed by B, you must create a +"tags" file. This is usually done by renaming one of the +candidate-tags files and deleting the others. Don't forget to fill in +the DATE tag in the selected candidate before renaming it. If +B could not find any tag information from MusicBrainz, you'll +have to fill out the candidate-tags-0 template. -cdrdao read-toc --device ${CDDEV} --driver generic-mmc toc -toc2cue toc cue +=head1 OPTIONS -trackcount=$(grep -c 'TRACK.*AUDIO' cue) +=over 4 -fa-tags ${trackcount} +=item B<-d> [B<--device>] I -exec cdparanoia -d ${CDDEV} 1-$(grep -c 'TRACK.*AUDIO' cue) wav +Use I as the CD-ROM device, instead of the default +"/dev/cdrom" or the environment variable CDDEV. + +=item B<-p> [B<--post-processor>] I + +Create a "post-processor" file in the temporary directory containing +the line 'I "$@"'. See B's man page for +information about this hook. + +=item B<-t> [B<--tracks>] I + +Archive only the first I tracks. This is handy for +ignoring data tracks. + +=back + +=head1 ENVIRONMENT + +=over 4 + +=item CDDEV + +B uses this to rip audio and save the cuesheet for a CD. +MusicBrainz::Client can usually figure this out automatically. + +=back + +=head1 AUTHORS + +Written by Eric Gillespie . + +flac-archive is free software; you may redistribute it and/or modify +it under the same terms as Perl itself. + +=cut + +""" + +import os, re, sys, tempfile, traceback +from optparse import OptionParser +import urllib + +import musicbrainz2.disc +import musicbrainz2.webservice + +from org.diplodocus.util import catch_EnvironmentError as c + +def mkcue(disc, trackcount=None): + fp = c(file, 'cue', 'w') + c(fp.write, 'FILE "dummy.wav" WAVE\n') + c(fp.write, ' TRACK 01 AUDIO\n') + c(fp.write, ' INDEX 01 00:00:00\n') + + if trackcount == None: + trackcount = disc.lastTrackNum + else: + trackcount = min(trackcount, disc.lastTrackNum) + + pregap = disc.tracks[0][0] + for i in xrange(disc.firstTrackNum, trackcount): + offset = disc.tracks[i][0] + offset -= pregap + + minutes = seconds = 0 + sectors = offset % 75 + if offset >= 75: + seconds = offset / 75 + if seconds >= 60: + minutes = seconds / 60 + seconds = seconds % 60 + + c(fp.write, ' TRACK %02d AUDIO\n' % (i + 1,)) + c(fp.write, + ' INDEX 01 %02d:%02d:%02d\n' % (minutes, seconds, sectors)) + + c(fp.close) + + return trackcount + +def tags_file(fn, trackcount, various, artist=None, album=None, + release_dates={}, tracks=[]): + fp = c(file, fn, 'w') + c(fp.write, 'ARTIST=') + if artist != None: + c(fp.write, artist.encode('utf-8')) + c(fp.write, '\nALBUM=') + if album != None: + c(fp.write, album.encode('utf-8')) + c(fp.write, '\n') + + have_date = False + for (country, date) in release_dates.items(): + have_date = True + c(fp.write, 'DATE[%s]=%s\n' % (country, date)) + have_date or c(fp.write, 'DATE=\n') + + if len(tracks) > 0: + trackcount = min(trackcount, len(tracks)) + for i in xrange(1, trackcount + 1): + try: + track = tracks.pop(0) + title = track.title + artist = track.artist + except IndexError: + title = '' + artist = '' + various and c(fp.write, 'ARTIST[%d]=%s\n' % (i, + artist.encode('utf-8'))) + c(fp.write, 'TITLE[%d]=%s\n' % (i, title.encode('utf-8'))) + + c(fp.close) + +def cover_art(i, asin): + url = 'http://images.amazon.com/images/P/%s.01.MZZZZZZZ.jpg' % (asin,) + fp = file('cover.front-' + i, 'w') + fp.write(urllib.urlopen(url).read()) + fp.close() + +def tags(q, releases, trackcount): + results = [] + seen_various = False + + tags_file('candidate-tags-0', trackcount, False) + + include = musicbrainz2.webservice.ReleaseIncludes(tracks=True) + + i = 0 + for album in releases: + i += 1 + various = not album.release.isSingleArtistRelease() + + if various and not seen_various: + seen_various = True + tags_file('candidate-tags-0v', trackcount, True) + + tags_file('candidate-tags-' + str(i), trackcount, various, + album.release.artist.name, album.release.title, + album.release.getReleaseEventsAsDict(), + q.getReleaseById(album.release.id, include).tracks) + + cover_art(str(i), album.release.asin) + +def rip(device, trackcount, single_file): + if device == None: + device = '/dev/cdrom' + + argv = ['cdparanoia', '-d', device, '1-' + str(trackcount)] + + if single_file: + argv.append('wav') + else: + argv.append('-B') + + c(os.execvp, argv[0], argv) + +def make_post_processor(command): + if command == None: + return + + fd = c(os.open, 'post-processor', os.O_CREAT | os.O_WRONLY, 0555) + fp = c(os.fdopen, fd, 'w') + c(fp.write, command +' "$@"\n') + c(fp.close) + +def releases_by_disc(q, disc): + filter = musicbrainz2.webservice.ReleaseFilter(discId=disc.getId()) + return q.getReleases(filter) + +def releases_by(q, title, artist=None): + r = q.getReleases(musicbrainz2.webservice.ReleaseFilter(title=title)) + if artist == None: + return r + + artist = re.sub(r'\s+', r'\s+', artist.strip()) + return [x for x in r if re.match(artist, x.release.artist.name, + re.IGNORECASE) != None] + +def main(argv): + # Control the exit code for any uncaught exceptions. + try: + parser = OptionParser() + parser.disable_interspersed_args() + parser.add_option('--artist') + parser.add_option('--title') + parser.add_option('-d', '--device') + parser.add_option('-m', '--no-musicbrainz', + action='store_true', default=False) + parser.add_option('-p', '--post-processor') + parser.add_option('-s', '--single-file', + action='store_true', default=False) + parser.add_option('-t', '--tracks', type='int', default=99) + except: + traceback.print_exc() + return 2 + + try: + # Raises SystemExit on invalid options in argv. + (options, args) = parser.parse_args(argv[1:]) + except Exception, error: + if isinstance(error, SystemExit): + return 1 + traceback.print_exc() + return 2 + + try: + device = options.device + trackcount = options.tracks + + tempdir = c((lambda x: tempfile.mkdtemp(prefix=x, dir='.')), + 'flac-archive.') + c(os.chdir, tempdir) + + make_post_processor(options.post_processor) + + q = musicbrainz2.webservice.Query() + if options.title != None: + releases = releases_by(q, options.title, options.artist) + else: + disc = musicbrainz2.disc.readDisc(device) + trackcount = mkcue(disc, trackcount) + if options.no_musicbrainz: + releases = [] + else: + releases = releases_by_disc(q, disc) + + tags(q, releases, trackcount) + + if options.title == None: + rip(device, trackcount, options.single_file) + except Exception, error: + if isinstance(error, SystemExit): + raise + # check all print_exc and format_exc in fa-flacd.py; i think + # for some i don't do this msg print check + sys.stderr.write(getattr(error, 'msg', '')) + traceback.print_exc() + return 2 + + return 0 + +if __name__ == '__main__': + sys.exit(main(sys.argv))