Rename Movies

gistfile1.py

import os
import tmdbsimple


tmdbsimple.API_KEY = "<KEY>"
MOVIES_PATH = "/<PATH>/"


def lookup(title):
    title = title.replace("_", " ")
    title = title.replace("-", " - ")

    print "\nLooking Up: %s\n" % title

    search = tmdbsimple.Search()
    search.movie(query=title)

    results = []
    for number, result in enumerate(search.results):
        title = "%s (%s)" % (result["title"], result["release_date"][0:4])
        results.append(title)

        print "[%s] %s" % (number, title)

        if number > 9:
            break

    return results


def rename(old_name):
    file_name, extension = old_name.split(".", 1)

    results = lookup(file_name)

    selection = raw_input("\nChoose Movie: [0] ") or 0

    try:
        new_name = "%s.%s" % (results[int(selection)], extension)
    except IndexError:
        return

    rename = raw_input("Rename: %s => %s [y/N] " % (old_name, new_name))

    if rename:
        old_name = MOVIES_PATH + old_name
        new_name = MOVIES_PATH + new_name
        os.rename(old_name, new_name)


if __name__ == "__main__":
    movies = os.listdir(MOVIES_PATH)
    for movie in sorted(movies):
        rename(movie)