Beispiel #1
0
def movie_from_rottentomatoes_input(rottentomatoes_input):
    movie = Movies()
    directors, writers, actors, genres = [], [], [], []
    rottentomatoes_id = rottentomatoes_id_from_input(rottentomatoes_input)
    if rottentomatoes_id:
        try:
            movie_dict = get_rottentomatoes_dict(rottentomatoes_id)
            try:
                if movie_dict.get("Response") == False:
                    return {"error_msg": "Invalid"}
            except Exception:
                pass
            movie.RottenTomatoesId = str(movie_dict.get("id"))
            movie.Title = movie_dict.get("title")
            movie.Year = movie_dict.get("year")
            if movie_dict.get("runtime"):
                runtime_str = str(movie_dict.get("runtime"))
            if movie_dict.get("abridged_directors"):
                for j in range(len(movie_dict.get("abridged_directors"))):
                    directors.append({"name": movie_dict.get("abridged_directors")[j].get("name")})
            if movie_dict.get("abridged_cast"):
                for j in range(len(movie_dict.get("abridged_cast"))):
                    actors.append(
                        {
                            "name": movie_dict.get("abridged_cast")[j].get("name"),
                            "id": movie_dict.get("abridged_cast")[j].get("id"),
                        }
                    )
            if movie_dict.get("genres"):
                for j in range(len(movie_dict.get("genres"))):
                    genres.append(movie_dict.get("genres")[j])
        except Exception:
            return {"error_msg": "Rotten Tomatoes API failed, please try again."}
    else:
        return {"error_msg": "Could not be parsed from input."}
    return {"movie": movie, "directors": directors, "writers": writers, "actors": actors, "genres": genres}
Beispiel #2
0
def movie_from_inputs(imdb_input, netflix_input, rottentomatoes_input, wikipedia_input):
    imdb_movie, rottentomatoes_movie, netflix_movie, wikipedia_movie = None, None, None, None
    directors, writers, actors, genres = [], [], [], []
    movie = Movies()
    error_list = {}
    success = True
    if imdb_input and len(imdb_input) > 0:
        imdb_res = movie_from_imdb_input(imdb_input)
        if imdb_res.get("movie"):
            imdb_movie = imdb_res.get("movie")
            movie.ImdbId = imdb_res.get("movie").ImdbId
            movie.Title = imdb_res.get("movie").Title
            movie.Year = imdb_res.get("movie").Year
            movie.Runtime = imdb_res.get("movie").Runtime
            directors = imdb_res.get("directors")
            writers = imdb_res.get("writers")
            actors = imdb_res.get("actors")
            genres = imdb_res.get("genres")
        else:
            error_list["ImdbId"] = imdb_res.get("error_msg")
            success = False
            # Imdb success required for now
    else:
        error_list["ImdbId"] = "This field cannot be blank."
        success = False
    if netflix_input and len(netflix_input) > 0:
        netflix_res = movie_from_netflix_input(netflix_input)
        if netflix_res.get("movie"):
            netflix_movie = netflix_res.get("movie")
            movie.NetflixId = netflix_movie.NetflixId
        else:
            error_list["NetflixId"] = netflix_res.get("error_msg")
            success = False
            # If no netflix input, get netflix movie given imdb movie
    elif movie.Title and movie.Year:
        netflix_res = netflix_movie_from_title(movie)
        if netflix_res.get("movie"):
            netflix_movie = netflix_res.get("movie")
            movie.NetflixId = netflix_movie.NetflixId
    if rottentomatoes_input and len(rottentomatoes_input) > 0:
        rottentomatoes_res = movie_from_rottentomatoes_input(rottentomatoes_input)
        if rottentomatoes_res.get("movie"):
            rottentomatoes_movie = rottentomatoes_res.get("movie")
            movie.RottenTomatoesId = rottentomatoes_movie.RottenTomatoesId
            # Use better cast list from rotten tomatoes
            actors = rottentomatoes_res.get("actors")
        else:
            error_list["RottenTomatoesId"] = rottentomatoes_res.get("error_msg")
            success = False
            # If no rotten tomatoes input, get rotten tomatoes movie given imdb movie
    elif movie.Title and movie.Year:
        rottentomatoes_res = rottentomatoes_movie_from_title(movie)
        if rottentomatoes_res.get("movie"):
            rottentomatoes_movie = rottentomatoes_res.get("movie")
            movie.RottenTomatoesId = rottentomatoes_movie.RottenTomatoesId
            # Use better cast list from rotten tomatoes
            actors = rottentomatoes_res.get("actors")
            # Rotten Tomatoes success required for now
    else:
        error_list["RottenTomatoesId"] = "This field cannot be blank."
        success = False
        # Use given wikipedia input with no validation, maybe include in future
    if wikipedia_input and len(wikipedia_input) > 0:
        wikipedia_res = movie_from_wikipedia_input(wikipedia_input)
        if wikipedia_res.get("movie"):
            wikipedia_movie = wikipedia_res.get("movie")
            movie.WikipediaId = wikipedia_movie.WikipediaId
        else:
            error_list["WikipediaId"] = wikipedia_res.get("error_msg")
            success = False
    elif movie.Title and movie.Year:
        wikipedia_res = wikipedia_movie_from_title(movie)
        if wikipedia_res.get("movie"):
            wikipedia_movie = wikipedia_res.get("movie")
            movie.WikipediaId = wikipedia_movie.WikipediaId
    return {
        "success": success,
        "movie": movie,
        "directors": directors,
        "writers": writers,
        "actors": actors,
        "genres": genres,
        "error_list": error_list,
    }
Beispiel #3
0
def view_list(request):
	try:
		logged_in_profile_info = { }
		permission_response = check_and_get_session_info(request, logged_in_profile_info)
		if permission_response != True:
			return permission_response
		if not logged_in_profile_info['id']:
			set_msg(request, 'Action Failed!', 'You must be logged in to perform that action', 'warning')
			return redirect('webapp.views.profile.login')
		type_dict = get_type_dict()
		if request.GET.get('add') and request.GET.get('i') and request.GET.get('r'):
			'''*****************************************************************************
			Add movie given imdb identifier (rotten tomatoes id required as well) and redirect to movie page by way of association functions if success otherwise back to search with errors
			PATH: webapp.views.movie.view_list; METHOD: none; PARAMS: get - add,i,r; MISC: none;
			*****************************************************************************'''
			try:
				has_error = False
				error_text = None
				movie = Movies ()
				# Start movie from imdb id
				res_dict = movie_from_imdb_input(request.GET.get('i'))
				if res_dict.get('movie'):
					movie = res_dict.get('movie')
				else:
					has_error = True
					error_text = 'IMDb ID Validation: ' + res_dict.get('error_msg')
					raise ValidationError('')
				# Get wikipedia id if resolved
				res = wikipedia_movie_from_title(movie)
				if res.get('movie'):
					movie.WikipediaId = res.get('movie').WikipediaId
				# Set rotten tomatoes, and wikipedia id
				movie.RottenTomatoesId = request.GET.get('r')
				test = movie_from_rottentomatoes_input(movie.RottenTomatoesId)
				if test.get('movie'):
					# Use better cast list from rotten tomatoes
					res_dict['actors'] = test.get('actors')
				else:
					has_error = True
					error_text = 'Rotten Tomatoes ID Validation: ' + test.get('error_msg')
					raise ValidationError('')
				movie.full_clean()
				movie.save()
				create_properties(movie, res_dict.get('directors'), res_dict.get('writers'), res_dict.get('actors'), res_dict.get('genres'), logged_in_profile_info['username'])
				movie_logger.info(movie.UrlTitle + ' Create Success by '  + logged_in_profile_info['username'])
				# Redirect to 'add association' function and add option 'seen' if already present
				res = redirect('webapp.views.movie.view', urltitle=movie.UrlTitle)
				res['Location'] += '?assoc=1&add=1'
				if request.GET.get('seen'):
					res['Location'] += '&seen=1'
				return res
			except ValidationError as e:
				urltitle = movie.UrlTitle if movie.UrlTitle else 'Unknown'
				movie_logger.info(urltitle + ' Create Failure by ' + logged_in_profile_info['username'])
				error_msg = None
				if has_error:
					error_msg = {'Error' : error_text}
				else:
					error_msg = e.message_dict
					for key in error_msg:
						error_msg[key] = str(error_msg[key][0])
				return render_to_response('movie/search.html', {'header' : generate_header_dict(request, 'Search'), 'success' : False, 'results' : error_msg}, RequestContext(request))
		elif logged_in_profile_info['admin'] and request.GET.get('add'):
			if request.method == 'POST':
				if request.POST.get('api_search_term'):
					'''*****************************************************************************
					Return results from imdb, netflix, rottentomatoes, and wikipedia searches as options to add movie
					PATH: webapp.views.movie.view_list; METHOD: post; PARAMS: get - add, post - api_search_term; MISC: logged_in_profile.IsAdmin;
					*****************************************************************************'''
					api_search_term = request.POST.get('api_search_term')
					api_search_length = int(request.POST.get('api_search_length')) if request.POST.get('api_search_length') and request.POST.get('api_search_length').isdigit() else 5
					api_search_length = api_search_length if api_search_length > 0 and api_search_length <= 25 else 5
					res_dict = movies_from_apis_term(api_search_term, api_search_length)
					imdb_possibilities = res_dict.get('imdb_movies')
					for movie in imdb_possibilities:
						movie.UrlTitle = imdb_link_for_movie(movie)
					netflix_possibilities = res_dict.get('netflix_movies')
					for movie in netflix_possibilities:
						movie.UrlTitle = netflix_link_for_movie(movie)
					rottentomatoes_possibilities = res_dict.get('rottentomatoes_movies')
					for movie in rottentomatoes_possibilities:
						movie.UrlTitle = rottentomatoes_link_for_movie(movie)
					wikipedia_possibilities = res_dict.get('wikipedia_movies')
					for movie in wikipedia_possibilities:
						movie.UrlTitle = wikipedia_link_for_movie(movie)
					return render_to_response('movie/add.html', {'header' : generate_header_dict(request, 'Add Movie'), 'api_search_term' : api_search_term, 'api_search_length' : api_search_length, 'imdb_possibilities' : imdb_possibilities, 'netflix_possibilities' : netflix_possibilities, 'rottentomatoes_possibilities' : rottentomatoes_possibilities, 'wikipedia_possibilities' : wikipedia_possibilities}, RequestContext(request))
				else:
					'''*****************************************************************************
					Add movie given user input of imdb, netflix, rottentomatoes, and wikipedia urls or ids and redirect to movie page
					PATH: webapp.views.movie.view_list; METHOD: post; PARAMS: get - add; MISC: logged_in_profile.IsAdmin;
					*****************************************************************************'''
					res_dict = movie_from_inputs(request.POST.get('imdb_url'), request.POST.get('netflix_url'), request.POST.get('rottentomatoes_id'), request.POST.get('wikipedia_id'))
					if res_dict.get('success'):
						try:
							movie = res_dict.get('movie')
							movie.full_clean()
							movie.save()
							create_properties(movie, res_dict.get('directors'), res_dict.get('writers'), res_dict.get('actors'), res_dict.get('genres'), logged_in_profile_info['username'])
							movie_logger.info(movie.UrlTitle + ' Create Success by ' + logged_in_profile_info['username'])
							return redirect('webapp.views.movie.view', urltitle=movie.UrlTitle)
						except ValidationError as e:
							urltitle = movie.UrlTitle if movie.UrlTitle else 'Unknown'
							movie_logger.info(urltitle + ' Create Failure by ' + logged_in_profile_info['username'])
							# Restore values for add page
							movie.ImdbId = request.POST.get('imdb_url')
							movies.RottenTomatoesId = request.POST.get('rottentomatoes_id')
							movies.NetflixId = request.POST.get('netflix_url')
							movies.WikipediaId = request.POST.get('wikipedia_id')
							error_msg = e.message_dict
							for key in error_msg:
								error_msg[key] = str(error_msg[key][0])
							return render_to_response('movie/add.html', {'header' : generate_header_dict(request, 'Add Movie'), 'error_msg' : error_msg, 'movie' : movie, 'links' : generate_links_dict(movie)}, RequestContext(request))
					else:
						movie_logger.info('Movie Create Failure by ' + logged_in_profile_info['username'])
						# Restore values for add page
						movie = res_dict.get('movie')
						movie.ImdbId = request.POST.get('imdb_url')
						movie.RottenTomatoesId = request.POST.get('rottentomatoes_id')
						movie.NetflixId = request.POST.get('netflix_url')
						movie.WikipediaId = request.POST.get('wikipedia_id')
						return render_to_response('movie/add.html', {'header' : generate_header_dict(request, 'Add Movie'), 'movie' : movie, 'error_msg' : res_dict.get('error_list'), 'links' : generate_links_dict(movie)}, RequestContext(request))
			else:
				'''*****************************************************************************
				Display admin add movie page
				PATH: webapp.views.movie.view_list; METHOD: none; PARAMS: get - add; MISC: logged_in_profile.IsAdmin;
				*****************************************************************************'''
				return render_to_response('movie/add.html', {'header' : generate_header_dict(request, 'Add Movie')}, RequestContext(request))
		else:
			'''*****************************************************************************
			Display movie list page
			PATH: webapp.views.movie.view_list; METHOD: none; PARAMS: none; MISC: none;
			*****************************************************************************'''
			movies, paginated_movies = [], None
			movie_list = Movies.objects.all().order_by('-Year', 'Title')
			length = int(request.GET.get('length')) if request.GET.get('length') and request.GET.get('length').isdigit() else 25
			length = length if length <= 100 else 100
			paginator = Paginator(movie_list, length)
			page = request.GET.get('page')
			try:
				paginated_movies = paginator.page(page)
			except PageNotAnInteger:
				paginated_movies = paginator.page(1)
			except EmptyPage:
				paginated_movies = paginator.page(paginator.num_pages)
			# Get all associations with logged in profile to correctly display links (seen later as well)
			for movie in paginated_movies:
				try:
					association = Associations.objects.get(ProfileId = logged_in_profile_info['id'], ConsumeableId = movie,  ConsumeableTypeId = type_dict['CONSUMEABLE_MOVIE'])
					movies.append((movie, True))
				except Exception:
					movies.append((movie, False))
			return render_to_response('movie/view_list.html', {'header' : generate_header_dict(request, 'Movie List'), 'movies' : movies, 'page' : paginated_movies}, RequestContext(request))
	except Exception:
		movie_logger.error('Unexpected error: ' + str(sys.exc_info()[0]))
		return render_to_response('500.html', {'header' : generate_header_dict(request, 'Error')}, RequestContext(request))