def findImdbMoviesLikeTitle(fileName):
imdbContext = imdb.IMDb()
fileNameArray = titleHandler.returnWellFormatedArrayFromTitle(fileName)
titles = {}
movies = []
lastSuccessfulTitle = ''
for i in range(0, len(fileNameArray)):
partialFileName = titleStringFromIndexOfTitleArray(fileNameArray, i)
imdbResults = imdbContext.search_movie(partialFileName)
loop = 15 if len(imdbResults) >= 15 else len(imdbResults)
for j in range(0, loop):
imdbResult = imdbResults[j]
newMovie = movie.create(imdbResult['title'])
if imdbResult.has_key('year'):
newMovie.year = imdbResult['year']
try:
titleIndex = '{0} {1}'.format(newMovie.title, newMovie.year)
if not titleIndex in titles:
titles[titleIndex] = True
movies.append(newMovie)
except UnicodeEncodeError:
continue
lastSuccessfulTitle = partialFileName
movies = titleHandler.orderMovieArrayByMatchingTitle(movies, lastSuccessfulTitle)
return movies
def findImdbSeriesLikeTitle(fileName):
imdbContext = imdb.IMDb()
fileNameArray = titleHandler.returnWellFormatedArrayFromTitle(fileName)
seriesTitles = {}
serieses = []
lastSuccessfulSeries = ''
for i in range(0, len(fileNameArray)):
partialFileName = titleStringFromIndexOfTitleArray(fileNameArray, i)
imdbResults = imdbContext.search_movie(partialFileName)
loop = 10 if len(imdbResults) >= 10 else len(imdbResults)
for j in range(0, loop):
imdbResult = imdbResults[j]
if imdbResult['kind'] == 'tv series':
newSeries = series.create(imdbResult['title'])
if not newSeries.title in seriesTitles:
seriesTitles[newSeries.title] = True
serieses.append(newSeries)
lastSuccessfulSeries = partialFileName
serieses = titleHandler.orderTvArrayByMatchingSeries(serieses, lastSuccessfulSeries)
return serieses
