box-office-cli 0.1.2 → 0.1.3

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 2db3e8d8d4c070cb193b30668a6965607fa8f0840f3d3fd57c96836895d37811
-  data.tar.gz: cb8c03f19bda4e492a1c2ca82940e97d3833e64dccc5625b6508b546c0954ba6
+  metadata.gz: 8cac9d25c7c1a316f2ba5fd7c905988c4f181b06f41b1e6b3ca342ac96561b84
+  data.tar.gz: 0761fb9cab67eeec86ef98d0c68c9604488d446363abb331c725858be02b1436
 SHA512:
-  metadata.gz: dd57b5cf5c00bf1f8a69e2a555b7396899d4ba72634fe9e14fde7cf26d38b659dd91238b1bb93dcaae9767d88dbfb0fc9a69579cdbf1ff5cc2859368f7721a43
-  data.tar.gz: e7b7151e26748e91590065aa601f5be35455f91bb8f1e7ee0151c81c77316402f674c758a55c3a13b470f3b9ba5cc9aa9f1cbaee876d384d89c90bd40aabbcd2
+  metadata.gz: 79d3f5a870f219c8d27141736ead407d562cbaec8b19e78ed51e5d8e0793a65a27e4461e95ddb30c02f5aea5e8c4c8c69664e0823bbf58aa09b33889ff65921d
+  data.tar.gz: 06f7b7bd1184b9c32246cbcb5c31f8bddae8f892bab2c3c08ae53c9e326e82d631287c2684ac067cd1b87e507c93cd15afa0818994e91d2fd841e8fad42acb50

data/lib/box_office/cli.rb

@@ -14,44 +14,23 @@ class BoxOffice::CLI
   def list_movies
     puts "---"
     puts "Last Weekend's Box Office:".colorize(:red)
-    @movies_list.each_with_index do |(movie, earnings), i|
-      puts "#{i + 1}.".colorize(:blue) + " #{movie}, #{earnings}"
+    BoxOffice::Movie.all.each_with_index do |movie, i|
+      puts "#{i + 1}.".colorize(:blue) + " #{movie.title}, #{movie.earnings}"
     end
     puts "---"
   end
 
-  def add_attributes_to_movie(user_input)
-    movie = BoxOffice::Movie.all[user_input]
-    attributes = BoxOffice::Scraper.scrape_movie_page(user_input)
-    movie.add_movie_attributes(attributes)
-  end
-
-  def display_movie_info(user_input)
-    movie = BoxOffice::Movie.all[user_input]
-    puts "---"
-    puts "#{movie.title}".colorize(:red)
-    puts "#{movie.synopsis}" if movie.synopsis != ""
-    puts "" if movie.synopsis != ""
-    puts "Genres:".colorize(:blue) + " #{movie.genres}" if !movie.genres.nil?
-    puts "Rating:".colorize(:blue) + " #{movie.rating}" if !movie.rating.nil?
-    puts "Studio:".colorize(:blue) + " #{movie.studio}" if !movie.studio.nil?
-    puts "Director:".colorize(:blue) + " #{movie.director}" if !movie.director.nil?
-    puts "Writers:".colorize(:blue) + " #{movie.writers}" if !movie.writers.nil?
-    puts "Cast:".colorize(:blue) + " #{movie.cast}" if movie.cast != ""
-    puts "Release Date:".colorize(:blue) + " #{movie.release_date}" if !movie.release_date.nil?
-    puts "Runtime:".colorize(:blue) + " #{movie.runtime}" if !movie.runtime.nil?
-    puts "Critic Score:".colorize(:blue) + " #{movie.critic_score}"
-    puts "Audience Score:".colorize(:blue) + " #{movie.audience_score}"
-    puts "---"
-  end
-
   def menu
     input = nil
     until input == "exit"
