filerenamer.py

#!/usr/bin/python

import sys, os, shutil
import re
import optparse
import logging

FULLFORMAT = "%(asctime)s  [%(levelname)s]  [%(module)s] %(message)s"
BASICFORMAT = "%(message)s"
logger = logging.getLogger()
delimiters = " _"
seasonre = re.compile(r"season\s*(\d)", re.IGNORECASE)
prefixre = re.compile(r"(\w+)[%s]." % delimiters)

def filerename(directory, new, original=None, simulate=False):
    root = os.path.abspath(directory)
    logger.debug("walking %s, renaming to %s" % (root, new))

    for filename in os.listdir(root):
        if not os.path.isfile(os.path.join(directory, filename)):
            continue
        try:
            original = prefixre.match(filename).group(1) if original is None else original
        except AttributeError:
            original = "Episode"

        fullfile = os.path.join(root, filename)
        basedir = os.path.basename(directory)
        logger.debug("parsing %s from %s" % (fullfile, basedir))

        name, extension = os.path.splitext(filename)
        # extract season from directory:
        season = "01"
        try:
            season = "%02d" % int(seasonre.match(basedir).group(1))
        except (IndexError, AttributeError):
            logger.debug("could not determine season, assuming it's the first")

        for delimiter in delimiters:
            name = name.replace(delimiter, ".")
        # renaming quirks:
        name = name.replace(":.", ":")
        name = name.replace(".-.", "-")
        prefix = '%s.S%sE' % (new, season)
        name = name.replace("%s." % original, "%s0" % prefix)
        # renaming quirks: remove 0 for double digits episodes
        name = re.sub(r'(%s)0(\d{2})' % prefix, r'\1\2', name)
        # renaming quirks: join double episodes again, put 0
        name = re.sub(r'(%s\d{2}).(\d{1,2})' % prefix, r'\1-0\2', name)
        # renaming quirks: remove 0 if trailing episode is > 10
        name = re.sub(r'(%s\d{2}-)0(\d{2})' % prefix, r'\1\2', name)

        newfile = "%s/%s%s" % (directory, name, extension)
        logger.debug("mv %s %s" % (fullfile, newfile))
        if not simulate:
            shutil.move(fullfile, newfile)

def main():
    # Setup the command line arguments.
    optp = optparse.OptionParser()
    # options.
    optp.add_option("-v", "--verbose", dest="verbose",
                    help="log verbosity.", action="store_true", default=False)

    optp.add_option("-s", "--simulate", dest="simulate",
                    help="do nothing, just simulate.", action="store_true", default=False)

    optp.add_option("-o", "--origin", dest="origin",
                    help="origin naming.")

    optp.add_option("-n", "--new", dest="new",
                    help="new naming.")

    opts, args = optp.parse_args()

    loglevel = logging.DEBUG if opts.simulate or opts.verbose else logging.INFO
    logformat = FULLFORMAT if opts.verbose else BASICFORMAT
    # log to stderr in fg
    logging.basicConfig(level=loglevel,
                        format=logformat)

    if len(args) < 1:
        print >> sys.stderr, "please specify a valid directory"
        optp.print_help()
        sys.exit(-1)

    if not os.path.isdir(args[0]):
        print >> sys.stderr, "%s is not a valid directory" % args[0]
        sys.exit(-1)

    if opts.origin is None:
        print >> sys.stderr, "Will try to guess origin name"

    if opts.new is None:
        opts.new = os.path.basename(os.path.abspath(args[0]))
        print >> sys.stderr, "Assuming new name %s" % opts.new

    for root, _, _ in os.walk(args[0]):
        filerename(root, opts.new, opts.origin, opts.simulate)

if __name__ == "__main__":
    main()