Das deutsche Python-Forum

Code: Alles auswählen

#!/usr/bin/python

# (all?) possible keys for movie data?
# see res.get()

# from http://imdbpy.sourceforge.net/?page=docs
from imdb import IMDb

def getPersons(moviename):
    print "fetching actors for movie '" + moviename + "'"
    ia = IMDb()
    s_result = ia.search_movie(moviename)
    if (len(s_result) == 0):
        print "got no movie for '" + moviename + "'"
        return false
    res = s_result[0]
    print res.get('music')
    ia.update(res)
    actors = s_result[0].get('actors')
    ret = {}
    for actor in actors:
        ret[str(actor.getID())] = str(actor)
    print "got " + str(len(actors)) + " actors"
    return ret

def intersect(a, b):
     return list(set(a) & set(b))

def compareMovies(movieNames):
    movieActors = {}
    for movieName in movieNames:
        movieActors[movieName] = getPersons(movieName)
    i = 0
    for k in movieActors.keys():
        i += 1
        for k2 in movieActors.keys()[i:]:
            actors = intersect(movieActors[k], movieActors[k2])
            print "comparing '" + k + "' to '" + k2 + "' => got " + str(len(actors)) + " hit(s)"
            for actor in actors:
                print movieActors[k][actor]

compareMovies(['Titanic', 'The Untouchables', 'Godzilla', 'The Reader'])