-      puts "Enter movie number to see more info, 'list' to see the list again, or 'exit' to leave the app:".colorize(:green)
+      puts "What do you want to do?".colorize(:green)
+      puts "* Enter ".colorize(:green) + "movie number".colorize(:blue) + " to learn more about the movie".colorize(:green)
+      puts "* Type '".colorize(:green) + "list".colorize(:blue) + "' to see the list again".colorize(:green)
+      puts "* Type '".colorize(:green) + "exit".colorize(:blue) + "' to leave the app".colorize(:green)
+      puts "---"
       input = gets.strip.downcase
       if input.to_i.between?(1, BoxOffice::Movie.all.length)
-        add_attributes_to_movie(input.to_i - 1)
+        add_attributes_to_movie(input.to_i - 1) if BoxOffice::Movie.all[input.to_i - 1].rating.nil? # Only scrapes webpage if attributes are 'nil' to prevent scraping multiple times
         display_movie_info(input.to_i - 1)
       elsif input == "list"
         list_movies
@@ -61,7 +40,19 @@ class BoxOffice::CLI
     end
   end
 
+  def add_attributes_to_movie(user_input)
+    movie = BoxOffice::Movie.all[user_input]
+    attributes = BoxOffice::Scraper.scrape_movie_page(user_input)
+    movie.add_movie_attributes(attributes)
+  end
+
+  def display_movie_info(user_input)
+    movie = BoxOffice::Movie.all[user_input]
+    movie.print_info
+    puts "---"
+  end
+
   def goodbye
-    puts "Peace out homie!".colorize(:green) + " <3".colorize(:light_red)
+    puts "Peace out homie!".colorize(:cyan) + " <3".colorize(:light_red)
   end
 end

data/lib/box_office/movie.rb

@@ -1,18 +1,35 @@
 class BoxOffice::Movie
-  attr_accessor :title, :synopsis, :rating, :genres, :director, :writers, :cast, :critic_score, :audience_score, :runtime, :studio, :release_date
+  attr_accessor :title, :link, :earnings, :synopsis, :rating, :genres, :director,
+  :writers, :cast, :critic_score, :audience_score, :runtime, :studio, :release_date
+
   @@all = []
 
-  def initialize(title)
-    @title = title
+  def initialize
     @@all << self
   end
 
-  def add_movie_attributes(attr_hash)
-    attr_hash.each do |key, value|
-      self.send("#{key}=", value)
+  def add_movie_attributes(attributes_hash)
+    attributes_hash.each do |attr, value|
+      self.send("#{attr}=", value)
     end
   end
 
+  def print_info
+    puts "#{self.title}".colorize(:red)
+    puts "#{self.synopsis}" if self.synopsis != ""
+    puts "" if self.synopsis != ""
+    puts "Genres:".colorize(:blue) + " #{self.genres}" if !self.genres.nil?
+    puts "Rating:".colorize(:blue) + " #{self.rating}" if !self.rating.nil?
+    puts "Studio:".colorize(:blue) + " #{self.studio}" if !self.studio.nil?
+    puts "Director:".colorize(:blue) + " #{self.director}" if !self.director.nil?
+    puts "Writers:".colorize(:blue) + " #{self.writers}" if !self.writers.nil?
+    puts "Main Cast:".colorize(:blue) + " #{self.cast}" if self.cast != ""
+    puts "Release Date:".colorize(:blue) + " #{self.release_date}" if !self.release_date.nil?
+    puts "Runtime:".colorize(:blue) + " #{self.runtime}" if !self.runtime.nil?
+    puts "Critic Score:".colorize(:blue) + " #{self.critic_score}"
+    puts "Audience Score:".colorize(:blue) + " #{self.audience_score}"
+  end
+
   def self.all
     @@all
   end

data/lib/box_office/scraper.rb

@@ -1,27 +1,18 @@
 class BoxOffice::Scraper
-  @@movie_links = []
-
-  def self.scrape_movie_list # Creates Movie objects and creates hash with movie titles and earnings from scraped website
+  def self.scrape_movie_list # Scrapes website and creates Movie objects
     movie_list = Nokogiri::HTML(open("https://www.rottentomatoes.com/browse/box-office/?rank_id=0&country=us"))
