id3-tagger.py

#!/usr/bin/env python

# ID3 module example program
# $Id$

# version 1.2
# written 2 May 1999 by Ben Gertzfield <che@debian.org>

# This program is released under the GNU GPL, version 2 or later.

import getopt, string, re, sys
import locale
from ID3 import *

version = 1.2
name = "id3-tagger.py"


def usage():
    sys.stderr.write(
        "This is %s version %0.1f, a tool for setting ID3 tags in MP3 files.\n\n"
        "Usage: %s [-t title] [-a artist] [-A album] [-y year] [-c comment] \n"
        "       %s [-g genre] [-T tracknum] [-d] [-h] [-v] file1 [file2 ...]\n\n"
        "-d: Delete the ID3 tag from specified file(s) completely\n"
        "-h: Display this text\n"
        "-v: Display the version of this program\n\n"
        "With no arguments, display the ID3 tag of the given file(s).\n"
        % (name, version, name, " " * len(name))
    )


def main():
    options = {}

    try:
        opts, args = getopt.getopt(sys.argv[1:], "t:a:A:y:c:g:T:dhvl")
    except getopt.error as msg:
        print(msg)
        usage()
        sys.exit(2)

    for opt, arg in opts:
        if opt == "-v":
            sys.stderr.write("This is %s version %0.1f.\n" % (name, version))
            sys.exit(0)
        if opt == "-h":
            usage()
            sys.exit(0)
        if opt == "-t":
            options["TITLE"] = arg
        if opt == "-a":
            options["ARTIST"] = arg
        if opt == "-A":
            options["ALBUM"] = arg
        if opt == "-y":
            options["YEAR"] = arg
        if opt == "-c":
            options["COMMENT"] = arg
        if opt == "-g":
            options["GENRE"] = arg
        if opt == "-T":
            options["TRACKNUMBER"] = arg
        if opt == "-d":
            options["delete"] = 1

    if len(args) == 0:
        usage()
        sys.exit(2)

    for file in args:
        try:
            id3info = ID3(file)

            needs_write = 0

            if len(list(options.keys())) > 0:
                needs_write = 1

            for k, v in list(options.items()):
                if k == "GENRE" and re.match("\d+$", v):
                    id3info[k] = locale.atoi(v)
                else:
                    id3info[k] = v

            if "delete" in options:
                id3info.delete()

            print(id3info)

            if needs_write:
                id3info.write()

        except InvalidTagError as msg:
            print("Invalid ID3 tag:", msg)
            continue


main()