myimdb 0.3.3 → 0.3.4

data/VERSION

@@ -1 +1 @@
-0.3.3
+0.3.4

data/bin/myimdb-catalogue

@@ -1,34 +1,151 @@
 #!/usr/bin/env ruby
 
-# point this file to a folder and all the folder inside it will be renamed
-require 'myimdb.rb'
+require 'open-uri'
+require 'optparse'
+require 'myimdb'
 
-unless ARGV[0]
-  p "Source directory required - exiting"
-  exit(0)
+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("-f", "--force", "Force generate data even if already present") do
+    options[:force] = true
+  end
+  
+  opts.on("-m", "--metadata", "Generates metadata (renames folders)") do
+    options[:metadata] = true
+  end
+  
+  opts.on("-i", "--images", "Generates images (only applies to windows)") do
+    options[:images] = true
+  end
+
+  begin
+    opts.parse!(ARGV)
+  rescue OptionParser::ParseError => e
+    warn e.message
+    puts opts
+    exit 1
+  end
 end
 
-file_paths = Dir["#{ARGV[0]}/*"]
-search_scope  = "imdb.com"
+# add imdb as default
+options.merge!(:metadata=> true) if options.empty?
+
+# need a movie name
+if ARGV.empty?
+  abort "Source directory required - exiting"
+end
+
+class Platform
+  class << self
+    def windows?
+      RUBY_PLATFORM =~ /mswin|windows/i
+    end
+    
+    def mac?
+      RUBY_PLATFORM =~ /darwin/i
+    end
+    
+    def linux?
+      RUBY_PLATFORM =~ /linux/i
+    end
+  end
+end
+
+file_paths    = Dir["#{ARGV[0]}/*"]
+
+def save_image(target_dir, name, image_index=0)
+  image_data = Myimdb::Search::Google.search_images(name, :size=> 'medium')[image_index]
+  image_url  = image_data["url"] if image_data
+  if image_url
+    image_file_path = File.join(target_dir, 'movie.jpg')
+    conf_file_path = File.join(target_dir, 'desktop.ini')
+    open(image_url) do |file|
+      File.open(image_file_path, 'wb') do |image_file|
+        image_file.puts file.read
+      end
+    end
+
+    `convert "#{image_file_path}" -thumbnail 256x256 -gravity center -crop 256x256+0+0! -background transparent -flatten "#{File.join(target_dir, 'movie.ico')}"`
+
+    if Platform.windows?
+      File.delete(conf_file_path) if File.exists?(conf_file_path)
+      File.open(conf_file_path, 'wb') do |conf|
+        conf.puts "[.ShellClassInfo]"
+        conf.puts "IconResource=movie.ico,0"
+      end
+      `attrib -s +h "#{conf_file_path}"`
+      `attrib -r "#{target_dir}"`
+      `attrib +r "#{target_dir}"`
+    end
+
+    File.delete(image_file_path)
+  end
+end
+
+def generate_metadata(path, name)
+  search_scope  = "imdb.com"
+
+  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)
+  new_path
+end
+
+def filtered_movie_name(name)
+  name.gsub(/\[.*?\]/,'').strip
+end
+
+def filtered_movie_year(name)
+  name.scan(/\[(\d+)\]/).flatten.first
+end
+
+
+image_retry_limit = 2
 
 file_paths.each do |path|
+  retry_count       = 0
+  metadata_present  = false
+
   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}"
+  begin
+    name = File.basename(path)
+
+    metadata_present = (name.scan(/\[.*?\]/).size == 3)
+    # metadata exists and -f isn't supplied
+    if metadata_present and !options[:force]
+      p "Skipping: Metadata already exists for: #{name}"
+    elsif options[:metadata]
+      p "Fetching metadata for: #{name}"
+      path = generate_metadata(path, filtered_movie_name(name))
+    end
+    
+    # movie icon exists and -f isn't supplied
+    if !metadata_present
+      p "Skipping: Metadata not present: #{name}"
+    elsif File.exists?(File.join(path, 'movie.ico')) and !options[:force]
+      p "Skipping: Image already exists for: #{name}"
+    elsif options[:images]
+      p "Fetching image for: #{name}"
+      begin
+        save_image(path, "#{filtered_movie_name(name)} #{filtered_movie_year(name)}", retry_count)
+      rescue
+        retry_count += 1
+        retry_count < image_retry_limit ? retry : raise
+      end
     end
+  rescue
+    p "Unable to fetch details for: #{name}"
   end
 end

data/lib/myimdb/scraper/base.rb

@@ -31,9 +31,15 @@ module Myimdb
       include HandleExceptions
       def directors
       end
+      
+      def directors_with_url
+      end
     
       def writers
       end
+      
+      def writers_with_url
+      end
     
       def rating
       end

data/lib/myimdb/scraper/imdb.rb

@@ -8,10 +8,18 @@ module Myimdb
       def directors
         document.css('.info h5:contains("Director") + .info-content a:not(.tn15more)').collect{ |a| a.text }
       end
+      
+      def directors_with_url
+        document.css('.info h5:contains("Director") + .info-content a:not(.tn15more)').collect{ |a| {:name=> a.text, :url=> "http://www.imdb.com#{a['href']}" } }
+      end
     
       def writers
         document.css('.info h5:contains("Writer") + .info-content a:not(.tn15more)').collect{ |a| a.text }
       end
+      
+      def writers_with_url
+        document.css('.info h5:contains("Writer") + .info-content a:not(.tn15more)').collect{ |a| {:name=> a.text, :url=> "http://www.imdb.com#{a['href']}" } }
+      end
     
       def rating
         document.css(".starbar-meta b").inner_text.strip.split('/').first.to_f
@@ -46,7 +54,7 @@ module Myimdb
           @document ||= Nokogiri::HTML(open(@url))
         end
         
-        handle_exceptions_for :directors, :writers, :rating, :votes, :genres, :tagline, :plot, :year
+        handle_exceptions_for :directors, :directors_with_url, :writers, :writers_with_url, :rating, :votes, :genres, :tagline, :plot, :year
     end
   end
 end

data/myimdb.gemspec

@@ -5,13 +5,13 @@
 
 Gem::Specification.new do |s|
   s.name = %q{myimdb}
-  s.version = "0.3.3"
+  s.version = "0.3.4"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 0") if s.respond_to? :required_rubygems_version=
   s.authors = ["Gaurav"]
-  s.date = %q{2010-01-19}
+  s.date = %q{2010-01-22}
   s.email = %q{gaurav@vinsol.com}
-  s.executables = ["myimdb", "myimdb-catalogue", "myimdb-catalogue.bat", "myimdb.bat"]
+  s.executables = ["myimdb", "myimdb-catalogue"]
   s.extra_rdoc_files = [
     "LICENSE",
      "README.rdoc"
@@ -23,8 +23,6 @@ Gem::Specification.new do |s|
      "VERSION",
      "bin/myimdb",
      "bin/myimdb-catalogue",
-     "bin/myimdb-catalogue.bat",
-     "bin/myimdb.bat",
      "lib/myimdb.rb",
      "lib/myimdb/scraper.rb",
      "lib/myimdb/scraper/base.rb",

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: myimdb
 version: !ruby/object:Gem::Version 
-  version: 0.3.3
+  version: 0.3.4
 platform: ruby
 authors: 
 - Gaurav
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2010-01-19 00:00:00 +05:30
+date: 2010-01-22 00:00:00 +05:30
 default_executable: 
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -37,8 +37,6 @@ email: gaurav@vinsol.com
 executables: 
 - myimdb
 - myimdb-catalogue
-- myimdb-catalogue.bat
-- myimdb.bat
 extensions: []
 
 extra_rdoc_files: 
@@ -51,8 +49,6 @@ files:
 - VERSION
 - bin/myimdb
 - bin/myimdb-catalogue
-- bin/myimdb-catalogue.bat
-- bin/myimdb.bat
 - lib/myimdb.rb
 - lib/myimdb/scraper.rb
 - lib/myimdb/scraper/base.rb

data/bin/myimdb-catalogue.bat

@@ -1 +0,0 @@
-@"ruby.exe" "%~dpn0" %*