]> diplodocus.org Git - flac-archive/blobdiff - fa-rip
Update CHANGES, list another dependency in README.
[flac-archive] / fa-rip
diff --git a/fa-rip b/fa-rip
index 55b2fb26791cf7d69cd1f59afaa159ae594c8df7..33fcc0d5ad65ad5cf2b8c58f001b9cb991769653 100755 (executable)
--- a/fa-rip
+++ b/fa-rip
-#! /usr/bin/env zsh
+#! /usr/bin/env python2.4
 
-# $Id$
+'''
+=head1 NAME
 
-set -e
-setopt NULL_GLOB
+B<fa-rip> - rip a CD for B<fa-flacd>
 
-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<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]
 
-    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<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.
 
-discid=($(cd-discid ${CDDEV}))
+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.
 
-[[ -d $discid[1] ]] || mkdir $discid[1]
-cd $discid[1]
+=head1 OPTIONS
 
-eval fa-tags $discid &
+=over 4
 
-cdrdao read-toc --device ${CDDEV} --driver generic-mmc toc
-toc2cue toc cue &
+=item B<-d> [B<--device>] I<device>
 
-exec cdparanoia -d ${CDDEV} 1-$discid[2] wav
+Use I<device> as the CD-ROM device, instead of the default
+"/dev/cdrom" or the environment variable CDDEV.
+
+=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<-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
+
+''' #' # python-mode is sucks
+
+import os, 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)
+    c(fp.write, '\nALBUM=')
+    if album != None:
+        c(fp.write, album)
+    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')
+
+    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))
+        c(fp.write, 'TITLE[%d]=%s\n' % (i, title))
+
+    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(disc, trackcount, mb=True):
+    results = []
+    seen_various = False
+
+    tags_file('candidate-tags-0', trackcount, False)
+
+    if not mb:
+        return
+
+    include = musicbrainz2.webservice.ReleaseIncludes(tracks=True)
+    q = musicbrainz2.webservice.Query()
+    filter = musicbrainz2.webservice.ReleaseFilter(discId=disc.getId())
+
+    i = 0
+    for album in q.getReleases(filter):
+        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 main(argv):
+    # Control the exit code for any uncaught exceptions.
+    try:
+        parser = OptionParser()
+        parser.disable_interspersed_args()
+        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)
+
+        disc = musicbrainz2.disc.readDisc(device)
+
+        trackcount = mkcue(disc, trackcount)
+        tags(disc, trackcount, mb=not options.no_musicbrainz)
+        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))