#!/usr/bin/python """ =head1 NAME B - rip a CD for B =head1 SYNOPSIS B [B<--artist> I B<--title> I] [B<-d> I<device>] [B<-m>] [B<-p> I<post-processor>] [B<-s>] [B<-t> I<track-count>] =head1 DESCRIPTION B<fa-rip> creates a temporary directory for storage of its intermediate files, uses MusicBrainz to create the "cue" file and candidate tags files, and runs C<cdparanoia(1)> to rip the CD to the "wav" file. In order for this CD to be processed by B<fa-flacd>, 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<fa-rip> could not find any tag information from MusicBrainz, you'll have to fill out the candidate-tags-0 template. =head1 OPTIONS =over 4 =item B<--artist> I<artist> B<--title> I<title> Write candidate-tags files based on I<artist> and album I<title>. Useful if you've already ripped wav files with some other program and just need to set things up for B<fa-flacd>. =item B<-d> [B<--device>] I<device> Use I<device> as the CD-ROM device, instead of the default "/dev/cdrom" or the environment variable CDDEV. =item B<-m> [B<--no-musicbrainz>] Don't connect to MusicBrainz, just write candidate-tags-0. =item B<-p> [B<--post-processor>] I<post-processor> Create a "post-processor" file in the temporary directory containing the line 'I<post-processor> "$@"'. See B<fa-flacd>'s man page for information about this hook. =item B<-s> [B<--single-file>] Rip whole disc to one wav file and configure B<fa-flacd> to encode it to one FLAC file with embedded cuesheet. =item B<-t> [B<--tracks>] I<track-count> Archive only the first I<track-count> tracks. This is handy for ignoring data tracks. =back =head1 ENVIRONMENT =over 4 =item CDDEV B<fa-rip> 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 <epg@pretzelnet.org>. 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 # http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=439790 MSF_OFFSET = 150 def mkcue(disc, trackcount=None): fp = c(file, 'cue', 'w') c(fp.write, 'FILE "dummy.wav" WAVE\n') if trackcount == None: trackcount = disc.lastTrackNum else: trackcount = min(trackcount, disc.lastTrackNum) for i in xrange(disc.firstTrackNum, trackcount+1): offset = disc.tracks[i-1][0] offset -= MSF_OFFSET 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,)) if i == 1 and offset > 0: c(fp.write, ' INDEX 00 00:00:00\n') 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') if various: c(fp.write, 'ALBUMARTIST=') else: 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(artist=True, tracks=True) i = 0 for rel_id in releases: i += 1 release = q.getReleaseById(rel_id, include) various = not release.isSingleArtistRelease() if various and not seen_various: seen_various = True tags_file('candidate-tags-0v', trackcount, various) tags_file('candidate-tags-' + str(i), trackcount, various, release.artist.name, release.title, release.getReleaseEventsAsDict(), release.tracks) # XXX Not sure if .asin is here after my change above; may need to # include urlRelations=True. See also: # for i in release.getRelations(): print i.type # http://musicbrainz.org/ns/rel-1.0#Wikipedia # ... # http://musicbrainz.org/ns/rel-1.0#AmazonAsin cover_art(str(i), 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()) # XXX had to change x.getId to x.release.getId on 10.04; is that an API # change in a newer version (handle both) or is this code I never tested? return (x.release.getId() for x in 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.getId() 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.') sys.stderr.write('ripping to %s\n\n' % (tempdir,)) 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))