fix-music-tags now supports Ogg files in a hacky way

joostsijm · joostsijm · commit d79c6aad7233 · 2016-07-05T10:01:14.000+02:00
diff --git a/python2.7/fix-music-tags.py b/python2.7/fix-music-tags.py
@@ -8,14 +8,19 @@
 """
 
 from mutagen.easyid3 import EasyID3
+from mutagen.oggvorbis import OggVorbis 
 import os
 import argparse
 import glob
 import re
 
 
 def fixTags(fname, keep):
-    audio = EasyID3(fname)
+    ext = os.path.splitext(fname)[1]
+    if ext == ".mp3":
+        audio = EasyID3(fname)
+    elif ext == ".ogg":
+        audio = OggVorbis(fname)
 
     delKeys = []
     for k, v in audio.items():
@@ -27,7 +32,11 @@ def fixTags(fname, keep):
     audio.save()
 
 def fixNumber(fname):
-    audio = EasyID3(fname)
+    ext = os.path.splitext(fname)[1]
+    if ext == ".mp3":
+        audio = EasyID3(fname)
+    elif ext == ".ogg":
+        audio = OggVorbis(fname)
 
     if `'tracknumber'` in audio:
         return
@@ -55,3 +64,9 @@ def fixNumber(fname):
         fixTags(fname, args.keep)
         if args.fixnumber:
             fixNumber(fname)
+
+    for fname in glob.glob("{}/*.ogg".format(args.directory)):
+        print("Fixing tags for {}".format(fname))
+        fixTags(fname, args.keep)
+        if args.fixnumber:
+            fixNumber(fname)
