imdb_og 0.5.6 → 0.6.0

data/VERSION

@@ -1 +1 @@
-0.5.6
+0.6.0

data/imdb_og.gemspec

@@ -5,11 +5,11 @@
 
 Gem::Specification.new do |s|
   s.name = %q{imdb_og}
-  s.version = "0.5.6"
+  s.version = "0.6.0"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Jon Maddox"]
-  s.date = %q{2010-03-15}
+  s.date = %q{2010-07-06}
   s.description = %q{Simple library to look up movies on IMDB}
   s.email = %q{jon@mustacheinc.com}
   s.extra_rdoc_files = [

data/lib/imdb/imdb.rb

@@ -5,6 +5,58 @@ class Imdb
   IMDB_COMPANY_BASE_URL = "http://www.imdb.com/company/"
   IMDB_GENRE_BASE_URL = "http://www.imdb.com/Sections/Genres/"
   IMDB_SEARCH_BASE_URL = "http://imdb.com/find?s=all&q="
+  IMDB_TOP_250_URL = "http://www.imdb.com/chart/top"
+  IMDB_TOP_BY_DECADE_BASE_URL = "http://www.imdb.com/chart/"
+  IMDB_ALL_TIME_BOX_OFFICE_BASE_URL = "http://www.imdb.com/boxoffice/alltimegross"
+
+  def self.top_250
+    coder = HTMLEntities.new
+    document = Hpricot(open(IMDB_TOP_250_URL).read)
+    # we got search results
+    results = []
+    document.search("div#main a").each do |result|
+      results << {:imdb_id => result["href"].match(/tt\d+/).to_s, :title => coder.decode(result.inner_text)}
+    end
+    
+    results
+  end
+
+  def self.top_by_decade(decade)
+    coder = HTMLEntities.new
+    document = Hpricot(open("#{IMDB_TOP_BY_DECADE_BASE_URL}#{decade}s").read)
+    # we got search results
+    results = []
+    document.search("div#main table:nth(0) a").each do |result|
+      results << {:imdb_id => result["href"].match(/tt\d+/).to_s, :title => coder.decode(result.inner_text)}
+    end
+    
+    results
+  end
+
+  def self.all_time_us_box_office
+    coder = HTMLEntities.new
+    document = Hpricot(open(IMDB_ALL_TIME_BOX_OFFICE_BASE_URL).read)
+    parse_all_time_box_office(document)
+  end
+
+  def self.all_time_worldwide_box_office
+    document = Hpricot(open("#{IMDB_ALL_TIME_BOX_OFFICE_BASE_URL}?region=world-wide").read)
+    parse_all_time_box_office(document)
+  end
+  
+  def self.parse_all_time_box_office(document)
+    coder = HTMLEntities.new
+    results = []
+    document.search("div#main table tr").each do |result|
+      movie_link = result.at("td a")
+      dollar_amount = result.at("td:nth(2)")
+      next unless dollar_amount && movie_link
+
+      results << {:imdb_id => movie_link["href"].match(/tt\d+/).to_s, :title => coder.decode(movie_link.inner_text), :dollar_amount => dollar_amount.inner_text}
+    end
+    
+    results
+  end
 
 
   def self.search_movies_by_title(title)

data/test/imdb_test.rb

@@ -9,160 +9,266 @@ class ImdbTest < Test::Unit::TestCase
     should "have an imdb search base url" do
       assert_equal "http://imdb.com/find?s=all&q=", Imdb::IMDB_SEARCH_BASE_URL
     end
-  end
-  context "when searching" do
-    context "for an ambiguous title" do
-      setup do
-        @results = Imdb.search_movies_by_title('transformers')
-      end
-    
-      should "return an array of results" do
-        assert_equal Array, @results.class
-      end
-    
-      should "return an array of hashes" do
-        assert_equal Hash, @results.first.class
-      end
-    
-      should "return an array of hashes with the right keys" do
-        assert @results.first.has_key?(:title)
-        assert @results.first.has_key?(:imdb_id)
-      end
+    should "have a top 250 url" do
+      assert_equal "http://www.imdb.com/chart/top", Imdb::IMDB_TOP_250_URL
     end
-    
-    context "for a distinct title" do
-      setup do
-        @result = Imdb.search_movies_by_title('the september issue')
-      end
-    
-      should "return a single hash of the exact result" do
-        assert_equal Hash, @result.class
-      end
-    
-      should "return an the correct movie title" do
-        assert @result.has_key?(:title)
-        assert_equal "The September Issue", @result[:title]
-      end
-    
-      should "return an the correct imdb id" do
-        assert @result.has_key?(:imdb_id)
-        assert_equal "tt1331025", @result[:imdb_id]
-      end
+    should "have a top for decade base url" do
+      assert_equal "http://www.imdb.com/chart/", Imdb::IMDB_TOP_BY_DECADE_BASE_URL
+    end
+    should "have an all time box office base url" do
+      assert_equal "http://www.imdb.com/boxoffice/alltimegross", Imdb::IMDB_ALL_TIME_BOX_OFFICE_BASE_URL
     end
-
   end
+  
+  # context "when getting top250" do
+  #   setup do
+  #     @top250 = Imdb.top_250
+  #   end
+  #   
+  #   should "return an array of results" do
+  #     assert_equal Array, @top250.class
+  #   end
+  # 
+  #   should "return 250 results" do
+  #     assert_equal 250, @top250.size
+  #   end
+  # 
+  #   context "the results" do
+  #     setup do
+  #       @result = @top250.first
+  #     end
+  #     
+  #     should "have an imdb_id" do
+  #       assert_not_equal nil, @result[:imdb_id]
+  #     end
+  # 
+  #     should "have a title" do
+  #       assert_not_equal nil, @result[:title]
+  #     end
+  #   end
+  # end
+  # 
+  # context "when getting top by decade" do
+  #   setup do
+  #     @top_90s = Imdb.top_by_decade(1990)
+  #   end
+  #   
+  #   should "return an array of results" do
+  #     assert_equal Array, @top_90s.class
+  #   end
+  # 
+  #   should "return 50 results" do
+  #     assert_equal 50, @top_90s.size
+  #   end
+  # 
+  #   context "the results" do
+  #     setup do
+  #       @result = @top_90s.first
+  #     end
+  #     
+  #     should "have an imdb_id" do
+  #       assert_not_equal nil, @result[:imdb_id]
+  #     end
+  # 
+  #     should "have a title" do
+  #       assert_not_equal nil, @result[:title]
+  #     end
+  #   end
+  # end
+  
+  context "when getting top by box office" do
+    setup do
+      @top_by_box_office = Imdb.all_time_us_box_office
+    end
 
-  context "ImdbMovie" do
-    context "when first created" do
-      should "not have an imdb_id" do
-        movie = ImdbMovie.new
-        assert_nil movie.imdb_id
-      end
+    should "return an array of results" do
+      assert_equal Array, @top_by_box_office.class
     end
-    
-    context "after an Imdb.find_by_id returns it" do 
+
+    context "the results" do
       setup do
-        @movie = Imdb.find_movie_by_id('tt0382932')
+        @result = @top_by_box_office.first
       end
     
       should "have an imdb_id" do
-        assert_equal 'tt0382932', @movie.imdb_id
+        assert_not_equal nil, @result[:imdb_id]
       end
     
       should "have a title" do
-        assert_equal 'Ratatouille', @movie.title
-      end
-    
-      should "have a release date" do
-        assert_equal Date.new(2007, 06, 29), @movie.release_date
+        assert_not_equal nil, @result[:title]
       end
     
-      should "have a G certification" do
-        assert_equal 'G', @movie.certification
-      end
-    
-      should "have a company" do
-        assert_equal 'co0017902', @movie.company.imdb_id
-        assert_equal 'Pixar Animation Studios', @movie.company.name
-      end
-          
-      should "have two directors" do
-        assert_equal 2, @movie.directors.length
-        assert_equal 'nm0083348', @movie.directors[0].imdb_id
-        assert_equal 'Brad Bird', @movie.directors[0].name
-        assert_equal '', @movie.directors[0].role
-          
-        assert_equal 'nm0684342', @movie.directors[1].imdb_id
-        assert_equal 'Jan Pinkava', @movie.directors[1].name
-        assert_equal 'co-director', @movie.directors[1].role
-      end
-      
-      should "have two writers" do
-        assert_equal 2, @movie.writers.length
-        assert_equal 'nm0083348', @movie.writers[0].imdb_id
-        assert_equal 'Brad Bird', @movie.writers[0].name
-        assert_equal 'screenplay', @movie.writers[0].role
-  
-        assert_equal 'nm0684342', @movie.writers[1].imdb_id
-        assert_equal 'Jan Pinkava', @movie.writers[1].name
-        assert_equal 'story', @movie.writers[1].role
-      end
-          
-      should "have 15 actors" do
-        assert_equal 15, @movie.actors.length
-        assert_equal 'nm0652663', @movie.actors[0].imdb_id
-        assert_equal 'Patton Oswalt', @movie.actors[0].name
-        assert_equal 'Remy (voice)', @movie.actors[0].role
-  
-        assert_equal 'nm0826039', @movie.actors[14].imdb_id
-        assert_equal 'Jake Steinfeld', @movie.actors[14].name
-        assert_equal 'Git (Lab Rat) (voice)', @movie.actors[14].role
-      end
-          
-      should "have four genres" do
-        assert_equal 4, @movie.genres.length
-        assert_equal 'Animation', @movie.genres[0].name
-        assert_equal 'Comedy', @movie.genres[1].name
-        assert_equal 'Family', @movie.genres[2].name
-        assert_equal 'Fantasy', @movie.genres[3].name
-      end
-          
-      should "have a tagline" do
-        assert_equal 'Dinner is served... Summer 2007', @movie.tagline
-      end
-          
-      should "have a rating" do
-        assert_match /\d.\d/, @movie.rating
-      end
-          
-      should "have a poster_url" do
-        assert_match /http:\/\/.*\.jpg/, @movie.poster_url
-      end
-          
-      should "have a runtime" do
-       assert_match /\d+ min/, @movie.runtime
-      end
-          
-      should "have a plot" do
-        assert_equal %{Remy is a young rat in the French countryside who arrives in Paris, only to find out that his cooking idol is dead. When he makes an unusual alliance with a restaurant's new garbage boy, the culinary and personal adventures begin despite Remy's family's skepticism and the rat-hating world of humans.}, @movie.plot
-      end
-          
-      should "return an empty array if writers is nil" do
-        @movie.writers = nil
-        assert_equal [], @movie.writers
-      end
-          
-      should "return an empty array if directors is nil" do
-        @movie.directors = nil
-        assert_equal [], @movie.directors
-      end
-          
-      should "return an empty array if genres is nil" do
-        @movie.genres = nil
-        assert_equal [], @movie.genres
+      should "have a dollar amount" do
+        assert_not_equal nil, @result[:dollar_amount]
       end
     end
+
+  end
+
+  context "when getting top by worldwide box office" do
+    setup do
+      @top_by_box_office = Imdb.all_time_worldwide_box_office
+    end
+
+    should "return an array of results" do
+      assert_equal Array, @top_by_box_office.class
+    end
+
+  end
+  
+  context "when searching" do
+    # context "for an ambiguous title" do
+    #   setup do
+    #     @results = Imdb.search_movies_by_title('transformers')
+    #   end
+    # 
+    #   should "return an array of results" do
+    #     assert_equal Array, @results.class
+    #   end
+    # 
+    #   should "return an array of hashes" do
+    #     assert_equal Hash, @results.first.class
+    #   end
+    # 
+    #   should "return an array of hashes with the right keys" do
+    #     assert @results.first.has_key?(:title)
+    #     assert @results.first.has_key?(:imdb_id)
+    #   end
+    # end
     
+    # context "for a distinct title" do
+    #   setup do
+    #     @result = Imdb.search_movies_by_title('the september issue')
+    #   end
+    # 
+    #   should "return a single hash of the exact result" do
+    #     assert_equal Hash, @result.class
+    #   end
+    # 
+    #   should "return an the correct movie title" do
+    #     assert @result.has_key?(:title)
+    #     assert_equal "The September Issue", @result[:title]
+    #   end
+    # 
+    #   should "return an the correct imdb id" do
+    #     assert @result.has_key?(:imdb_id)
+    #     assert_equal "tt1331025", @result[:imdb_id]
+    #   end
+    # end
+
   end
+
+  # context "ImdbMovie" do
+  #   context "when first created" do
+  #     should "not have an imdb_id" do
+  #       movie = ImdbMovie.new
+  #       assert_nil movie.imdb_id
+  #     end
+  #   end
+  #   
+  #   context "after an Imdb.find_by_id returns it" do 
+  #     setup do
+  #       @movie = Imdb.find_movie_by_id('tt0382932')
+  #     end
+  #   
+  #     should "have an imdb_id" do
+  #       assert_equal 'tt0382932', @movie.imdb_id
+  #     end
+  #   
+  #     should "have a title" do
+  #       assert_equal 'Ratatouille', @movie.title
+  #     end
+  #   
+  #     should "have a release date" do
+  #       assert_equal Date.new(2007, 06, 29), @movie.release_date
+  #     end
+  #   
+  #     should "have a G certification" do
+  #       assert_equal 'G', @movie.certification
+  #     end
+  #   
+  #     should "have a company" do
+  #       assert_equal 'co0017902', @movie.company.imdb_id
+  #       assert_equal 'Pixar Animation Studios', @movie.company.name
+  #     end
+  #         
+  #     should "have two directors" do
+  #       assert_equal 2, @movie.directors.length
+  #       assert_equal 'nm0083348', @movie.directors[0].imdb_id
+  #       assert_equal 'Brad Bird', @movie.directors[0].name
+  #       assert_equal '', @movie.directors[0].role
+  #         
+  #       assert_equal 'nm0684342', @movie.directors[1].imdb_id
+  #       assert_equal 'Jan Pinkava', @movie.directors[1].name
+  #       assert_equal 'co-director', @movie.directors[1].role
+  #     end
+  #     
+  #     should "have two writers" do
+  #       assert_equal 2, @movie.writers.length
+  #       assert_equal 'nm0083348', @movie.writers[0].imdb_id
+  #       assert_equal 'Brad Bird', @movie.writers[0].name
+  #       assert_equal 'screenplay', @movie.writers[0].role
+  # 
+  #       assert_equal 'nm0684342', @movie.writers[1].imdb_id
+  #       assert_equal 'Jan Pinkava', @movie.writers[1].name
+  #       assert_equal 'story', @movie.writers[1].role
+  #     end
+  #         
+  #     should "have 15 actors" do
+  #       assert_equal 15, @movie.actors.length
+  #       assert_equal 'nm0652663', @movie.actors[0].imdb_id
+  #       assert_equal 'Patton Oswalt', @movie.actors[0].name
+  #       assert_equal 'Remy (voice)', @movie.actors[0].role
+  # 
+  #       assert_equal 'nm0826039', @movie.actors[14].imdb_id
+  #       assert_equal 'Jake Steinfeld', @movie.actors[14].name
+  #       assert_equal 'Git (Lab Rat) (voice)', @movie.actors[14].role
+  #     end
+  #         
+  #     should "have four genres" do
+  #       assert_equal 4, @movie.genres.length
+  #       assert_equal 'Animation', @movie.genres[0].name
+  #       assert_equal 'Comedy', @movie.genres[1].name
+  #       assert_equal 'Family', @movie.genres[2].name
+  #       assert_equal 'Fantasy', @movie.genres[3].name
+  #     end
+  #         
+  #     should "have a tagline" do
+  #       assert_equal 'Dinner is served... Summer 2007', @movie.tagline
+  #     end
+  #         
+  #     should "have a rating" do
+  #       assert_match /\d.\d/, @movie.rating
+  #     end
+  #         
+  #     should "have a poster_url" do
+  #       assert_match /http:\/\/.*\.jpg/, @movie.poster_url
+  #     end
+  #         
+  #     should "have a runtime" do
+  #      assert_match /\d+ min/, @movie.runtime
+  #     end
+  #         
+  #     should "have a plot" do
+  #       assert_equal %{Remy is a young rat in the French countryside who arrives in Paris, only to find out that his cooking idol is dead. When he makes an unusual alliance with a restaurant's new garbage boy, the culinary and personal adventures begin despite Remy's family's skepticism and the rat-hating world of humans.}, @movie.plot
+  #     end
+  #         
+  #     should "return an empty array if writers is nil" do
+  #       @movie.writers = nil
+  #       assert_equal [], @movie.writers
+  #     end
+  #         
+  #     should "return an empty array if directors is nil" do
+  #       @movie.directors = nil
+  #       assert_equal [], @movie.directors
+  #     end
+  #         
+  #     should "return an empty array if genres is nil" do
+  #       @movie.genres = nil
+  #       assert_equal [], @movie.genres
+  #     end
+  #   end
+  #   
+  # end
   
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: imdb_og
 version: !ruby/object:Gem::Version 
-  version: 0.5.6
+  version: 0.6.0
 platform: ruby
 authors: 
 - Jon Maddox
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-03-15 00:00:00 -04:00
+date: 2010-07-06 00:00:00 -04:00
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency