Add fix-music-tags script.

bamos · bamos · commit 1b3aabb9e8a0 · 2015-12-30T11:30:01.000-05:00
diff --git a/README.md b/README.md
@@ -74,6 +74,15 @@ This script computes the mean of a directory of images for Caffe.
 
 
 
+## [python2.7/fix-music-tags.py](https://github.com/bamos/python-scripts/blob/master/python2.7/fix-music-tags.py)
++ Authors: [Brandon Amos](http://bamos.github.io)
++ Created: 2015.12.30
+
+
+This script (fix-music-tags.py) mass-removes unwanted music tags.
+
+
+
 ## [python2.7/mt.py](https://github.com/bamos/python-scripts/blob/master/python2.7/mt.py)
 + Authors: [Brandon Amos](http://bamos.github.io)
 + Created: 2014.11.30
@@ -345,8 +354,8 @@ and I'm happy to merge pull requests of other projects.
 
 Name | Stargazers | Description
 ----|----|----
-[averagesecurityguy/Python-Examples](https://github.com/averagesecurityguy/Python-Examples) | 20 | Example scripts for common python tasks
-[ClarkGoble/Scripts](https://github.com/ClarkGoble/Scripts) | 28 | My scripts - primarily using python and appscript
-[computermacgyver/twitter-python](https://github.com/computermacgyver/twitter-python) | 50 | Simple example scripts for Twitter data collection with Tweepy in Python
+[averagesecurityguy/Python-Examples](https://github.com/averagesecurityguy/Python-Examples) | 22 | Example scripts for common python tasks
+[ClarkGoble/Scripts](https://github.com/ClarkGoble/Scripts) | 29 | My scripts - primarily using python and appscript
+[computermacgyver/twitter-python](https://github.com/computermacgyver/twitter-python) | 62 | Simple example scripts for Twitter data collection with Tweepy in Python
 [gpambrozio/PythonScripts](https://github.com/gpambrozio/PythonScripts) | 38 | A bunch of Python scripts I made and that might interest somebody else
-[realpython/python-scripts](https://github.com/realpython/python-scripts) | 67 | because i'm tired of gists
+[realpython/python-scripts](https://github.com/realpython/python-scripts) | 549 | because i'm tired of gists
diff --git a/python2.7/fix-music-tags.py b/python2.7/fix-music-tags.py
@@ -0,0 +1,40 @@
+#!/usr/bin/env python2
+
+__author__ = ['[Brandon Amos](http://bamos.github.io)']
+__date__ = '2015.12.30'
+
+"""
+This script (fix-music-tags.py) mass-removes unwanted music tags.
+"""
+
+from mutagen.easyid3 import EasyID3
+import argparse
+import glob
+
+
+def fixTags(fname, keep):
+    audio = EasyID3(fname)
+    artist = audio['artist']
+    title = audio['title']
+    album = audio['album']
+
+    delKeys = []
+    for k, v in audio.items():
+        if k not in keep:
+            delKeys.append(k)
+
+    for k in delKeys:
+        del audio[k]
+    audio.save()
+
+if __name__ == '__main__':
+    parser = argparse.ArgumentParser()
+    parser.add_argument('directory', help='Directory with mp3 files to fix.')
+    parser.add_argument('--keep', default=['title', 'artist', 'album', 'genre'],
+                        type=str, nargs='+', metavar='TAG',
+                        help="Tags to keep. Default: title, artist, album, genre")
+    args = parser.parse_args()
+
+    for fname in glob.glob("{}/*.mp3".format(args.directory)):
+        print("Fixing tags for {}".format(fname))
+        fixTags(fname, args.keep)
