diplodocus.org Git - flac-archive/blob - fa-rip

   1 #! /usr/bin/env python2.4
   2
   3 '''
   4 =head1 NAME
   5
   6 B<fa-rip> - rip a CD for B<fa-flacd>
   7
   8 =head1 SYNOPSIS
   9
  10 B<fa-rip> [B<-d> I<device>] [B<-p> I<post-processor> [B<-t> I<track-count>]
  11
  12 =head1 DESCRIPTION
  13
  14 B<fa-rip> creates a temporary directory for storage of its
  15 intermediate files, uses MusicBrainz to create the "cue" file and
  16 candidate tags files, and runs C<cdparanoia(1)> to rip the CD to the
  17 "wav" file.
  18
  19 In order for this CD to be processed by B<fa-flacd>, you must create a
  20 "tags" file.  This is usually done by renaming one of the
  21 candidate-tags files and deleting the others.  Don't forget to fill in
  22 the DATE tag in the selected candidate before renaming it.  If
  23 B<fa-rip> could not find any tag information from MusicBrainz, you'll
  24 have to fill out the candidate-tags-0 template.
  25
  26 =head1 OPTIONS
  27
  28 =over 4
  29
  30 =item B<-d> [B<--device>] I<device>
  31
  32 Use I<device> as the CD-ROM device, instead of the default
  33 "/dev/cdrom" or the environment variable CDDEV.
  34
  35 =item B<-p> [B<--post-processor>] I<post-processor>
  36
  37 Create a "post-processor" file in the temporary directory containing
  38 the line 'I<post-processor> "$@"'.  See B<fa-flacd>'s man page for
  39 information about this hook.
  40
  41 =item B<-t> [B<--tracks>] I<track-count>
  42
  43 Archive only the first I<track-count> tracks.  This is handy for
  44 ignoring data tracks.
  45
  46 =back
  47
  48 =head1 ENVIRONMENT
  49
  50 =over 4
  51
  52 =item CDDEV
  53
  54 B<fa-rip> uses this to rip audio and save the cuesheet for a CD.
  55 MusicBrainz::Client can usually figure this out automatically.
  56
  57 =back
  58
  59 =head1 AUTHORS
  60
  61 Written by Eric Gillespie <epg@pretzelnet.org>.
  62
  63 flac-archive is free software; you may redistribute it and/or modify
  64 it under the same terms as Perl itself.
  65
  66 =cut
  67
  68 ''' #' # python-mode is sucks
  69
  70 import os, sys, tempfile, traceback
  71 from optparse import OptionParser
  72
  73 import musicbrainz2.disc
  74 import musicbrainz2.webservice
  75
  76 from org.diplodocus.util import catch_EnvironmentError as c
  77
  78 def mkcue(disc, trackcount=None):
  79     fp = c(file, 'cue', 'w')
  80     c(fp.write, 'FILE "dummy.wav" WAVE\n')
  81     c(fp.write, '  TRACK 01 AUDIO\n')
  82     c(fp.write, '    INDEX 01 00:00:00\n')
  83
  84     if trackcount == None:
  85         trackcount = disc.lastTrackNum
  86     else:
  87         trackcount = min(trackcount, disc.lastTrackNum)
  88
  89     pregap = disc.tracks[0][0]
  90     for i in xrange(disc.firstTrackNum, trackcount):
  91         offset = disc.tracks[i][0]
  92         offset -= pregap
  93
  94         minutes = seconds = 0
  95         sectors = offset % 75
  96         if offset >= 75:
  97             seconds = offset / 75
  98             if seconds >= 60:
  99                 minutes = seconds / 60
 100                 seconds = seconds % 60
 101
 102         c(fp.write, '  TRACK %02d AUDIO\n' % (i + 1,))
 103         c(fp.write,
 104           '    INDEX 01 %02d:%02d:%02d\n' % (minutes, seconds, sectors))
 105
 106     c(fp.close)
 107
 108     return trackcount
 109
 110 def tags_file(fn, trackcount, various, artist=None, album=None,
 111               release_dates={}, tracks=[]):
 112     fp = c(file, fn, 'w')
 113     c(fp.write, 'ARTIST=')
 114     if artist != None:
 115         c(fp.write, artist)
 116     c(fp.write, '\nALBUM=')
 117     if album != None:
 118         c(fp.write, album)
 119     c(fp.write, '\n')
 120
 121     have_date = False
 122     for (country, date) in release_dates.items():
 123         have_date = True
 124         c(fp.write, 'DATE[%s]=%s\n' % (country, date))
 125     have_date or c(fp.write, 'DATE=\n')
 126
 127     for i in xrange(1, trackcount):
 128         try:
 129             track = tracks.pop(0)
 130             title = track.title
 131             artist = track.artist
 132         except IndexError:
 133             title = ''
 134             artist = ''
 135         various and c(fp.write, 'ARTIST[%d]=%s\n' % (i, artist))
 136         c(fp.write, 'TITLE[%d]=%s\n' % (i, title))
 137
 138     c(fp.close)
 139
 140 def tags(disc, trackcount, mb=True):
 141     results = []
 142     seen_various = False
 143
 144     tags_file('candidate-tags-0', trackcount, False)
 145
 146     if not mb:
 147         return
 148
 149     include = musicbrainz2.webservice.ReleaseIncludes(artist=True, tracks=True)
 150     q = musicbrainz2.webservice.Query()
 151     filter = musicbrainz2.webservice.ReleaseFilter(discId=disc.getId())
 152
 153     i = 0
 154     for album in q.getReleases(filter):
 155         i += 1
 156         r = q.getReleaseById(album.release.id, include)
 157         various = not r.isSingleArtistRelease()
 158
 159         if various and not seen_various:
 160             seen_various = True
 161             tags_file('candidate-tags-0v', trackcount, True)
 162
 163         tags_file('candidate-tags-' + str(i), trackcount, various,
 164                   r.artist.name, r.title,
 165                   album.release.getReleaseEventsAsDict(),
 166                   r.tracks)
 167
 168 def rip(device, trackcount, single_file):
 169     if device == None:
 170         device = '/dev/cdrom'
 171
 172     argv = ['cdparanoia', '-d', device]
 173
 174     if single_file:
 175         argv.extend(['1-' + str(trackcount), 'wav'])
 176     else:
 177         argv.append('-B')
 178
 179     c(os.execvp, argv[0], argv)
 180
 181 def make_post_processor(command):
 182     if command == None:
 183         return
 184
 185     fd = c(os.open, 'post-processor', os.O_CREAT | os.O_WRONLY, 0555)
 186     fp = c(os.fdopen, fd, 'w')
 187     c(fp.write, command +' "$@"\n')
 188     c(fp.close)
 189
 190 def main(argv):
 191     # Control the exit code for any uncaught exceptions.
 192     try:
 193         parser = OptionParser()
 194         parser.disable_interspersed_args()
 195         parser.add_option('-d', '--device')
 196         parser.add_option('-m', '--no-musicbrainz',
 197                           action='store_true', default=False)
 198         parser.add_option('-p', '--post-processor')
 199         parser.add_option('-s', '--single-file',
 200                           action='store_true', default=False)
 201         parser.add_option('-t', '--tracks', type='int', default=99)
 202     except:
 203         traceback.print_exc()
 204         return 2
 205
 206     try:
 207         # Raises SystemExit on invalid options in argv.
 208         (options, args) = parser.parse_args(argv[1:])
 209     except Exception, error:
 210         if isinstance(error, SystemExit):
 211             return 1
 212         traceback.print_exc()
 213         return 2
 214
 215     try:
 216         device = options.device
 217         trackcount = options.tracks
 218
 219         tempdir = c((lambda x: tempfile.mkdtemp(prefix=x, dir='.')),
 220                     'flac-archive.')
 221         c(os.chdir, tempdir)
 222
 223         make_post_processor(options.post_processor)
 224
 225         disc = musicbrainz2.disc.readDisc(device)
 226
 227         trackcount = mkcue(disc, trackcount)
 228         tags(disc, trackcount, mb=not options.no_musicbrainz)
 229         rip(device, trackcount, options.single_file)
 230     except Exception, error:
 231         if isinstance(error, SystemExit):
 232             raise
 233         # check all print_exc and format_exc in fa-flacd.py; i think
 234         # for some i don't do this msg print check
 235         sys.stderr.write(getattr(error, 'msg', ''))
 236         traceback.print_exc()
 237         return 2
 238
 239     return 0
 240
 241 if __name__ == '__main__':
 242     sys.exit(main(sys.argv))