imdb_og 0.5.0

data/.gitignore

@@ -0,0 +1,5 @@
+*.sw?
+.DS_Store
+coverage
+rdoc
+pkg

data/LICENSE

@@ -0,0 +1,20 @@
+Copyright (c) 2009 Jon Maddox
+
+Permission is hereby granted, free of charge, to any person obtaining
+a copy of this software and associated documentation files (the
+"Software"), to deal in the Software without restriction, including
+without limitation the rights to use, copy, modify, merge, publish,
+distribute, sublicense, and/or sell copies of the Software, and to
+permit persons to whom the Software is furnished to do so, subject to
+the following conditions:
+
+The above copyright notice and this permission notice shall be
+included in all copies or substantial portions of the Software.
+
+THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND,
+EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF
+MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND
+NONINFRINGEMENT. IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE
+LIABLE FOR ANY CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION
+OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION
+WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE.

data/README.textile

@@ -0,0 +1,24 @@
+h1. IMDB!
+
+h2. What?
+
+A simple ruby library to scrape IMDB
+
+h2. How?
+
+<pre>
+	<code>
+movie = Imdb.find_movie_by_id('tt1099212')
+movie.title
+</code>
+=> "Twilight"
+
+</pre>
+
+It does more of course, just look at the source.
+
+h2. Installation
+
+gem install imdb
+
+gem hosted on Gemcutter

data/Rakefile

@@ -0,0 +1,57 @@
+require 'rubygems'
+require 'rake'
+
+begin
+  require 'jeweler'
+  Jeweler::Tasks.new do |gem|
+    gem.name = "imdb_og"
+    gem.summary = %Q{Simple library to look up movies on IMDB}
+    gem.description = %Q{Simple library to look up movies on IMDB}
+    gem.email = "jon@mustacheinc.com"
+    gem.homepage = "http://github.com/maddox/imdb"
+    gem.authors = ["Jon Maddox"]
+    gem.add_development_dependency "thoughtbot-shoulda"
+    gem.add_dependency "htmlentities"
+    gem.add_dependency "hpricot"
+  end
+rescue LoadError
+  puts "Jeweler (or a dependency) not available. Install it with: sudo gem install jeweler"
+end
+
+require 'rake/testtask'
+Rake::TestTask.new(:test) do |test|
+  test.libs << 'lib' << 'test'
+  test.pattern = 'test/**/*_test.rb'
+  test.verbose = true
+end
+
+begin
+  require 'rcov/rcovtask'
+  Rcov::RcovTask.new do |test|
+    test.libs << 'test'
+    test.pattern = 'test/**/*_test.rb'
+    test.verbose = true
+  end
+rescue LoadError
+  task :rcov do
+    abort "RCov is not available. In order to run rcov, you must: sudo gem install spicycode-rcov"
+  end
+end
+
+task :test => :check_dependencies
+
+task :default => :test
+
+require 'rake/rdoctask'
+Rake::RDocTask.new do |rdoc|
+  if File.exist?('VERSION')
+    version = File.read('VERSION')
+  else
+    version = ""
+  end
+
+  rdoc.rdoc_dir = 'rdoc'
+  rdoc.title = "imdb_gem #{version}"
+  rdoc.rdoc_files.include('README*')
+  rdoc.rdoc_files.include('lib/**/*.rb')
+end

data/VERSION

@@ -0,0 +1 @@
+0.5.0

data/imdb.gemspec

@@ -0,0 +1,62 @@
+# Generated by jeweler
+# DO NOT EDIT THIS FILE
+# Instead, edit Jeweler::Tasks in Rakefile, and run `rake gemspec`
+# -*- encoding: utf-8 -*-
+
+Gem::Specification.new do |s|
+  s.name = %q{imdb}
+  s.version = "0.5.0"
+
+  s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
+  s.authors = ["Jon Maddox"]
+  s.date = %q{2009-10-10}
+  s.description = %q{Simple library to look up movies on IMDB}
+  s.email = %q{jon@mustacheinc.com}
+  s.extra_rdoc_files = [
+    "LICENSE",
+     "README.textile"
+  ]
+  s.files = [
+    ".gitignore",
+     "LICENSE",
+     "README.textile",
+     "Rakefile",
+     "VERSION",
+     "lib/imdb.rb",
+     "lib/imdb/imdb.rb",
+     "lib/imdb/imdb_company.rb",
+     "lib/imdb/imdb_genre.rb",
+     "lib/imdb/imdb_movie.rb",
+     "lib/imdb/imdb_name.rb",
+     "test/imdb_test.rb",
+     "test/test_helper.rb"
+  ]
+  s.homepage = %q{http://github.com/maddox/imdb}
+  s.rdoc_options = ["--charset=UTF-8"]
+  s.require_paths = ["lib"]
+  s.rubygems_version = %q{1.3.5}
+  s.summary = %q{Simple library to look up movies on IMDB}
+  s.test_files = [
+    "test/imdb_test.rb",
+     "test/test_helper.rb"
+  ]
+
+  if s.respond_to? :specification_version then
+    current_version = Gem::Specification::CURRENT_SPECIFICATION_VERSION
+    s.specification_version = 3
+
+    if Gem::Version.new(Gem::RubyGemsVersion) >= Gem::Version.new('1.2.0') then
+      s.add_development_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_runtime_dependency(%q<htmlentities>, [">= 0"])
+      s.add_runtime_dependency(%q<hpricot>, [">= 0"])
+    else
+      s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+      s.add_dependency(%q<htmlentities>, [">= 0"])
+      s.add_dependency(%q<hpricot>, [">= 0"])
+    end
+  else
+    s.add_dependency(%q<thoughtbot-shoulda>, [">= 0"])
+    s.add_dependency(%q<htmlentities>, [">= 0"])
+    s.add_dependency(%q<hpricot>, [">= 0"])
+  end
+end

data/lib/imdb/imdb.rb

@@ -0,0 +1,127 @@
+class Imdb
+
+  IMDB_MOVIE_BASE_URL = "http://www.imdb.com/title/"
+  IMDB_NAME_BASE_URL = "http://www.imdb.com/name/"
+  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="
+
+  def self.find_movie_by_id(id)
+    coder = HTMLEntities.new
+
+    data = Hpricot(open(IMDB_MOVIE_BASE_URL + id))
+    
+    movie = ImdbMovie.new
+    
+    movie.imdb_id = id
+    movie.title = coder.decode(data.at("meta[@name='title']")['content'].gsub(/\((\d{4}(\/[^)]*)?|[A-Z]+)\)/,'').strip)
+
+    rating_text = (data/"div.rating/div.meta/b").inner_text
+    if rating_text =~ /([\d\.]+)\/10/
+      movie.rating = $1
+    end
+
+    begin
+      movie.poster_url = data.at("div.photo/a[@name='poster']/img")['src']
+    rescue
+      movie.poster_url = nil
+    end
+
+    infos = (data/"div.info")
+    infos.each do |info|
+      info_title = (info/"h5").inner_text
+      case info_title
+      when /Directors?:/
+        movie.directors = parse_names(info)
+      when /Writers?[^:]+:/
+        movie.writers = parse_names(info)
+      when /Company:/
+        movie.company = parse_company(info)
+      when "Tagline:"
+        movie.tagline = coder.decode(parse_info(info).strip)
+      when "Runtime:"
+        movie.runtime = parse_info(info).strip
+        if (movie.runtime)
+          movie.runtime.gsub!(/^[^:]+:\s*/, '')
+          movie.runtime.gsub!(/min .*/, 'min')
+        end
+      when "Plot:"
+        movie.plot = parse_info(info).strip
+        movie.plot = movie.plot.gsub(/\s*\|\s*add synopsis$/, '')
+        movie.plot = movie.plot.gsub(/\s*\|\s*full synopsis$/, '')
+        movie.plot = movie.plot.gsub(/\s*\|\s*add summary$/, '')
+        movie.plot = movie.plot.gsub(/full summary$/, '')
+        movie.plot = movie.plot.gsub(/more$/, '')
+        movie.plot = coder.decode(movie.plot.strip)
+      when "Genre:"
+        movie.genres = parse_genres(info)
+      when "Release Date:"
+        begin
+          if (parse_info(info).strip =~ /(\d{1,2}) ([a-zA-Z]+) (\d{4})/)
+            movie.release_date = Date.parse("#{$2} #{$1}, #{$3}")
+          end
+        rescue
+          movie.release_date = nil
+        end
+      when "Certification:"
+        begin
+          movie.certification = (info/"a").map { |v| v.inner_html }.select { |v| v =~ /^USA:/ && v !~ /Unrated/ }.map { |v| v[/^USA:/]=''; v.strip }.first
+        end
+      end
+    end 
+
+    cast = (data/"table.cast"/"tr")
+    cast.each do |cast_member|
+        actor_a = (cast_member/"td.nm").inner_html
+        actor_a =~ /name\/([^"]+)\//
+        actor_id = $1
+        actor_name = coder.decode((cast_member/"td.nm"/"a").inner_text)
+        actor_role = coder.decode((cast_member/"td.char").inner_text)
+        movie.actors = movie.actors << ImdbName.new(actor_id, actor_name, actor_role)
+    end
+
+    movie # return movie
+
+  end
+
+
+  protected
+  
+  def self.parse_info(info)
+    value = info.inner_text.gsub(/\n/,'') 
+    if value =~ /\:(.+)/ 
+      value = $1
+    end
+    value.gsub(/ more$/, '')
+  end
+  
+  def self.parse_names(info)
+    coder = HTMLEntities.new
+
+    # <a href="/name/nm0083348/">Brad Bird</a><br/><a href="/name/nm0684342/">Jan Pinkava</a> (co-director)<br/>N
+    info.inner_html.scan(/<a href="\/name\/([^"]+)\/"[^>]*>([^<]+)<\/a>( \(([^)]+)\))?/).map do |match|
+      ImdbName.new(coder.decode(match[0]), coder.decode(match[1]), coder.decode(match[3]))
+    end
+  end
+  
+  def self.parse_company(info)
+    coder = HTMLEntities.new
+    # <a href="/company/co0017902/">Pixar Animation Studios</a>
+    match = info.inner_html =~ /<a href="\/company\/([^"]+)\/">([^<]+)<\/a>/
+    ImdbCompany.new(coder.decode($1), coder.decode($2))
+  end
+
+  def self.parse_genres(info)
+    coder = HTMLEntities.new
+    # <a href="/Sections/Genres/Animation/">Animation</a> / <a href="/Sections/Genres/Adventure/">Adventure</a>
+    genre_links = (info/"a").find_all do |link|
+      link['href'] =~ /^\/Sections\/Genres/
+    end 
+    genre_links.map do |link|
+      genre = link['href'] =~ /([^\/]+)\/$/
+      ImdbGenre.new(coder.decode($1), coder.decode($1))
+    end
+  end
+
+  
+end

data/lib/imdb/imdb_company.rb

@@ -0,0 +1,8 @@
+class ImdbCompany
+  attr_accessor :imdb_id, :name
+  
+  def initialize(imdb_id, name)
+    self.imdb_id = imdb_id;
+    self.name = name;
+  end
+end

data/lib/imdb/imdb_genre.rb

@@ -0,0 +1,8 @@
+class ImdbGenre
+  attr_accessor :imdb_id, :name
+  
+  def initialize(imdb_id, name)
+    self.imdb_id = imdb_id;
+    self.name = name;
+  end
+end

data/lib/imdb/imdb_movie.rb

@@ -0,0 +1,20 @@
+class ImdbMovie
+  attr_accessor :imdb_id, :title, :directors, :writers, :tagline, :company, :plot, :runtime, :rating, :poster_url, :release_date, :certification, :genres, :actors
+
+  def writers
+     self.instance_variable_get(:@writers) || []
+  end
+
+  def actors
+     self.instance_variable_get(:@actors) || []
+  end
+
+  def directors
+    self.instance_variable_get(:@directors) || []
+  end
+  
+  def genres
+    self.instance_variable_get(:@genres) || []
+  end
+
+end

data/lib/imdb/imdb_name.rb

@@ -0,0 +1,9 @@
+class ImdbName
+  attr_accessor :imdb_id, :name, :role
+  
+  def initialize(imdb_id, name, role)
+    self.imdb_id = imdb_id;
+    self.name = name;
+    self.role = role;
+  end
+end

data/lib/imdb.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'hpricot'
+require 'open-uri'
+require 'date'
+require 'htmlentities'
+require 'imdb/imdb'
+require 'imdb/imdb_company'
+require 'imdb/imdb_movie'
+require 'imdb/imdb_name'
+require 'imdb/imdb_genre'

data/test/imdb_test.rb

@@ -0,0 +1,126 @@
+require 'test_helper'
+
+class ImdbTest < Test::Unit::TestCase
+
+  context "Imdb" do
+    should "have an imdb movie base url" do
+      assert_equal "http://www.imdb.com/title/", Imdb::IMDB_MOVIE_BASE_URL
+    end
+    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 "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 five genres" do
+        assert_equal 3, @movie.genres.length
+        assert_equal 'Animation', @movie.genres[0].name
+        assert_equal 'Comedy', @movie.genres[1].name
+        assert_equal 'Family', @movie.genres[2].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

data/test/test_helper.rb

@@ -0,0 +1,10 @@
+require 'rubygems'
+require 'test/unit'
+require 'shoulda'
+
+$LOAD_PATH.unshift(File.join(File.dirname(__FILE__), '..', 'lib'))
+$LOAD_PATH.unshift(File.dirname(__FILE__))
+require 'imdb'
+
+class Test::Unit::TestCase
+end

metadata

@@ -0,0 +1,99 @@
+--- !ruby/object:Gem::Specification 
+name: imdb_og
+version: !ruby/object:Gem::Version 
+  version: 0.5.0
+platform: ruby
+authors: 
+- Jon Maddox
+autorequire: 
+bindir: bin
+cert_chain: []
+
+date: 2009-10-10 00:00:00 -04:00
+default_executable: 
+dependencies: 
+- !ruby/object:Gem::Dependency 
+  name: thoughtbot-shoulda
+  type: :development
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: htmlentities
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+- !ruby/object:Gem::Dependency 
+  name: hpricot
+  type: :runtime
+  version_requirement: 
+  version_requirements: !ruby/object:Gem::Requirement 
+    requirements: 
+    - - ">="
+      - !ruby/object:Gem::Version 
+        version: "0"
+    version: 
+description: Simple library to look up movies on IMDB
+email: jon@mustacheinc.com
+executables: []
+
+extensions: []
+
+extra_rdoc_files: 
+- LICENSE
+- README.textile
+files: 
+- .gitignore
+- LICENSE
+- README.textile
+- Rakefile
+- VERSION
+- imdb.gemspec
+- lib/imdb.rb
+- lib/imdb/imdb.rb
+- lib/imdb/imdb_company.rb
+- lib/imdb/imdb_genre.rb
+- lib/imdb/imdb_movie.rb
+- lib/imdb/imdb_name.rb
+- test/imdb_test.rb
+- test/test_helper.rb
+has_rdoc: true
+homepage: http://github.com/maddox/imdb
+licenses: []
+
+post_install_message: 
+rdoc_options: 
+- --charset=UTF-8
+require_paths: 
+- lib
+required_ruby_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+required_rubygems_version: !ruby/object:Gem::Requirement 
+  requirements: 
+  - - ">="
+    - !ruby/object:Gem::Version 
+      version: "0"
+  version: 
+requirements: []
+
+rubyforge_project: 
+rubygems_version: 1.3.5
+signing_key: 
+specification_version: 3
+summary: Simple library to look up movies on IMDB
+test_files: 
+- test/imdb_test.rb
+- test/test_helper.rb