X-Git-Url: https://diplodocus.org/git/flac-archive/blobdiff_plain/c069d00b1502aa08be2bdf0a197ab1110ba350d5..003b5f1c71fd698ad5f66478c170e9d90ea7a4ec:/fa-rip

diff --git a/fa-rip b/fa-rip
index 1a9bc2a..4b7f76d 100755
--- a/fa-rip
+++ b/fa-rip
@@ -1,171 +1,20 @@
-#! /usr/bin/env perl
-
-# $Id$
-# $URL$
+#! /usr/bin/env python2.4
 
+"""
 =head1 NAME
 
 B<fa-rip> - rip a CD for B<fa-flacd>
 
 =head1 SYNOPSIS
 
-B<fa-rip> [B<-d> I<device>] [B<-t> I<track-count>]
-
-=cut
-
-use strict;
-use warnings;
-
-use Env qw(
-           CDDEV
-);
-
-use File::Temp;
-use Getopt::Long qw(:config gnu_getopt no_ignore_case);
-use POSIX ':sys_wait_h';
-use Pod::Usage;
-
-use MusicBrainz::Client::Simple;
-
-sub run_or_die {
-    my $command = shift;
-    my $status;
-
-    $status = system($command);
-
-    if (WIFEXITED($status)) {
-        if (($status = WEXITSTATUS($status)) != 0) {
-            die("$command exited with status $status");
-        }
-    } elsif (WIFSIGNALED($status)) {
-        die("$command killed with signal ", WTERMSIG($status));
-    } elsif (WIFSTOPPED($status)) {
-        die("$command stopped with signal ", WSTOPSIG($status));
-    } else {
-        die("Major horkage on system($command): \$? = $? \$! = $!");
-    }
-}
-
-sub mkcue {
-    my $device = shift;
-    my $trackcount = shift;
-    my @command;
-
-    push(@command, 'mkcue');
-
-    if (defined($trackcount)) {
-        push(@command, "-t $trackcount");
-    }
-
-    if (defined($device)) {
-        push(@command, $device);
-    }
-
-    push(@command, '> cue');
-    run_or_die(join(' ', @command));
-
-    if (not defined($trackcount)) {
-        open(F, 'cue') or die("open(cue): $!");
-        $trackcount = grep(/TRACK.*AUDIO/, <F>);
-        close(F);
-    }
-
-    return $trackcount;
-}
-
-sub tags {
-    my $device = shift;
-    my $trackcount = shift;
-    my $mb;
-    my @results;
-    my $album;
-    my $i;
-    my @tracks;
-    my $name;
-    my $track;
-    my $j;
-
-    if (defined($device)) {
-        $mb = new MusicBrainz::Client::Simple (device=>$device);
-    } else {
-        $mb = new MusicBrainz::Client::Simple;
-    }
-
-    @results = $mb->lookup_cd;
-    if (not $mb->success) {
-        die($mb->get_error);
-    }
-
-    open(F, '>candidate-tags-0') or die("open('>candidate-tags-0'): $!");
-    print(F "$_=\n") for ('ARTIST', 'ALBUM', 'DATE');
-    print(F "TITLE=\n") for 1 .. $trackcount;
-    close(F) or die("close('>candidate-tags-0'): $!");
-
-    for $album (@results) {
-        $i++;
-        open(F, ">candidate-tags-$i") or die("open(>candidate-tags-$i): $!");
-
-        print(F 'ARTIST=', $album->get_artist->get_name, "\n");
-        print(F 'ALBUM=', $album->get_name, "\n");
-
-        # MusicBrainz doesn't have dates yet; these are usually wrong anyway.
-        print(F "DATE=\n");
-
-        @tracks = $album->get_tracks;
-        for $j (1 .. $trackcount) {
-            if ($track = shift(@tracks)) {
-                $name = $track->get_name;
-            } else {
-                $name = '';
-            }
-            print(F "TITLE=$name\n");
-        }
-
-        close(F) or die("close(>candidate-tags-$i): $!");
-    }
-}
-
-sub rip {
-    my $device = shift;
-    my $trackcount = shift;
-
-    $device ||= '/dev/cdrom';
-
-    exec('cdparanoia', '-d', $device, "1-$trackcount", 'wav');
-    # exec prints its own error message so just
-    die;
-}
-
-MAIN: {
-    my $trackcount;
-    my $help;
-    my $tempdir;
-
-    GetOptions(
-               'device|d=s' => \$CDDEV,
-               'tracks|t=i' => \$trackcount,
-               'help|h|?' => \$help,
-              ) or pod2usage();
-    $help and pod2usage(-exitstatus=>0, -verbose=>1);
-
-    # File::Temp::tempdir calls die on error.
-    $tempdir = File::Temp::tempdir('flac-archive.XXXXXXXXXX');
-    chdir($tempdir) or die("chdir($tempdir): $!");
-
-    $trackcount = mkcue($CDDEV, $trackcount);
-    tags($CDDEV, $trackcount);
-    rip($CDDEV, $trackcount);
-}
-
-
-__END__
+B<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]
 
 =head1 DESCRIPTION
 
 B<fa-rip> creates a temporary directory for storage of its
-intermediate files, runs C<mkcue(1)> to create the "cue" file, uses
-MusicBrainz to generate candidate tags files, and runs
-C<cdparanoia(1)> to rip the CD to the "wav" file.
+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
@@ -183,6 +32,12 @@ have to fill out the candidate-tags-0 template.
 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
@@ -196,9 +51,8 @@ ignoring data tracks.
 
 =item CDDEV
 
-B<fa-rip> uses this to rip audio and save the cuesheet for a CD.  It
-makes some effort to check some common device names for FreeBSD,
-Linux, and NetBSD by default.
+B<fa-rip> uses this to rip audio and save the cuesheet for a CD.
+MusicBrainz::Client can usually figure this out automatically.
 
 =back
 
@@ -211,10 +65,209 @@ it under the same terms as Perl itself.
 
 =cut
 
-# Local variables:
-# cperl-indent-level: 4
-# perl-indent-level: 4
-# indent-tabs-mode: nil
-# End:
-
-# vi: set tabstop=4 expandtab:
+"""
+
+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))