pirate-autonzb 0.3 → 0.3.2

data/Manifest

@@ -1,4 +1,5 @@
 asset/failure.png
+autonzb.gemspec
 bin/autonzb
 lib/imdb.rb
 lib/inspector.rb

data/Rakefile

@@ -3,7 +3,7 @@ require 'rubygems'
 require 'rake'
 require 'echoe'
 
-Echoe.new('autonzb', '0.3') do |p|
+Echoe.new('autonzb', '0.3.2') do |p|
   p.description    = "Ruby tool to automatically download x264 HD nzb movies files from newzleech.com"
   p.url            = "http://github.com/pirate/autonzb"
   p.author         = "Pirate"

data/autonzb.gemspec

@@ -2,17 +2,17 @@
 
 Gem::Specification.new do |s|
   s.name = %q{autonzb}
-  s.version = "0.3"
+  s.version = "0.3.2"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
   s.authors = ["Pirate"]
-  s.date = %q{2008-12-25}
+  s.date = %q{2008-12-26}
   s.default_executable = %q{autonzb}
   s.description = %q{Ruby tool to automatically download x264 HD nzb movies files from newzleech.com}
   s.email = %q{pirate.2061@gmail.com}
   s.executables = ["autonzb"]
   s.extra_rdoc_files = ["bin/autonzb", "lib/imdb.rb", "lib/inspector.rb", "lib/movie.rb", "lib/nfo.rb", "lib/nzb.rb", "README.markdown"]
