def save_tmdb_poster(film): try: from film20.import_films.tmdb_poster_fetcher import fetch_film_by_title, save_tmdb_poster as _save_tmdb_poster tmdb_film = fetch_film_by_title(film) if tmdb_film and _save_tmdb_poster(film, tmdb_film): film.tmdb_import_status = Film.IMPORTED_TMDB film.save() logger.debug("tmdb hires poster saved: %r", film.poster) return True except Exception, e: logger.warning(unicode(e))
def test_fetch_by_title(self): """ Test fetching by title """ self.initialize() films = Film.objects.filter(image="", tmdb_import_status=Film.NOT_IMPORTED) for film in films: tmdb_movie = fetch_film_by_title(film) self.assertEqual(tmdb_movie is not None, True) result = save_tmdb_poster(film, tmdb_movie) self.assertEqual(result, True) film = Film.objects.get(title="The Terminator") self.assertEqual(film.image is not None, True)