Panda Tiger Eagle

lib/api.rb

@@ -9,6 +9,7 @@ class Api
  def self.search_by_title(title)
  url = "http://api.rottentomatoes.com/api/public/v1.0/movies.json?apikey=#{APIKEY}&q=#{URI.encode(title)}&page_limit=1"
  struct = OpenStruct.new(get_url_as_json(url).fetch("movies").first)
+ return false if struct.title == nil
  Movie.new(id: struct.id.to_i,
  title: struct.title,
  year: struct.year,

movie_json.rb

@@ -2,16 +2,63 @@
 require_relative "lib/api"
 
 def find_movie
- puts "OH HAI. Search?"
+ puts 'Add a movie you really like:'
  movie_title = gets
- movie = Api.search_by_title(movie_title)
- puts "Found: #{movie.title}. Score: #{movie.score}"
+ if !(current_movie = Api.search_by_title(movie_title))
+ puts 'Oops. Something went wrong or title not found.'
+ else
+ @movies << current_movie
+ puts "Found: #{@movies.last.title}. Score: #{@movies.last.score} Year: #{@movies.last.year}" 
+ if @movies.length > 1
+ puts "Average movie score: #{get_average(@movies, "score", "float")}"
+ puts "Average movie year: #{get_average(@movies, "year", "integer")}"
+ puts get_satisfaction(@movies)
+ end
+ end
+end
+
+def get_average(movies, attribute, type)
+ attribute_sum = 0
+ movies.each { |rating| attribute_sum += rating.send(attribute) }
+ if type == "float" 
+ return attribute_sum / movies.length.to_f
+ else
+ return attribute_sum / movies.length
+ end
 end
 
+def get_satisfaction(movies)
+ min_movie = movies.min_by(&:year)
+ max_movie = movies.max_by(&:year)
+ return "neutral" if min_movie == max_movie
+ max_movie_count, max_movie_score_total = 0, 0
+ min_movie_count, min_movie_score_total = 0, 0
+ movies.each do |movie|
+ if movie.year == max_movie.year
+ max_movie_count += 1 
+ max_movie_score_total += movie.score
+ end
+ if movie.year == min_movie.year
+ min_movie_count += 1 
+ min_movie_score_total += movie.score
+ end
+ end
+ user_slope = ((max_movie_score_total/max_movie_count).to_f - (min_movie_score_total/min_movie_count).to_f) / (max_movie.year - min_movie.year).to_f
+ if user_slope == 0
+ return "neutral"
+ elsif user_slope > 0
+ return "getting happier"
+ else
+ return "getting madder"
+ end
+end
+
+@movies = []
+
 find_movie
 
 while true do
- puts "Search Again (Y/N)" 
+ puts 'Search Again (Y/N)'
  answer = gets.upcase[0]
  if answer == "Y"
  find_movie