-  s.files = ["asset/failure.png", "bin/autonzb", "lib/imdb.rb", "lib/inspector.rb", "lib/movie.rb", "lib/nfo.rb", "lib/nzb.rb", "Manifest", "Rakefile", "README.markdown", "autonzb.gemspec"]
+  s.files = ["asset/failure.png", "autonzb.gemspec", "bin/autonzb", "lib/imdb.rb", "lib/inspector.rb", "lib/movie.rb", "lib/nfo.rb", "lib/nzb.rb", "Manifest", "Rakefile", "README.markdown"]
   s.has_rdoc = true
   s.homepage = %q{http://github.com/pirate/autonzb}
   s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Autonzb", "--main", "README.markdown"]

data/bin/autonzb

@@ -46,5 +46,6 @@ begin
 
   nzbmatrix = NZB.new(inspector, ARGV.flags.d, :age => ARGV.flags.age, :pages => ARGV.flags.pages)
 rescue => e
-  Inspector.growl("AutoNZB Error!", e)
+  p e.to_s
+  Inspector.growl("AutoNZB Error!", 'look into the console log')
 end

data/lib/inspector.rb

@@ -2,7 +2,7 @@ require File.join(File.dirname(__FILE__), 'movie')
 
 class Inspector
   
-  attr_accessor :backup
+  attr_accessor :backup_path, :movies, :nzbs
   
   def initialize(paths, options = {})
     @paths = paths.split(',').map { |p| p.gsub(/\/$/,'') }
@@ -11,59 +11,61 @@ class Inspector
     @options[:imdb_score] = @options[:imdb_score] ? @options[:imdb_score].to_f : 7.0
     @options[:year] = @options[:year] ? @options[:year].to_i : 1950
     
-    if @options[:backup]
-      @backup = @options[:backup].gsub(/\/$/,'')
-      @paths << @backup
-    end
-    
     @movies = []
     initialize_movies
     
+    if @options[:backup]
+      @backup_path = @options[:backup].gsub(/\/$/,'') 
+      @nzbs = []
+      initialize_nzbs
+      @movies = @nzbs + @movies
+    end
+    
     $stdout.print "Movie criteria: imdb score >= #{@options[:imdb_score]}, year >= #{@options[:year]} and srt [#{@options[:srt].join(',')}]\n"
   end
   
-  def need?(movie)
-    if valid?(movie)
-      $stdout.print " => movie has required criteria "
-      if m = @movies.detect { |m| m == movie }
-        $stdout.print "but is already owned "
+  def need?(movie, not_validate = false, movies = @movies, log = true)
+    if not_validate || valid?(movie)
+      $stdout.print " => movie has required criteria " if log
+      if m = movies.detect { |m| m == movie }
+        $stdout.print "but is already owned " if log
         if srt_score(movie) > srt_score(m)
-          $stdout.print "but new movie has better subtitle: [#{movie.srt.join(',')}]\n"
+          $stdout.print "but new movie has better subtitle: [#{movie.srt.join(',')}]\n" if log
           true
         elsif srt_score(movie) == srt_score(m)
           if format_score(movie) > format_score(m)
-            $stdout.print "but new movie has better format: #{movie.format}\n"
+            $stdout.print "but new movie has better format: #{movie.format}\n" if log
             true
           elsif format_score(movie) == format_score(m)
             if source_score(movie) > source_score(m)
-              $stdout.print "but new movie has better source: #{movie.source}\n"
+              $stdout.print "but new movie has better source: #{movie.source}\n" if log
               true
             elsif source_score(movie) == source_score(m)
               if sound_score(movie) > sound_score(m)
-                $stdout.print "but new movie has better sound: #{movie.sound}\n"
+                $stdout.print "but new movie has better sound: #{movie.sound}\n" if log
                 true
               else
-                $stdout.print "with same srt, format, source and sound\n"
+                $stdout.print "with same srt, format, source and sound\n" if log
                 false
               end
             else
-              $stdout.print "with same srt, format and better source: #{m.source}\n"
+              $stdout.print "with same srt, format and better source: #{m.source}\n" if log
               false
             end
           else
-            $stdout.print "with same srt and better format: #{m.format}\n"
+            $stdout.print "with same srt and better format: #{m.format}\n" if log
             false
           end
         else
-          $stdout.print "with better subtitle: [#{m.srt.join(',')}]\n"
+          $stdout.print "with better subtitle: [#{m.srt.join(',')}]\n" if log
           false
         end
       else
-        $stdout.print "and is not already owned\n"
+        $stdout.print "and is not already owned\n" if log
         true
       end
     else
-      $stdout.print " => movie doesn't have required criteria\n"
+      $stdout.print " => movie doesn't have required criteria\n" if log
       false
     end
   end
@@ -80,14 +82,19 @@ private
       base_dir = clean_dir(Dir.new(path))
       base_dir.each do |movie|
         movie_path = "#{path}/#{movie}"
-        @movies << Movie.new(movie) if File.directory?(movie_path) || File.extname(movie_path) == '.nzb'
-      end
-      if path == @backup
-        $stdout.print "Found #{@movies.size - old_movies_size} backuped nzb(s) in #{path}\n"
-      else
-        $stdout.print "Found #{@movies.size - old_movies_size} movie(s) in #{path}\n"
+        @movies << Movie.new(movie, :path => movie_path) if File.directory?(movie_path)
       end
+      $stdout.print "Found #{@movies.size - old_movies_size} movie(s) in #{path}\n"
+    end
+  end
+  
+  def initialize_nzbs
+    base_dir = clean_dir(Dir.new(@backup_path))
+    base_dir.each do |nzb|
+      nzb_path = "#{@backup_path}/#{nzb}"
+      @nzbs << Movie.new(nzb, :path => nzb_path) if File.extname(nzb_path) == '.nzb'
     end
+    $stdout.print "Found #{@nzbs.size} backuped nzb(s) in #{@backup_path}\n"
   end
 
   def clean_dir(dir)

data/lib/nzb.rb

@@ -11,6 +11,7 @@ class NZB
   attr_accessor :movies, :agent
   
   def initialize(inspector, download_path, options = {})
+    @inspector, @download_path = inspector, download_path.gsub(/\/$/,'')
     @options = options
     @options[:age] ||= 160
     @options[:pages] ||= 2
@@ -23,24 +24,8 @@ class NZB
     end
     
     parse_newzleech
-    movies.each do |movie|
-      $stdout.print "#{movie.dirname}, imdb score: #{movie.score}, age: #{movie.age.to_i} day(s)\n"
-      if inspector.need?(movie)
-        $stdout.print  " => DOWNLOAD: #{movie.name} (#{movie.year})\n"
-        download_nzb(download_path, movie, inspector.backup)
-      end
-    end
-    $stdout.print "No nzb found, maybe change -age or -page setting\n" if @movies.empty?
-  end
-  
-  def download_nzb(download_path, movie, backup_path = nil)
-    path = download_path.gsub(/\/$/,'') # removed / at the end
-    Tempfile.open("movie.nzb") do |tempfile|
-      tempfile.write(open(movie.nzb_link).read) # download the nzb
-      tempfile.close
-      File.move(tempfile.path, "#{path}/#{movie.dirname}.nzb")
-      File.copy("#{path}/#{movie.dirname}.nzb", "#{backup_path}/#{movie.dirname}.nzb") if backup_path
-    end
+    parse_movies
+    keep_only_best_nzb if @inspector.backup_path
   end
   
 private
@@ -70,6 +55,27 @@ private
     $stdout.print "\n"
   end
   
+  def parse_movies
+    movies.each do |movie|
+      $stdout.print "#{movie.dirname}, imdb score: #{movie.score}, age: #{movie.age.to_i} day(s)\n"
+      if @inspector.need?(movie)
+        $stdout.print " => DOWNLOAD: #{movie.name} (#{movie.year})\n"
+        download_nzb(movie, @inspector.backup_path)
+        @inspector.movies << movie
+      end
+    end
+    $stdout.print "No nzb found, maybe change -age or -page setting\n" if @movies.empty?
+  end
+  
+  def download_nzb(movie, backup_path = nil)
+    Tempfile.open("movie.nzb") do |tempfile|
+      tempfile.write(open(movie.nzb_link).read) # download the nzb
+      tempfile.close
+      File.move(tempfile.path, "#{@download_path}/#{movie.dirname}.nzb")
+      File.copy("#{@download_path}/#{movie.dirname}.nzb", "#{backup_path}/#{movie.dirname}.nzb") if backup_path
+    end
+  end
+  
   def parse_age(string)
     case string
     when /h/i
@@ -78,5 +84,20 @@ private
       string.to_f
     end
   end
+  
+  def keep_only_best_nzb
+    size = 0
+    @inspector.nzbs.each do |nzb|
+      nzbs = @inspector.nzbs.select { |item| item.path != nzb.path }
+      unless @inspector.need?(nzb, true, nzbs, false)
+        File.delete(nzb.path)
+        size += 1
+      end
+    end
+    if size > 0
+      $stdout.print "#########################################################################\n"
+      $stdout.print "Deleted #{size} useless backuped nzb(s) (keep only the best nzb by movie)\n"
+    end
+  end
     
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: pirate-autonzb
 version: !ruby/object:Gem::Version 
-  version: "0.3"
+  version: 0.3.2
 platform: ruby
 authors: 
 - Pirate
@@ -9,7 +9,7 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2008-12-25 00:00:00 -08:00
+date: 2008-12-26 00:00:00 -08:00
 default_executable: autonzb
 dependencies: 
 - !ruby/object:Gem::Dependency 
@@ -73,6 +73,7 @@ extra_rdoc_files:
 - README.markdown
 files: 
 - asset/failure.png
+- autonzb.gemspec
 - bin/autonzb
 - lib/imdb.rb
 - lib/inspector.rb
@@ -82,7 +83,6 @@ files:
 - Manifest
 - Rakefile
 - README.markdown
-- autonzb.gemspec
 has_rdoc: true
 homepage: http://github.com/pirate/autonzb
 post_install_message: