X-Git-Url: https://diplodocus.org/git/flac-archive/blobdiff_plain/5fbfc29eae00bde7562adba75406161c9bf756d2..73112f3f050b8f78a6d03449c00a60c6a30f715e:/rewrite-tags

diff --git a/rewrite-tags b/rewrite-tags
index e164f3f..a3b5eb6 100755
--- a/rewrite-tags
+++ b/rewrite-tags
@@ -1,7 +1,9 @@
 #! /usr/bin/python
 
-import sys
-from subprocess import Popen, PIPE
+import os, sys
+import subprocess
+import tempfile
+from errno import EEXIST
 
 from flac_archive.tags import Tags
 
@@ -18,21 +20,33 @@ class SubprocessError(Exception):
         Exception.__init__(self, msg % (abs(status), command_msg))
         self.status = status
         self.command = command
-        self.stderr = stderr
+        self.stderr = ''.join(stderr)
 
 def get_tags(fn):
     tags = Tags()
 
     command = ['metaflac', '--no-utf8-convert', '--export-tags-to=-', fn]
-    p = Popen(command, stdout=PIPE)
+    p = subprocess.Popen(command, stdout=subprocess.PIPE,
+                         stderr=subprocess.STDOUT)
     tags.load(p.stdout)
 
     status = p.wait()
     if status != 0:
-        raise SubprocessError(status, command=command, stderr=p.stderr)
+        raise SubprocessError(status, command=command, stderr=p.stdout)
 
     return tags
 
+def rewrite_track_tags(track, tags, fn, tmp):
+    tmp.write('\n'.join(tags.track(track)) + '\n')
+    tmp.close()
+    command = ['metaflac', '--no-utf8-convert', '--dont-use-padding',
+               '--remove-all-tags', '--import-tags-from', tmp.name, fn]
+    p = subprocess.Popen(command, stdout=subprocess.PIPE,
+                         stderr=subprocess.STDOUT)
+    status = p.wait()
+    if status != 0:
+        raise SubprocessError(status, command=command, stderr=p.stdout)
+
 def do_read(filenames):
     # Use this mapping of tag names to sets of tag values to detect global tags.
     all_tags = {}
@@ -76,7 +90,7 @@ def do_write(args):
                          % (len(tags), len(args)))
         return 4
     artist = tags.get_path_safe('ARTIST')
-    album = tags.get_path_safe('ALBUM'),
+    album = tags.get_path_safe('ALBUM')
     try:
         os.mkdir(artist)
     except OSError, e:
@@ -92,8 +106,16 @@ def do_write(args):
         track = i + 1
         fn = '%s/%s/%s.flac' % (artist, album, tags.make_filename(track))
         if fn != old_fn:
-            #os.rename(old_fn, fn)
-            pass
+            os.rename(old_fn, fn)
+        tmp = tempfile.NamedTemporaryFile(delete=False)
+        try:
+            rewrite_track_tags(track, tags, fn, tmp)
+        finally:
+            try:
+                os.unlink(tmp.name)
+            except:
+                pass
+    return 0
 
 def main(args):
     if len(args) < 3: