myimdb 0.3.2 → 0.3.3

data/VERSION

@@ -1 +1 @@
-0.3.2
+0.3.3

data/bin/myimdb

@@ -1,20 +1,62 @@
 #!/usr/bin/env ruby
 
-require 'myimdb.rb'
+require 'optparse'
+require 'myimdb'
 
-name = ARGV.join(' ')
+options = {}
+
+OptionParser.new do |opts|
+  opts.banner = "Usage: #{File.basename($0)} [movie name]"
+
+  opts.on("-h", "--help", "Displays this help info") do
+    puts opts
+    exit 0
+  end
+  
+  opts.on("-r", "--rottentomatoes", "Generates data from Rotten Tomatoes") do
+    options[:rottentomatoes] = true
+  end
+  
+  opts.on("-m", "--metacritic", "Generates data from Metacritic") do
+    options[:metacritic] = true
+  end
+  
+  opts.on("-i", "--imdb", "Generates data from Imdb") do
+    options[:imdb] = true
+  end
+
+  begin
+    opts.parse!(ARGV)
+  rescue OptionParser::ParseError => e
+    warn e.message
+    puts opts
+    exit 1
+  end
+end
+
+# add imdb as default
+options.merge!(:imdb=> true) if options.empty?
 
-if name.blank?
-  p "Movie name required - exiting"
-  exit(0)
+# need a movie name
+if ARGV.empty?
+  abort "Movie name required - exiting"
 end
 
-search_scope  = "imdb.com"
+name = ARGV.join(' ')
 
-begin
-  search_result = Myimdb::Search::Google.search_text(name, search_scope)[0]
-  imdb          = Myimdb::Scraper::Imdb.new(search_result["url"])
-  print "#{imdb.summary}\n"
+def details(klass_name, name)
+  search_scope  = "#{klass_name.downcase}.com"
+  
+  search_result = Myimdb::Search::Google.search_text(name, :restrict_to=> search_scope)[0]
+  site          = "Myimdb::Scraper::#{klass_name}".constantize.new(search_result["url"])
+  print "====================================================\n"
+  print "#{klass_name} details for: #{name}\n"
+  print "====================================================\n"
+  print "#{site.summary}\n"
 rescue
-  p "Unable to fetch details for: #{name}"
+  p "Unable to fetch #{klass_name} details for: #{name}"
 end
+
+['Imdb', 'RottenTomatoes', 'Metacritic'].each do |site|
+  details(site, name) if options[site.downcase.to_sym]
+end

data/bin/myimdb-catalogue

@@ -1,34 +1,34 @@
-#!/usr/bin/env ruby
-
-# point this file to a folder and all the folder inside it will be renamed
-require 'myimdb.rb'
-
-unless ARGV[0]
-  p "Source directory required - exiting"
-  exit(0)
-end
-
-file_paths = Dir["#{ARGV[0]}/*"]
-search_scope  = "imdb.com"
-
-file_paths.each do |path|
-  next if !File.directory?(path)
-  p "============================================"
-  name = File.basename(path)
-  if name.scan(/\[.*?\]/).size == 3
-    p "Skipping: #{name}"
-  else
-    p "Fetching details for: #{name}"
-    begin
-      search_result = Myimdb::Search::Google.search_text(name, search_scope)[0]
-      imdb          = Myimdb::Scraper::Imdb.new(search_result["url"])
-      new_name = name.gsub(/\[\S+\]/, "").strip
-      new_name << " [#{imdb.year}] [#{imdb.rating},#{imdb.votes}] [#{imdb.directors.join(',')}]"
-      p "Renaming: #{name} to: #{new_name}"
-      new_path = File.join(File.dirname(path), new_name)
-      File.rename(path, new_path)
-    rescue
-      p "Unable to fetch details for: #{name}"
-    end
-  end
+#!/usr/bin/env ruby
+
+# point this file to a folder and all the folder inside it will be renamed
+require 'myimdb.rb'
+
+unless ARGV[0]
+  p "Source directory required - exiting"
+  exit(0)
+end
+
+file_paths = Dir["#{ARGV[0]}/*"]
+search_scope  = "imdb.com"
+
+file_paths.each do |path|
+  next if !File.directory?(path)
+  p "============================================"
+  name = File.basename(path)
+  if name.scan(/\[.*?\]/).size == 3
+    p "Skipping: #{name}"
+  else
+    p "Fetching details for: #{name}"
+    begin
+      search_result = Myimdb::Search::Google.search_text(name, :restrict_to=> search_scope)[0]
+      imdb          = Myimdb::Scraper::Imdb.new(search_result["url"])
+      new_name = name.gsub(/\[\S+\]/, "").strip
+      new_name << " [#{imdb.year}] [#{imdb.rating},#{imdb.votes}] [#{imdb.directors.join(',')}]"
+      p "Renaming: #{name} to: #{new_name}"
+      new_path = File.join(File.dirname(path), new_name)
+      File.rename(path, new_path)
+    rescue
+      p "Unable to fetch details for: #{name}"
+    end
+  end
 end

data/lib/myimdb/scraper/string_extensions.rb

@@ -23,6 +23,17 @@ module Myimdb
       def humanize
         gsub(/_/, " ").capitalize
       end
+      
+      def constantize
+        names = self.split('::')
+        names.shift if names.empty? || names.first.empty?
+      
+        constant = Object
+        names.each do |name|
+          constant = constant.const_defined?(name) ? constant.const_get(name) : constant.const_missing(name)
+        end
+        constant
+      end
     end
   end
 end

data/myimdb.gemspec

@@ -5,7 +5,7 @@
 
 Gem::Specification.new do |s|
   s.name = %q{myimdb}
-  s.version = "0.3.2"
+  s.version = "0.3.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Gaurav"]

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: myimdb
 version: !ruby/object:Gem::Version 
-  version: 0.3.2
+  version: 0.3.3
 platform: ruby
 authors: 
 - Gaurav