imdb_movies 1.0.5 → 1.0.6

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA1:
-  metadata.gz: 08eaec2d38a2b2ea5220be1f57eb3c0990325bd6
-  data.tar.gz: bf8cdb67a8dba50a0f965c95c9e16e389aa0df92
+  metadata.gz: 0358af84ea07e6522acd36a70735567d372d46a0
+  data.tar.gz: ae8259f1a5cedd56ed433956f827d388bff30957
 SHA512:
-  metadata.gz: df6ac6219b59a209c4c584ed86e5e49dee833525af76384b837b977b361ca8d49388f2687de31bdd2fe2527d63e25f5be7a7ccb21dd7419ae3086224e404d7f7
-  data.tar.gz: d88ddb4724b05455729679502a125f680c6afc3fc8f3b4fb1cba68e1f0f4f0007edd5a3c4c964b2d2a7680ef76f73e99728b3065aa3a206abbec49de4114bfb0
+  metadata.gz: d4ff009df862f8ca7cd9c95bcb02304cb17ca90ee9ad88307687489c2e8bd670553a7176bea63f59ddfd9aa4f2b1131e8459a8a000c1fde73eb018ae8abfc1d7
+  data.tar.gz: 31014695b1c78565ce2976ac9265a0a127c67a9d48acbc24a232f3c573d301cb467cc99bb035b36ad59b47398c606fdd714e83bbd0f261d2bcf06d0aaf1f43cd

data/lib/imdb_movies/cli.rb

@@ -1,16 +1,22 @@
 # CLI Controller
-  class ImdbMovies::CLI
+class ImdbMovies::CLI
 
-	attr_accessor :imdb_page, :nav_links, :movie_links
+	attr_accessor :opening_movie_names, :opening_week, :now_playing_movie_names, :coming_soon_movie_names, :coming_soon_week, :all_movie_names, :all_movie_links
 
 	SKIP_MENU_DISPLAY = ["1", "2", "3", "opening this week", "now playing", "coming soon", "exit", "quit", ""]
 
 	def initialize
-		@nav_links, @movie_links = [], []
-		@imdb_page = ImdbScraper.new.scrape
+		
+		imdb_movies_info = ImdbScraper.new.scrape
+
+		@opening_movie_names = imdb_movies_info[0]
+		@opening_week = imdb_movies_info[1]
+		@now_playing_movie_names = imdb_movies_info[2]
+		@coming_soon_movie_names = imdb_movies_info[3]
+		@coming_soon_week = imdb_movies_info[4]
+		@all_movie_names = imdb_movies_info[5]
+		@all_movie_links = imdb_movies_info[6]
 
-		@imdb_page.each{|cat| @nav_links << cat[0][1] }
-		@imdb_page.each{|cat| cat[1].movie_links.each{|movie| @movie_links << movie}}
 	end
 
 	# Main CLI
@@ -23,52 +29,72 @@
 
 		until user_input == "exit" || user_input == "quit"
 			puts "\nYou can enter 1-3, a specific movie name, 'menu' or 'exit/quit'."
-			print "Input : "
+			puts "The movie name does not have to be case sensitive to get the info."
+			print "\nInput : "
 
 			user_input = gets.strip.downcase
-			input(user_input, movie_links)
+			input(user_input)
 
 			# Only displays the menu again if they entered a movie name or 'menu'
 			main_menu if !SKIP_MENU_DISPLAY.include?(user_input)
 		end
 
-		puts "\n\nThank You for using IMDB Movies."
+		puts "\nThank You for using IMDB Movies."
 	end
 
-	# Displays categories (opening this week, now playing, coming soon) 
+	# Displays movie categories (now playing, opening this week, coming soon) as well as first 5 movies if applicable
 	def main_menu
 
+		# Displays movies already in theaters
+		puts "\n1 - Now Playing"
+		@now_playing_movie_names[0...5].each {|movie_name| puts "\t\t#{movie_name}"}
+		puts "\tEnter '1' to see more movies that are playing."
+
+		# Displays opening week movies
+		puts "\n2 - #{opening_week}"
+		@opening_movie_names[0...5].each {|movie_name| puts "\t\t#{movie_name}"}
+		puts "\tEnter '2' to see more movies that are opening this week."
+
 		# Displays categories
-		@imdb_page.each.with_index(1){|section, index| 
-			puts "\n#{index} - #{section[0][0].gsub(/\(\w+\)/, '')}"
-			section[1].movie_links[0...5].each{|movie|
-				puts "\t\t#{movie[0]}"
-			}
-			puts "\tEnter '#{index}' to see more."
-		}
+		puts "\n3 - #{coming_soon_week}"
+		@coming_soon_movie_names[0...5].each {|movie_name| puts "\t\t#{movie_name}"}
+		puts "\tEnter '3' to see more movies that are coming after this week."
 	end
 
 	# Gets user's input and calls a different navigation methood based on input
-	def input(user_input, movie_links)
+	def input(user_input)
 
 		case user_input
 		when "",  "menu", "exit", "quit"
 		when "1", "opening this week"
-			Display.new(@imdb_page[0][1]).opening
+			display_movie_names(@opening_movie_names)
 		when "2", "now playing"
-			Display.new(@imdb_page[1][1]).now_playing
+			display_movie_names(@now_playing_movie_names)
 		when "3", "coming soon"
-			Display.new(@imdb_page[2][1]).coming_soon
+			display_movie_names(@coming_soon_movie_names)
 
-		# When movie is entered it calls Display.movie
-		# Otherwise it outputs a message.
+		# When movie is found is entered it creates a new instance of Movie from
 		else
-			# Nil if movie was not found
-			movie_link = movie_links.detect { |m| 
-				m[0].downcase.gsub(/[^\w\s]+/, '').squeeze(" ").include?(user_input) }
-
-			movie_link ? (Display.new().movie(movie_link[1])) : (puts "\nI'm sorry I don't know what you mean.")
+			# Search inside the @all_movie_names array. If you user input is included inside a movie title then store the index you found it at
+			found_movie_index = @all_movie_names.index{|movie_name| movie_name.downcase.gsub(/[^\w\s]+/, '').squeeze(" ").include?(user_input)}
+
+			# Create a new instance of Movie and have it display the info
+			# Otherwise show a message saying you're not sure what that was and display the menu again
+			if found_movie_index
+				movie = Movie.new(@all_movie_links[found_movie_index])
+				movie.display_info
+
+				print "\nDo you want to see the trailer on Youtube? Y / N "
+				input = gets.strip.downcase
+				movie.open_trailer if input == "yes" || input == "y"
+			else
+				puts "\nI'm sorry I don't know what you mean."
+			end
 		end
 	end
 
+	def display_movie_names(movie_names)
+		movie_names.each{|movie_name| puts "\t\t#{movie_name}"}
+	end
+
 end

data/lib/imdb_movies/imdb_scraper.rb

@@ -3,49 +3,65 @@
 
 	IMDB = "http://www.imdb.com"
 
-	attr_accessor :category_info
+	attr_accessor :opening_movie_names, :opening_movie_links, :opening_week, :now_playing_movie_names, :now_playing_movie_links, :coming_soon_movie_names, :coming_soon_movie_links, :coming_soon_week, :all_movie_names, :all_movie_links
 
 	def initialize
-		@category_info = []
+		@opening_movie_names = []
+		@opening_movie_links = []
+		@opening_week = ""
+		@now_playing_movie_names = []
+		@now_playing_movie_links = []
+		@coming_soon_movie_names = []
+		@coming_soon_movie_links = []
+		@coming_soon_week = ""
+		@all_movie_names = []
+		@all_movie_links = []
 	end
 
 	def scrape
 		doc = Nokogiri::HTML(open(IMDB))
+
+		# Gets the HTML for the divs on the sidebar
 		sections = doc.css("#sidebar  div.aux-content-widget-2")
 
-		categories = []
+		# For each section, it gets and stores the movie names, links and the week if applicable
+		sections.each do |section| 
 
