-#! /usr/bin/env python2.4
+#!/usr/bin/python
-'''
+"""
=head1 NAME
B<fa-rip> - rip a CD for B<fa-flacd>
=head1 SYNOPSIS
-B<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]
+B<fa-rip> [B<--artist> I<artist> B<--title> I<title>] [B<-d> I<device>] [B<-m>] [B<-p> I<post-processor>] [B<-s>] [B<-t> I<track-count>]
=head1 DESCRIPTION
=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
=cut
-''' #' # python-mode is sucks
+"""
-import os, sys, tempfile, traceback
+import os, re, sys, tempfile, time, traceback
from optparse import OptionParser
+import urllib
-import musicbrainz2.disc
-import musicbrainz2.webservice
+import discid.disc
+import musicbrainzngs.musicbrainz
from org.diplodocus.util import catch_EnvironmentError as c
-def mkcue(disc, trackcount=None):
- fp = c(file, 'cue', 'w')
+musicbrainzngs.musicbrainz.set_useragent(
+ 'flac-archive', '0.1', 'https://diplodocus.org/git/flac-archive')
+
+#import logging
+#logging.basicConfig(level=logging.DEBUG)
+
+# http://bugs.debian.org/cgi-bin/bugreport.cgi?bug=439790
+MSF_OFFSET = 150
+
+def mkcue(fp, disc, trackcount=None):
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
+ trackcount = len(disc.tracks)
else:
- trackcount = min(trackcount, disc.lastTrackNum)
+ trackcount = min(trackcount, len(disc.tracks))
- pregap = disc.tracks[0][0]
- for i in xrange(disc.firstTrackNum, trackcount):
- offset = disc.tracks[i][0]
- offset -= pregap
+ for i in xrange(1, trackcount+1):
+ track = disc.tracks[i-1]
+ offset = track.offset - MSF_OFFSET
minutes = seconds = 0
sectors = offset % 75
minutes = seconds / 60
seconds = seconds % 60
- c(fp.write, ' TRACK %02d AUDIO\n' % (i + 1,))
+ c(fp.write, ' TRACK %02d AUDIO\n' % (track.number,))
+ 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')
- c(fp.write, 'ARTIST=')
+ if various:
+ c(fp.write, 'ALBUMARTIST=')
+ else:
+ c(fp.write, 'ARTIST=')
if artist != None:
- c(fp.write, artist)
+ c(fp.write, artist.encode('utf-8'))
c(fp.write, '\nALBUM=')
if album != None:
- c(fp.write, album)
+ c(fp.write, album.encode('utf-8'))
c(fp.write, '\n')
have_date = False
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:
+ artist = title = ''
+ if len(tracks) > 0:
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))
+ if track.artist:
+ artist = track.artist.name
+ 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 tags(disc, trackcount, mb=True):
+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(releases, trackcount):
results = []
+ seen_asins = set()
seen_various = False
tags_file('candidate-tags-0', trackcount, False)
- if not mb:
- return
-
- include = musicbrainz2.webservice.ReleaseIncludes(artist=True, tracks=True)
- q = musicbrainz2.webservice.Query()
- filter = musicbrainz2.webservice.ReleaseFilter(discId=disc.getId())
-
i = 0
- for album in q.getReleases(filter):
+ for release in releases:
i += 1
- r = q.getReleaseById(album.release.id, include)
- various = not r.isSingleArtistRelease()
+ various = not release.isSingleArtistRelease()
if various and not seen_various:
seen_various = True
- tags_file('candidate-tags-0v', trackcount, True)
+ tags_file('candidate-tags-0v', trackcount, various)
tags_file('candidate-tags-' + str(i), trackcount, various,
- r.artist.name, r.title,
- album.release.getReleaseEventsAsDict(),
- r.tracks)
+ release.artist.name, release.title,
+ release.getReleaseEventsAsDict(),
+ release.tracks)
+ if release.asin:
+ # 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
+ asin = release.asin
+ if asin not in seen_asins:
+ seen_asins.add(asin)
+ cover_art(str(i), asin)
def rip(device, trackcount, single_file):
if device == None:
device = '/dev/cdrom'
- argv = ['cdparanoia', '-d', device]
+ argv = ['cdparanoia', '-d', device, '1-' + str(trackcount)]
if single_file:
- argv.extend(['1-' + str(trackcount), 'wav'])
+ argv.append('wav')
else:
argv.append('-B')
c(fp.write, command +' "$@"\n')
c(fp.close)
+def get_releases(filter_, tries=5):
+ sleep = 1
+ query = musicbrainz2.webservice.Query()
+ while True:
+ try:
+ return query.getReleases(filter_)
+ except musicbrainz2.webservice.WebServiceError, e:
+ if '503' not in e.msg:
+ raise
+ tries -= 1
+ sys.stderr.write('getReleases: %s: ' % e)
+ if tries == 0:
+ sys.stderr.write('giving up\n')
+ raise
+ sleep *= 2
+ sys.stderr.write('sleeping %ds before retry...\n' % sleep)
+ time.sleep(sleep)
+
+def releases_by_disc(disc_id):
+ try:
+ musicbrainzngs.musicbrainz.get_releases_by_discid(disc_id)
+ except musicbrainzngs.musicbrainz.ResponseError:
+ return []
+ raise 'what now'
+
+def releases_by(q, title, artist=None):
+ filter_ = musicbrainz2.webservice.ReleaseFilter(title=title)
+ results = get_releases(filter_)
+ releases = (result.release for result in results)
+ if artist:
+ pattern = re.sub(r'\s+', r'\s+', artist.strip())
+ releases = (x for x in releases
+ if re.match(pattern, x.artist.name, re.IGNORECASE))
+ return releases
+
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('--discid')
+ parser.add_option('--title')
+ parser.add_option('--print-discid', action='store_true', default=False)
parser.add_option('-d', '--device')
parser.add_option('-m', '--no-musicbrainz',
action='store_true', default=False)
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)
- disc = musicbrainz2.disc.readDisc(device)
-
- trackcount = mkcue(disc, trackcount)
- tags(disc, trackcount, mb=not options.no_musicbrainz)
- rip(device, trackcount, options.single_file)
+ if options.title != None:
+ tags(releases_by(q, options.title, options.artist), trackcount)
+ elif options.discid != None:
+ tags(releases_by_disc(options.discid), trackcount)
+ else:
+ disc = discid.disc.read(device)
+ if options.print_discid:
+ print disc.id
+ return 0
+ fp = c(file, 'cue', 'w')
+ trackcount = mkcue(fp, disc, trackcount)
+ c(fp.close)
+ if options.no_musicbrainz:
+ releases = []
+ else:
+ releases = releases_by_disc(disc.id)
+ tags(releases, trackcount)
+ rip(device, trackcount, options.single_file)
except Exception, error:
if isinstance(error, SystemExit):
raise
projects / flac-archive / blobdiff