Python Script to Check Movie Runtimes

Code:
#!/usr/bin/env python

# cklength.py

# count_zero 2011

# This free software is licensed under the GPL (http://www.gnu.org/licenses/gpl.html)

# Check encoded movie length, compare it to feature length using IMDb 

# Requires mplayer and IMDbPY to run



import sys

import os

import re

import subprocess

import os.path

import imdb



# Change this to your mplayer executable.

mplayer_exe = '/usr/bin/mplayer'



def get_feature_length(myfile):

    """ Check IMDb for the movie, search based on the filename.

    Will return the runtime of the first movie found, in minutes."""

    name = os.path.basename(myfile).split('.')[0]

    i = imdb.IMDb()

    try:

    m = i.search_movie(name)[0]

    i.update(m)

    runtime = m['runtimes'][0]

    title = m['title']

    return name, title, runtime

    except:

    return name, 'not found', 'unknown'



def get_encoded_length(myfile):

    """ Use mplayer to calculate the length of the movie file.

    Will return the runtime, in minutes."""

    abs_myfile = os.path.abspath(myfile)

    try:

    command = '%s -identify -frames 0 -vc null -vo null -ao null "%s"' % (mplayer_exe, abs_myfile)

    length_re = re.compile('ID_LENGTH=.*')

    p = subprocess.Popen(command, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)

    length = p.communicate()[0]

    mylength = int(length_re.findall(length)[0].lstrip('ID_LENGTH=').split('.')[0]) / 60

    return mylength

    except:

    return 'unknown'

    

if __name__ == "__main__":

    myfiles = sys.argv[1:]

    for myfile in myfiles:

    length = get_encoded_length(myfile)

    name, title, runtime = get_feature_length(myfile)

    print '%s (IMDb name: %s) - %s min (local) / %s min (IMDb)' % (myfile, title, length, runtime)