-    titles = []
-    earnings = []
-
-    movie_list.css("table td.left a").each do |title|
-      @@movie_links << title.attr("href")
-      movie = BoxOffice::Movie.new(title.text)
-      titles << movie.title
-    end
 
-    movie_list.css("table td[7]").each do |earning|
-      earnings << earning.text
+    movie_list.css("table.center.table tr[itemprop='itemListElement']").each do |row|
+      movie = BoxOffice::Movie.new
+      movie.title = row.css("td.left a").text.strip
+      movie.link = row.css("td.left a").attr("href").text.strip
+      movie.earnings = row.css("td[7]").text.strip
     end
-
-    [titles, earnings].transpose.to_h
   end
 
-  def self.scrape_movie_page(index) # Scrapes movie webpage and creates hash of movie info
-    all_info = {}
-    movie_page = Nokogiri::HTML(open("https://www.rottentomatoes.com/#{@@movie_links[index]}"))
+  def self.scrape_movie_page(index) # Scrapes movie webpage and creates hash of movie attributes
+    attributes = {}
+    movie_page = Nokogiri::HTML(open("https://www.rottentomatoes.com/#{BoxOffice::Movie.all[index].link}"))
 
     # Scraping synopsis and cast
     movie_synopsis = movie_page.css("div#movieSynopsis").text.strip
@@ -40,41 +31,39 @@ class BoxOffice::Scraper
       movie_audience_score = movie_page.css("div.audience-score.meter").text.split[0]
     end
 
-    # Scraping basic movie info and storing it into a hash
+    # Scraping basic movie info and storing it in a hash
     info_keys = []
     info_values = []
-
     movie_page.css("ul.content-meta.info div.meta-label.subtle").each { |key| info_keys << key.text.strip }
     movie_page.css("ul.content-meta.info div.meta-value").each { |value| info_values << value.text.gsub(/\n\s*/, "").strip }
-
     info_hash = [info_keys, info_values].transpose.to_h
 
-    # Combining all scraped info into one hash called 'all_info'
+    # Combining all scraped info into one hash called 'attributes'
     info_hash.each do |key, value|
       case key
       when "Rating:"
-        all_info[:rating] = value
+        attributes[:rating] = value
       when "Genre:"
-        all_info[:genres] = value
+        attributes[:genres] = value
       when "Directed By:"
-        all_info[:director] = value
+        attributes[:director] = value
       when "Written By:"
-        all_info[:writers] = value
+        attributes[:writers] = value
       when "In Theaters:"
-        all_info[:release_date] = value
+        attributes[:release_date] = value
       when "Runtime:"
-        all_info[:runtime] = value
+        attributes[:runtime] = value
       when "Studio:"
-        all_info[:studio] = value
+        attributes[:studio] = value
       end
      end
 
-    all_info[:synopsis] = movie_synopsis
-    all_info[:critic_score] = movie_critic_score
-    all_info[:audience_score] = movie_audience_score
-    all_info[:cast] = movie_cast
+    attributes[:synopsis] = movie_synopsis
+    attributes[:critic_score] = movie_critic_score
+    attributes[:audience_score] = movie_audience_score
+    attributes[:cast] = movie_cast
 
-    all_info
+    attributes
   end
 
 end

data/lib/box_office/version.rb

@@ -1,3 +1,3 @@
 module BoxOffice
-  VERSION = "0.1.2"
+  VERSION = "0.1.3"
 end

metadata

@@ -1,14 +1,14 @@
 --- !ruby/object:Gem::Specification
 name: box-office-cli
 version: !ruby/object:Gem::Version
-  version: 0.1.2
+  version: 0.1.3
 platform: ruby
 authors:
 - Jason Mei
 autorequire: 
 bindir: bin
 cert_chain: []
-date: 2018-09-26 00:00:00.000000000 Z
+date: 2018-09-27 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: bundler
@@ -131,5 +131,5 @@ rubyforge_project:
 rubygems_version: 2.7.7
 signing_key: 
 specification_version: 4
-summary: A gem that provides information on last weekend's movie box office.
+summary: A simple gem that provides information on last weekend's movie box office.
 test_files: []