]> diplodocus.org Git - flac-archive/blobdiff - flac2mp3
pointless quote change
[flac-archive] / flac2mp3
index 20ab4760344d64e189e4f9b535c2031172a80368..1e29ad0bc04f727d947667bf6dc8107d747cc02e 100755 (executable)
--- a/flac2mp3
+++ b/flac2mp3
@@ -1,6 +1,6 @@
 #! /usr/bin/env python2.4
 
-'''
+"""
 =head1 NAME
 
 B<flac2mp3> - transcode FLAC file to MP3 files
@@ -48,7 +48,7 @@ Written by Eric Gillespie <epg@pretzelnet.org>.
 
 =cut
 
-''' #' # python-mode is sucks
+"""
 
 import re, sys, traceback
 from optparse import OptionParser
@@ -61,13 +61,11 @@ from org.diplodocus.util import run_or_die
 ################################################################################
 # The child processes
 
-def flac2mp3(fn, title, artist, album, date, track, skip_until, pic=None):
-    (title, artist, album, date) = [(x == None and 'unknown') or x
-                                    for x in (title, artist, album, date)]
-    try:
-        (skip_arg, until_arg) = skip_until
-    except ValueError:
-        skip_arg = until_arg = ''
+def flac2mp3(fn, title, artist, album, date, track, skip_until, pics=None):
+    (title, artist, album) = [(x == None and 'unknown') or x
+                              for x in (title, artist, album)]
+    if date == None:
+        date = ''
 
     if quiet:
         flac_options = '--silent'
@@ -84,20 +82,24 @@ def flac2mp3(fn, title, artist, album, date, track, skip_until, pic=None):
     verbose and tmp.append('--verbose')
     lame_options = ' '.join(tmp)
 
+    outfile = ('%s (%s) %02d %s.mp3' % (artist, album,
+                                        track, title)).replace('/', '_')
+
     # Escape any single quotes ' so we can quote this.
     (fn, title, artist,
      album, date) = [x.replace("'", r"'\''")
                      for x in (fn, title, artist, album, date)]
 
-    outfile = ('%s (%s) %02d %s.mp3' % (artist, album,
-                                        track, title)).replace('/', '_')
+    quoted_outfile = ('%s (%s) %02d %s.mp3' % (artist, album,
+                                               track, title)).replace('/', '_')
 
-    run_or_die(3, "flac %s -cd %s %s '%s' | lame --add-id3v2 %s --tt '%s' --ta '%s' --tl '%s' --ty '%s' --tn %d - '%s'"
-               % (flac_options, skip_arg or '', until_arg or '', fn,
-                  lame_options, title, artist, album, date, track, outfile))
+    run_or_die(3, "flac %s -cd %s '%s' | lame --add-id3v2 %s --tt '%s' --ta '%s' --tl '%s' --ty '%s' --tn %d - '%s'"
+               % (flac_options, ' '.join(skip_until), fn,
+                  lame_options, title, artist, album, date, track,
+                  quoted_outfile))
 
-    if pic != None:
-        taglib.add_apic_frame_to_mp3(outfile, pic[0], pic[1], pic[2])
+    if pics != None:
+        taglib.add_apic_frame_to_mp3(outfile, pics)
 
     return 0
 
@@ -148,18 +150,18 @@ def get_decode_args(fn):
 # get PART as part of the filelname, same as mp3s.
 class Tags(object):
     def __init__(self):
+        self._global = {}
         self._tags = {}
     def __len__(self):
-        return len(self._tags)
+        # All files have at least one track.
+        return max(1, len(self._tags))
     def get(self, key, track=None):
         key = key.upper()
         try:
-            if track == None:
-                return self._tags[None][key]
             try:
                 return self._tags[track][key]
             except KeyError:
-                return self._tags[None][key]
+                return self._global[key]
         except KeyError:
             return None
     def gets(self, key, track=None):
@@ -168,15 +170,21 @@ class Tags(object):
             return None
         return '\n'.join(value)
     def set(self, key, value, track=None):
-        if track not in self._tags:
-            self._tags[track] = {}
-        if key not in self._tags[track]:
-            self._tags[track][key] = []
-        self._tags[track][key].append(value)
+        key = key.upper()
+        if track == None:
+            tags = self._global
+        else:
+            try:
+                tags = self._tags[track]
+            except KeyError:
+                tags = self._tags[track] = {}
+        if key not in tags:
+            tags[key] = []
+        tags[key].append(value)
 
 def get_tags(fn):
-    '''Return the ARTIST, ALBUM, and DATE tags followed by the TITLE tags
-    in the file FN.'''
+    """Return the ARTIST, ALBUM, and DATE tags followed by the TITLE tags
+    in the file FN."""
 
     tags = Tags()
 
@@ -197,18 +205,6 @@ def get_tags(fn):
 
     return tags
 
-def find_pic(fn, tags):
-    pic = tags.get('__flac2mp3_PICTURE')
-
-    if not isinstance(pic, tuple):
-        for i in flac.get_pictures(fn):
-            if i[1] == flac.PICTURE_TYPE_FRONT_COVER:
-                pic = i[:3]
-                break
-        tags.set('__flac2mp3_PICTURE', pic)
-
-    return pic
-
 def main(argv):
     # Control the exit code for any uncaught exceptions.
     try:
@@ -262,6 +258,8 @@ def main(argv):
                 else:
                     track = int(track)
 
+                pics = flac.get_pictures(fn)
+
                 for i in range(len(tags)):
                     title = tags.gets('TITLE', track)
                     part = tags.gets('PART', track)
@@ -271,7 +269,7 @@ def main(argv):
                                  tags.gets('ARTIST', track),
                                  album,
                                  tags.gets('DATE', track),
-                                 track, args[i], find_pic(fn, tags)])
+                                 track, args[i], pics])
                     track = i + 2
             except Exception, error:
                 sys.stderr.write(getattr(error, 'msg', ''))