-		sections.each{|s| 
-			category = s.css(".widget_header .oneline h3").text
-			begin
-				category_link = IMDB + s.css(".seemore a").attr("href").value if category != ""
-			rescue NoMethodError
-			end
+			# Stores the category header in category. 
+			# We only want the movies, not the television or social media
+			category =  section.css(".widget_header .oneline h3").text
+
+			case category
+				when "Opening This Week", "Now Playing (Box Office)", "Coming Soon"
+
+					# Gets the see more url so that MovieLinksScraper knows where to get the movie links from
+					category_link = IMDB + section.css(".seemore a").attr("href").value
+
+					# Looks inside the linked page to get the movies for that category
+					# Stores the array of movie names and links inside @movie_names and @movie_links to be flattened later
+					category_movies = MovieLinksScraper.new(category_link)
+					case category
+						when "Opening This Week"
+							@opening_movie_names, @opening_movie_links, @opening_week = category_movies.opening
+						when "Now Playing (Box Office)"
+							@now_playing_movie_names, @now_playing_movie_links = category_movies.now_playing
+						when "Coming Soon"
+							@coming_soon_movie_names, @coming_soon_movie_links, @coming_soon_week = category_movies.coming_soon
+							@coming_soon_week = "Coming Soon - Week of " + @coming_soon_week
+					end
 
-			begin
-				category_movies = MovieLinksScraper.new(category_link)
-				case category
-				when "Opening This Week"
-					category_movies.opening
-				when "Now Playing (Box Office)"
-					category_movies.now_playing
-				when "Coming Soon"
-					category_movies.coming_soon
 				end
-				movie_links = category_movies
-			rescue NoMethodError
 			end
 
-			# Only save parts with movies
-			# IMDB's site uses the same classes for every sidebar div
-			# No ids to differentiate movies from tv shows and social media
-			case category 
-			when "Opening This Week", "Now Playing (Box Office)", "Coming Soon"
-				categories.push([[category, category_link], movie_links])			
-			end
-		}
+			@all_movie_names = [@opening_movie_names, @now_playing_movie_names, @coming_soon_movie_names].flatten
+			@all_movie_links = [@opening_movie_links, @now_playing_movie_links, @coming_soon_movie_links].flatten
+
+			# Returning all the categories, movie names and links to the CLI to store
+			# We are sending back the category's link so that the Display class can look inside that page and get each movie name
+			# The CLI will not know what movies belong in which category, so it doesn't know how to display them
+			# The CLI calls the Display class so it can search for 
 
-		categories
+			return @opening_movie_names, @opening_week, @now_playing_movie_names, @coming_soon_movie_names, @coming_soon_week, @all_movie_names, @all_movie_links
 	end
 
 end

data/lib/imdb_movies/movie.rb

@@ -1,86 +1,103 @@
-# Accepts a url from IMDB on initialization and scrapes info for that movie
+# Accepts a url from IMDB on initialization and scrapes info from IMDB
   class Movie
-	attr_accessor :url, :name, :movie_rating, :length, :genres, :release_date, :imdb_rating, :description, :directors, :writers, :cast, :trailer_link
+		attr_accessor :url, :name, :movie_rating, :length, :genres, :release_date, :imdb_rating, :description, :directors, :writers, :cast, :trailer_link
 
-	def initialize(url)
-		@url = url
+		def initialize(url)
+			@url = url
+			scrape_info
+		end
+
+		# Scrapes and stores movie's info from IMDB page
+		def scrape_info
+			info = Nokogiri::HTML(open(@url)).css(".article.title-overview")
+
+			@name = info.css("tbody h1.header span[itemprop = 'name']").text.strip
+			
+			begin
+				@movie_rating = info.css(".infobar meta[itemprop = 'contentRating']").attr("content").value.strip
+			rescue NoMethodError
+			end
+
+			@length = info.css(".infobar time[itemprop = 'duration']").text.strip
+			@release_date = info.css("a[title = 'See all release dates']").text.strip.gsub(/\s+/, " ")
+			@imdb_rating = info.css(".titlePageSprite.star-box-giga-star").text.strip
+			@description = info.css("p[itemprop = 'description']").text.strip
 
-		scrape_info
-	end
+			genres = info.css("span[itemprop = 'genre']")
+			@genres = []
+			genres.each{|g| @genres << g.text }
+			@genres = @genres.join(", ")
 
