Use flac_archive.tags.Tags, greatly simplifying things. Put PART tag

[flac-archive] / flac2mp3

diff --git a/flac2mp3 b/flac2mp3
index 5ebc87396bc0d503ccfd39227fe245b30fed3d31..fa6e26dbc79985704145799b56d5eddb79f170af 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
@@ -58,6 +58,8 @@ import org.diplodocus.jobs
 from org.diplodocus import flac, taglib
 from org.diplodocus.util import run_or_die
 
+from flac_archive.tags import Tags
+
 ################################################################################
 # The child processes
 
@@ -146,59 +148,15 @@ def get_decode_args(fn):
 
     return args
 
-# XXX other things should usue this; flac files, for example, should
-# 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)
-    def get(self, key, track=None):
-        key = key.upper()
-        try:
-            try:
-                return self._tags[track][key]
-            except KeyError:
-                return self._global[key]
-        except KeyError:
-            return None
-    def gets(self, key, track=None):
-        value = self.get(key, track)
-        if value == None:
-            return None
-        return '\n'.join(value)
-    def set(self, key, value, track=None):
-        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()
 
     p = Popen(['metaflac', '--export-tags-to=-', fn], stdout=PIPE)
-    for line in (x.rstrip() for x in p.stdout):
-        (tag, value) = line.split('=', 1)
-
-        m = re.search(r'\[([0-9]+)]$', tag)
-        if m != None:
-            tag = tag[:m.start()]
-            track = int(m.group(1))
-        else:
-            track = None
+    tags.load(p.stdout)
 
-        tags.set(tag, value, track)
     # XXX dataloss!  check status
     status = p.wait()
 
@@ -227,6 +185,7 @@ def main(argv):
         traceback.print_exc()
         return 2
 
+    separator = ' '
     try:
         global debug, flac_options, lame_options, quiet, verbose
         debug = options.debug
@@ -240,7 +199,7 @@ def main(argv):
                 args = get_decode_args(fn)
 
                 tags = get_tags(fn)
-                album = tags.gets('ALBUM')
+                album = tags.gets('ALBUM', separator=separator)
                 discnum = tags.gets('DISCNUMBER')
                 track = tags.gets('TRACKNUMBER')
 
@@ -260,12 +219,12 @@ def main(argv):
                 pics = flac.get_pictures(fn)
 
                 for i in range(len(tags)):
-                    title = tags.gets('TITLE', track)
+                    title = tags.gets('TITLE', track, separator)
                     part = tags.gets('PART', track)
                     if part != None:
                         title = '%s - %s' % (title, part)
                     jobs.append([fn, title,
-                                 tags.gets('ARTIST', track),
+                                 tags.gets('ARTIST', track, separator=', '),
                                  album,
                                  tags.gets('DATE', track),
                                  track, args[i], pics])