X-Git-Url: https://diplodocus.org/git/flac-archive/blobdiff_plain/af28e6fa66b0592c23744868ee174dba8b1c8fd5..3e8794c0d39a363cd6da9f6a29e0b07fb8ede9ea:/fa-rip

diff --git a/fa-rip b/fa-rip
index 22140a5..92b709d 100755
--- a/fa-rip
+++ b/fa-rip
@@ -1,8 +1,6 @@
-#! /usr/bin/env perl
-
-# $Id$
-# $URL$
+#! /usr/bin/env python2.4
 
+'''
 =head1 NAME
 
 B<fa-rip> - rip a CD for B<fa-flacd>
@@ -11,214 +9,6 @@ B<fa-rip> - rip a CD for B<fa-flacd>
 
 B<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]
 
-=cut
-
-use strict;
-use warnings;
-
-use Env qw(
-           CDDEV
-);
-
-use Fcntl qw(O_CREAT O_WRONLY);
-use File::Temp;
-use Getopt::Long qw(:config gnu_getopt no_ignore_case);
-use List::Util qw(min);
-use POSIX ':sys_wait_h';
-use Pod::Usage;
-
-use MusicBrainz::Client;
-use MusicBrainz::Queries qw(
-                            MBQ_GetCDTOC
-                            MBE_TOCGetFirstTrack
-                            MBE_TOCGetLastTrack
-                            MBE_TOCGetTrackSectorOffset
-);
-use MusicBrainz::Client::Simple;
-
-sub mkcue {
-    my $device = shift;
-    my $trackcount = shift;
-
-    my $mb = MusicBrainz::Client->new;
-    defined($device) and $mb->set_device($device);
-
-    $mb->query(MBQ_GetCDTOC) or die($mb->get_query_error);
-
-    open(my $fh, '>cue') or die("open('>cue'): $!");
-    print($fh "FILE \"dummy.wav\" WAVE\n");
-    print($fh "  TRACK 01 AUDIO\n");
-    print($fh "    INDEX 01 00:00:00\n");
-
-    my $first = $mb->get_result_data(MBE_TOCGetFirstTrack) + 1;
-    $trackcount = min($trackcount, $mb->get_result_data(MBE_TOCGetLastTrack));
-    # There is frequently (always?) an offset of 150 sectors, so
-    # we'll subtract this offset from each track offset.
-    my $something = $mb->get_result_data1(MBE_TOCGetTrackSectorOffset, 2);
-
-    for my $track ($first .. $trackcount) {
-        my $off = $mb->get_result_data1(MBE_TOCGetTrackSectorOffset, $track+1);
-        $off -= $something;
-
-        my ($minutes,$seconds)=(0,0);
-        my $sectors = $off % 75;
-        if ($off >= 75) {
-            $seconds = $off / 75;
-            if ($seconds >= 60) {
-                $minutes = $seconds / 60;
-                $seconds = $seconds % 60;
-            }
-        }
-
-        printf($fh "  TRACK %02d AUDIO\n", $track);
-        printf($fh "    INDEX 01 %02d:%02d:%02d\n",
-               $minutes, $seconds, $sectors);
-    }
-
-    close($fh) or die("close(>cue): $!");
-
-    return $trackcount;
-}
-
-sub tags_file {
-    my $fn = shift;
-    my $trackcount = shift;
-    my $various = shift;
-    my $artist = shift;
-    my $album = shift;
-    my $release_dates = shift;
-    my $fh;
-    my $i;
-    my $track;
-    my $name;
-
-    open($fh, '>', $fn) or die("open('>$fn'): $!");
-    print($fh 'ARTIST=', (defined($artist) and $artist or ''), "\n");
-    print($fh 'ALBUM=', (defined($album) and $album or ''), "\n");
-
-    if (defined($release_dates) and %$release_dates) {
-        while (my ($country, $date) = each(%$release_dates)) {
-            print($fh "DATE[$country]=$date\n");
-        }
-    } else {
-        print($fh "DATE=\n");
-    }
-
-    for $i (1 .. $trackcount) {
-        $various and print($fh "ARTIST[$i]=\n");
-        if ($track = shift(@_)) {
-            $name = $track->get_name;
-        } else {
-            $name = '';
-        }
-        print($fh "TITLE[$i]=$name\n");
-    }
-
-    close($fh) or die("close(>$fn): $!");
-}
-
-sub tags {
-    my $device = shift;
-    my $trackcount = shift;
-    my $no_mb = shift;
-    my $mb;
-    my @results;
-    my $album;
-    my $i;
-    my $various;
-    my $seen_various;
-
-    tags_file('candidate-tags-0', $trackcount, 0);
-
-    defined($no_mb) and $no_mb and return;
-
-    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);
-    }
-
-    for $album (@results) {
-        $i++;
-
-        if ($various = $album->has_various_artists) {
-            if (not $seen_various) {
-                $seen_various = 1;
-                tags_file('candidate-tags-0v', $trackcount, 1);
-            }
-        }
-
-        my %dates = $album->get_release_dates;
-        tags_file("candidate-tags-$i", $trackcount, $various,
-                  $album->get_artist->get_name, $album->get_name,
-                  \%dates, $album->get_tracks);
-    }
-}
-
-sub rip {
-    my $device = shift;
-    my $trackcount = shift;
-    my $single_file = shift;
-    my @output;
-
-    $device ||= '/dev/cdrom';
-    if ($single_file) {
-        @output = ("1-$trackcount", 'wav');
-    } else {
-        @output = ('-B');
-    }
-
-    exec('cdparanoia', '-d', $device, @output);
-    die;
-}
-
-sub make_post_processor {
-    my $command = shift;
-
-    defined($command) or return;
-
-    sysopen(F, 'post-processor', O_CREAT | O_WRONLY, 0555)
-      or die("sysopen(post-processor, O_CREAT | O_WRONLY, 0555): $!");
-    print(F $command, ' "$@"', "\n");
-    close(F) or die("close(post-processor, O_CREAT | O_WRONLY, 0555): $!");
-}
-
-MAIN: {
-    my $no_mb;
-    my $post_processor;
-    my $single_file;
-    my $trackcount = 99;
-    my $help;
-    my $tempdir;
-
-    GetOptions(
-               'device|d=s' => \$CDDEV,
-               'no-musicbrainz|m' => \$no_mb,
-               'post-processor|p=s', \$post_processor,
-               'single-file|s' => \$single_file,
-               '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): $!");
-
-    make_post_processor($post_processor);
-    $trackcount = mkcue($CDDEV, $trackcount);
-    tags($CDDEV, $trackcount, $no_mb);
-    rip($CDDEV, $trackcount, $single_file);
-}
-
-
-__END__
-
 =head1 DESCRIPTION
 
 B<fa-rip> creates a temporary directory for storage of its
@@ -275,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:
+''' #' # python-mode is sucks
+
+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))