-	# Scrape and store info
-	def scrape_info
-		info = Nokogiri::HTML(open(@url)).css(".article.title-overview")
+			directors = info.css("div[itemprop = 'director'] span[itemprop = 'name']")
+			@directors = []
+			directors.each{|d| @directors << d.text}
 
-		@name = info.css("tbody h1.header span[itemprop = 'name']").text.strip
-		
-		begin
-			@movie_rating = info.css(".infobar meta[itemprop = 'contentRating']").attr("content").value.strip
-		rescue NoMethodError
+			writers = info.css("div[itemprop = 'creator'] span[itemprop = 'name']")
+			@writers = []
+			writers.each{|w| @writers << w.text}
+
+			cast = info.css("div[itemprop = 'actors'] span[itemprop = 'name']")
+			@cast = []
+			cast.each{|c| @cast << c.text}
+			
+			@trailer_link = trailer
 		end
 
-		@length = info.css(".infobar time[itemprop = 'duration']").text.strip
-		@release_date = info.css("a[title = 'See all release dates']").text.strip.gsub(/\s+/, " ")
-		@imdb_rating = info.css(".titlePageSprite.star-box-giga-star").text.strip
-		@description = info.css("p[itemprop = 'description']").text.strip
-
-		genres = info.css("span[itemprop = 'genre']")
-		@genres = []
-		genres.each{|g| @genres << g.text }
-		@genres = @genres.join(", ")
-
-		directors = info.css("div[itemprop = 'director'] span[itemprop = 'name']")
-		@directors = []
-		directors.each{|d| @directors << d.text}
-
-		writers = info.css("div[itemprop = 'creator'] span[itemprop = 'name']")
-		@writers = []
-		writers.each{|w| @writers << w.text}
-
-		cast = info.css("div[itemprop = 'actors'] span[itemprop = 'name']")
-		@cast = []
-		cast.each{|c| @cast << c.text}
-		
-		@trailer_link = trailer
-	end
-
-	# Scrapes Youtube search results for first video link
-	# 	searches movie name + ' trailer'
-	def trailer
-		name = @name.downcase.split(/\W+/).join("+")
-		youtube_search = "https://www.youtube.com/results?search_query="
-		youtube_search_query = Nokogiri::HTML(open(youtube_search + name + "+trailer",  ssl_verify_mode: OpenSSL::SSL::VERIFY_NONE))
-		
-		trailer_link = "https://youtube.com" + youtube_search_query.css("#results ol.section-list li ol.item-section li div.yt-lockup-dismissable div.yt-lockup-content h3.yt-lockup-title a").attr("href").value
-	end
-
-	def display_info
-		puts "\n#{@name}\n\n"
-		puts "Movie Rating: \t#{@movie_rating}"
-		puts "Length: \t#{@length}"
-		puts "Genre: \t\t#{@genres}"
-		puts "Release Date: \t#{@release_date}"
-		@imdb_rating != "" ? (puts "IMDB Rating: \t#{@imdb_rating}/10.0") : (puts "IMDB Rating: \tNo Ratings Yet")
-		puts "Description: \t#{@description}"
-		puts "Director(s): \t#{@directors.join(", ")}"
-		puts "Writer(s): \t#{@writers.join(", ")}"
-		puts "Cast: \t\t#{@cast.join(", ")}"
-		puts "\nTrailer: \t#{@trailer_link}"
-	end
-
-	def open_trailer
-		begin
-			exec("open #{@trailer_link}")
-		rescue SystemCallError
+		# Scrapes Youtube search results for first video link
+		# 	searches movie name + ' trailer'
+		def trailer
+			movie_name = @name.downcase.split(/\W+/).join("+")
+			youtube_search = "https://www.youtube.com/results?search_query=" + movie_name + "+trailer"
+
+			# Scrapes HTML for Youtube search results
+			# 'ssl_verify_mode: OpenSSL::SSL::VERIFY_NONE' is to get past SSL verification for open-uri when getting Youtube's HTML
+			youtube_search_query = Nokogiri::HTML(open(youtube_search, ssl_verify_mode: OpenSSL::SSL::VERIFY_NONE))
+			
+			# Scrape first link
+			trailer_link = "https://youtube.com" + youtube_search_query.css("#results ol.section-list li ol.item-section li div.yt-lockup-dismissable div.yt-lockup-content h3.yt-lockup-title a").attr("href").value
 		end
-		begin
-			exec("start #{@trailer_link}")
-		rescue SystemCallError
+
+		def display_info
+			puts "\n#{@name}\n\n"
+			puts "Movie Rating: \t#{@movie_rating}"
+			puts "Length: \t#{@length}"
+			puts "Genre: \t\t#{@genres}"
+			puts "Release Date: \t#{@release_date}"
+
+			if @imdb_rating != ""
+				puts "IMDB Rating: \t#{@imdb_rating}/10.0"
+			else
+				puts "IMDB Rating: \tNo Ratings Yet"
+			end
+			
+			puts "Description: \t#{@description}"
+			puts "Director(s): \t#{@directors.join(", ")}"
+			puts "Writer(s): \t#{@writers.join(", ")}"
+			puts "Cast: \t\t#{@cast.join(", ")}"
+			puts "\nTrailer: \t#{@trailer_link}"
 		end
-		begin
-			exec("xdg-open #{@trailer_link}")
-		rescue SystemCallError
+
+		# Executes command to open youtube link inside browser
+		def open_trailer
+
+			# Mac OS X
+			begin
+				exec("open #{@trailer_link}")
+			rescue SystemCallError
+			end
+
+			# Windows
+			begin
+				exec("start #{@trailer_link}")
+			rescue SystemCallError
+			end
+
+			# Linux/Unix
+			begin
+				exec("xdg-open #{@trailer_link}")
+			rescue SystemCallError
+			end
+
 		end
-	end
 end
 

data/lib/imdb_movies/movie_links_scraper.rb

@@ -1,46 +1,63 @@
   class MovieLinksScraper
-	attr_accessor :movie_links, :url, :opening_week
+		attr_accessor  :url, :movie_names, :movie_links, :opening_week, :upcomming_week
 
-	IMDB = "http://www.imdb.com"
+		IMDB = "http://www.imdb.com"
 
-	def initialize(url)
-		# Array of arrays, each with movie name and link
-		@movie_links = []
+		def initialize(url)
+			@url = url
+			@movie_names = []
+			@movie_links = []
+			@opening_week = ""
+			@comming_soon_week = ""
+		end
+
+		# Gets the links and opening week from IMDB for the movies that are opening up this week
+		# The movie names and links to the movies' IMDB pages are returned back to ImdbScraper
+		def opening
+			doc = Nokogiri::HTML(open(@url))
+			movie_links = doc.css(".article .list")[1].css(".list_item h4[itemprop = 'name'] a")
+			add_movie_name_and_links(movie_links)
 
-		@url = url
-	end
+			# Stores the header that is shown on the opening week movies page
+			# Example: 'Opening This Week - January 22' is stored
+			@opening_week = doc.css(".article .list h3").first.text
 
-	def opening
-		doc = Nokogiri::HTML(open(@url))
-		movie_links = doc.css(".article .list")[1].css(".list_item h4[itemprop = 'name'] a")
-		begin
-			movie_links.each{|m_link| @movie_links.push( [ m_link.text.gsub(/\(\w+\)/, ""), (IMDB + m_link.attr("href")) ] )}
-		rescue NoMethodError
+			return @movie_names, @movie_links, @opening_week
 		end
 
-		@opening_week = doc.css(".article .list h3").first.text
-		@movie_links
-	end
+		# Gets the links from IMDB for the movies that are popular and in theaters now
+		# The movie names and links to the movies' IMDB pages are returned back to ImdbScraper
+		def now_playing
+			doc = Nokogiri::HTML(open(@url))
+			movie_links = doc.css("tbody tr td.titleColumn a")
+			add_movie_name_and_links(movie_links)
 
-	def now_playing
-		doc = Nokogiri::HTML(open(@url))
-		movie_links = doc.css("tbody tr td.titleColumn a")
-		begin
-			movie_links.each{|m_link| @movie_links.push( [ m_link.text.gsub(/\(\w+\)/, ""), (IMDB + m_link.attr("href")) ] )}		
-		rescue NoMethodError
+			return @movie_names, @movie_links
 		end
-		@movie_links
-	end
-
-	def coming_soon
-		doc = Nokogiri::HTML(open(@url))
-		movie_links = doc.css(".list.detail .list_item h4[itemprop = 'name'] a")
-		begin
-			movie_links.each{|m_link| @movie_links.push( [ m_link.text.gsub(/\(\w+\)/, ""), (IMDB + m_link.attr("href")) ] )}		
-		rescue NoMethodError
+
+		# Gets the links and opening week from IMDB for the movies that are comming up in 2 weeks
+		# The movie names and links to the movies' IMDB pages are returned back to ImdbScraper
+		def coming_soon
+			doc = Nokogiri::HTML(open(@url))
+			movie_links = doc.css(".list.detail .list_item h4[itemprop = 'name'] a")
+			add_movie_name_and_links(movie_links)
+
+			# Stores the header that is show on the coming soon movies page
+			# Example: 'January 29' is stored
+			@coming_soon_week = doc.css(".list.detail h4.li_group a").first.text
+			return @movie_names, @movie_links, @coming_soon_week
 		end
 
-		@opening_week = doc.css(".list.detail h4.li_group a").first.text
-		@movie_links
-	end
+		# Gets an array movie links that were scraped from an IMDB
+		# Each movie link has the movie name as the text.
+		# We are pushing each movie name and link inside @movie_names and @movie_links
+		def add_movie_name_and_links(movie_links)
+			# Pulls anchors which are movie names with links to that movie's page on IMDB
+			movie_links.each do |link|
+				# Gets the movie name from text
+				@movie_names << link.text.gsub(/\(\w+\)/, '')
+				# Gets the movie's url
+				@movie_links << IMDB + link.attr("href")	
+			end
+		end
 end

data/lib/imdb_movies/version.rb

@@ -1,3 +1,3 @@
 module ImdbMovies	
-  VERSION = "1.0.5"
+  VERSION = "1.0.6"
 end

data/lib/imdb_movies.rb

@@ -1,6 +1,5 @@
 require_relative "./imdb_movies/version"
 require_relative "./imdb_movies/cli"
-require_relative "./imdb_movies/display"
 require_relative "./imdb_movies/imdb_scraper"
 require_relative "./imdb_movies/movie.rb"
 require_relative "./imdb_movies/movie_links_scraper"

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: imdb_movies
 version: !ruby/object:Gem::Version
-  version: 1.0.5
+  version: 1.0.6
 platform: ruby
 authors:
 - azheng249
@@ -102,7 +102,6 @@ files:
 - imdb_movies.gemspec
 - lib/imdb_movies.rb
 - lib/imdb_movies/cli.rb
-- lib/imdb_movies/display.rb
 - lib/imdb_movies/imdb_scraper.rb
 - lib/imdb_movies/movie.rb
 - lib/imdb_movies/movie_links_scraper.rb

data/lib/imdb_movies/display.rb

@@ -1,35 +0,0 @@
-  class Display
-
-	attr_accessor :links
-
-	def initialize(links = nil)
-		@links = links
-	end
-
-	def opening
-		puts @links.opening_week + "\n\n"
-		@links.movie_links.each{|m| puts "\t\t#{m[0]}"}
-
-	end
-
-	def now_playing
-		puts "Now Playing\n\n"
-		@links.movie_links.each{|m| puts "\t\t#{m[0]}"}	
-	end
-
-	def coming_soon
-		puts "Coming Soon - \t" + @links.opening_week + "\n\n"
-		@links.movie_links.each{|m| puts "\t\t#{m[0]}"}	
-	end
-
-	def movie(movie_url)
-		movie = Movie.new(movie_url)
-		movie.display_info
-
-		print "\nDo you want to see the trailer for this movie (Y/N):\t"
-		user_input = gets.strip.downcase
-
-		movie.open_trailer if user_input == "y" || user_input == "yes"
-	end
-
-end