def getMovieInfoByTMDB_ID(self, tmdb_id=''):
self.domain = 'movie'
self.action = 'getInfo'
if tmdb_id:
self.tmdb_id = tmdb_id
try:
movie_list = Movie.select(Movie.q.tmdb_id==self.tmdb_id)
if movie_list.count() == 1:
oMovie = movie_list[0]
elif movie_list.count() == 0:
raise SQLObjectNotFound
else:
raise AttributeError
except SQLObjectNotFound:
self.url = "%s%s" % (self.urls.generateURL(self.domain, self.action), self.tmdb_id)
self.movie_info = self._getResponse(self.url)
oMovie = Movie(tmdb_id = self._getKey('id', 0),
imdb_id = self._getKey('imdb_id', ''),
title = self._getKey('name', ''),
year = int(self._getYearFromDate(self._getKey('released', ''))),
genre = self._getPrimaryGenre(self._getKey('genres', [])),
mpaa = Movie.ratings.index(self._getKey('certification', 'NR')),
director = self._getDirector(self._getKey('cast', [])),
actors = self._getPrimaryActors(self._getKey('cast', [])),
description = self._getKey('overview', ''),
length = int(self._getKey('runtime', 0)),
poster_URL = self._getPosterURL(self._getKey('posters', '')),
)
return oMovie
def getMovieIDByName(self, name):
self.domain = 'movie'
self.action = 'search'
self.searchTerm = quote_plus(name)
try:
movie_list = Movie.select("""movie.title LIKE '%s'""" % name)
if movie_list.count() == 1:
self.tmdb_id = movie_list[0].tmdb_id
else:
raise SQLObjectNotFound
except SQLObjectNotFound:
self.url = "%s%s" % (self.urls.generateURL(self.domain, self.action), self.searchTerm)
movie_info = self._getResponse(self.url)
self.tmdb_id = movie_info['id']
return self.tmdb_id
