pirate-autosub 0.2.4 → 0.3

data/Rakefile

@@ -3,7 +3,7 @@ require 'rubygems'
 require 'rake'
 require 'echoe'
 
-Echoe.new('autosub', '0.2.4') do |p|
+Echoe.new('autosub', '0.3') do |p|
   p.description    = "Ruby tool to automatically download subtitles (srt) inside your TV Shows folder"
   p.url            = "http://github.com/pirate/autosub"
   p.author         = "Pirate"

data/autosub.gemspec

@@ -2,17 +2,17 @@
 
 Gem::Specification.new do |s|
   s.name = %q{autosub}
-  s.version = "0.2.4"
+  s.version = "0.3"
 
   s.required_rubygems_version = Gem::Requirement.new(">= 1.2") if s.respond_to? :required_rubygems_version=
   s.authors = ["Pirate"]
-  s.date = %q{2009-01-17}
+  s.date = %q{2009-03-15}
   s.default_executable = %q{autosub}
   s.description = %q{Ruby tool to automatically download subtitles (srt) inside your TV Shows folder}
   s.email = %q{pirate.2061@gmail.com}
   s.executables = ["autosub"]
-  s.extra_rdoc_files = ["bin/autosub", "lib/episode.rb", "lib/inspector.rb", "lib/sites/seriessub.rb", "lib/sites/tvsubtitle.rb", "README.markdown"]
-  s.files = ["asset/failure.png", "asset/srt.png", "autosub.gemspec", "bin/autosub", "lib/episode.rb", "lib/inspector.rb", "lib/sites/seriessub.rb", "lib/sites/tvsubtitle.rb", "Rakefile", "README.markdown", "Manifest"]
+  s.extra_rdoc_files = ["bin/autosub", "lib/episode.rb", "lib/inspector.rb", "lib/sites/podnapisi.rb", "lib/sites/seriessub.rb", "lib/sites/tvsubtitle.rb", "README.markdown"]
+  s.files = ["asset/failure.png", "asset/srt.png", "autosub.gemspec", "bin/autosub", "lib/episode.rb", "lib/inspector.rb", "lib/sites/podnapisi.rb", "lib/sites/seriessub.rb", "lib/sites/tvsubtitle.rb", "Manifest", "Rakefile", "README.markdown"]
   s.has_rdoc = true
   s.homepage = %q{http://github.com/pirate/autosub}
   s.rdoc_options = ["--line-numbers", "--inline-source", "--title", "Autosub", "--main", "README.markdown"]
@@ -31,14 +31,12 @@ Gem::Specification.new do |s|
       s.add_runtime_dependency(%q<optiflag>, [">= 0"])
       s.add_runtime_dependency(%q<simple-rss>, [">= 0"])
       s.add_runtime_dependency(%q<mechanize>, [">= 0"])
-      s.add_development_dependency(%q<echoe>, [">= 0"])
     else
       s.add_dependency(%q<hpricot>, [">= 0"])
       s.add_dependency(%q<rubyzip>, [">= 0"])
       s.add_dependency(%q<optiflag>, [">= 0"])
       s.add_dependency(%q<simple-rss>, [">= 0"])
       s.add_dependency(%q<mechanize>, [">= 0"])
-      s.add_dependency(%q<echoe>, [">= 0"])
     end
   else
     s.add_dependency(%q<hpricot>, [">= 0"])
@@ -46,6 +44,5 @@ Gem::Specification.new do |s|
     s.add_dependency(%q<optiflag>, [">= 0"])
     s.add_dependency(%q<simple-rss>, [">= 0"])
     s.add_dependency(%q<mechanize>, [">= 0"])
-    s.add_dependency(%q<echoe>, [">= 0"])
   end
 end

data/bin/autosub

@@ -11,12 +11,15 @@ module AutoSub extend OptiFlagSet
     description "Language(s) you want for your subtitles, default: fr,en
                   (Order is important to define if a srt is needed)"
   end
+  optional_flag "log" do
+    description "path for log file"
+  end
 
   and_process!
 end
 
 begin
-  inspector = Inspector.new(ARGV.flags.p, ARGV.flags.lang)
+  inspector = Inspector.new(ARGV.flags.p, :lang => ARGV.flags.lang, :log_path => ARGV.flags.log)
 rescue => e
   p e.to_s
   Inspector.growl("AutoSub Error!", 'look into the console log', 'failure.png')

data/lib/episode.rb

@@ -46,7 +46,7 @@ class Episode
   def good_format?(filename)
     case @format
     when 'hd'
-      filename.include?('720p')
+      filename.include?('720p') || filename.include?('720')
     when 'sd'
       !filename.include?('720p')
     end

data/lib/inspector.rb

@@ -1,15 +1,23 @@
 require File.join(File.dirname(__FILE__), '..', 'lib', 'episode')
 require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'seriessub')
 require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'tvsubtitle')
+require File.join(File.dirname(__FILE__), '..', 'lib', 'sites', 'podnapisi')
+require 'logger'
 
 class Inspector
   
-  attr_accessor :tv_shows, :episodes, :langs
+  attr_accessor :tv_shows, :episodes, :langs, :log
   
-  def initialize(path , langs = nil)
-    @path = path.gsub(/\/$/,'')
-    @langs = langs ? langs.split(',') : ['fr','en']
-
+  def initialize(path , options = {})
+    @path = Pathname.new(path).realpath.to_s
+    @langs = options[:lang] ? options[:lang].split(',') : ['fr','en']
+    
+    # Initialize Logger
+    @log = Logger.new(options[:log_path] || "#{@path}/_autosub_.log" )
+    @log.datetime_format = "%Y-%m-%d %H:%M:%S "
+    log.info "Begin searching for '#{langs.join(',')}' subtitles..."
+    
+    $stdout.print "--- Searching for '#{langs.join(',')}' subtitles ---\n"
     @tv_shows = [] # {:name => ..., :path => ...}
     initialize_tv_shows
     $stdout.print "Found #{@tv_shows.size} TV Show(s) in #{@path}\n"
@@ -20,8 +28,21 @@ class Inspector
     $stdout.print "Found #{@episodes.select { |e| e.need_srt?(@langs) }.size} episode(s) who need(s) srt\n"
     
     # Find new srt
-    TVSubtitle.new(self)
-    SeriesSub.new(self)
+    begin
+      Podnapisi.new(self)
+    rescue => e
+      log.fatal "Problem with Podnapisi: #{e}"
+    end
+    begin
+      TVSubtitle.new(self)
+    rescue => e
+      log.fatal "Problem with TVSubtitle: #{e}"
+    end
+    begin  
+      SeriesSub.new(self)
+    rescue => e
+      log.fatal "Problem with SeriesSub: #{e}"
+    end
   end
   
   def growl_episode(episode, lang)
@@ -61,7 +82,7 @@ private
   end
   
   def clean_dir(dir)
-    dir.select { |e| !["..", ".", ".DS_Store", ".com.apple.timemachine.supported", "Icon\r"].include?(e) }
+    dir.select { |e| !["..", ".", ".DS_Store", ".com.apple.timemachine.supported", "Icon\r", "autosub.log", "_autosub_.log"].include?(e) }
   end
 
   def lang_name(lang)

data/lib/sites/podnapisi.rb

@@ -0,0 +1,101 @@
+require 'hpricot'
+require 'open-uri'
+require 'zip/zip'
+require 'cgi'
+
+# http://www.podnapisi.net/ | http://www.sub-titles.net/
+class Podnapisi
+  
+  def initialize(inspector)
+    @inspector = inspector
+    episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
+
+    $stdout.print "------------------------------------------\n"
+    $stdout.print "Start srt searching from www.podnapisi.net\n"
+    $stdout.print "------------------------------------------\n"
+    episodes.each do |e|
+      @inspector.langs.each do |l|
+        unless e.srt.include?(l) 
+          $stdout.print "#{e.episode_name_with_format} [#{l}]: "
+          if get_srt(e, l)
+            @inspector.log.info "#{e.episode_name_with_format} [#{l}] // Podnapisi // #{@srt_name}"
+            $stdout.print "FOUND\n"
+            e.srt << l
+            @inspector.growl_episode(e, l)
+            break
+          else
+            $stdout.print "NOT FOUND\n"
+          end
+        end
+      end
+    end
+  end
+    
+private
+  
+  def get_srt(episode, lang)
+    found = false
+    doc = Hpricot(open(url(episode, lang)))
+    doc.search("table.list tr").each do |el|
+      if !found && have_srt_format?(el) && !hearing_impaired?(el) && url = open(download_url(el))
+        Tempfile.open("tv.zip") do |tempfile|
+          tempfile.write(open(url).read) # download the zip
+          tempfile.close # tempfile need to be closed for unzip it
+          Zip::ZipFile.open(tempfile.path) do |zip_file|
+            zip_file.each do |f|
+              if f.size > 5000
+                @srt_name = f.name
+                zip_file.extract(f, episode.srt_filename(lang))
+                found = true
+              end
+            end
+          end
+        end
+      end
+    end
+  rescue => e
+    @inspector.log.fatal "Podnapisi // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
+  end
+
+  def url(episode, lang)
+    "http://simple.podnapisi.net/ppodnapisi/search?tbsl=3&asdp=1&sK=#{tv_show(episode)}&sJ=#{lang_num(lang)}&sO=desc&sS=time&submit=Search&sTS=#{episode.season}&sTE=#{episode.number}&sY=&sR=#{format(episode)}&sT=1"
+  end
+  
+  def tv_show(episode)
+    CGI.escape(episode.tv_show)
+  end
+  
+  def format(episode)
+    case episode.format
+    when 'hd'
+      "720"
+    when 'sd'
+      ""
+    end
+  end
+  
+  def lang_num(lang)
+    case lang
+    when 'fr'
+      8
+    when 'en'
+      2
+    end
+  end
+  
+  def have_srt_format?(el)
+    !el.search("td[text()*='SubRip']").empty?
+  end
+  
+  def hearing_impaired?(el)
+    !el.search("img[@title='Subtitle is for hearing impaired']").empty?
+  end
+  
+  def download_url(el)
+    "http://simple.podnapisi.net#{el.search("a[text()*='download subtitle']").first[:href]}"
+  end
+  
+end

data/lib/sites/seriessub.rb

@@ -17,6 +17,7 @@ class SeriesSub
         unless e.srt.include?(l) 
           $stdout.print "#{e.episode_name_with_format} [#{l}]: "
           if get_srt(e, l)
+            inspector.log.info "#{e.episode_name_with_format} [#{l}] // SeriesSub // #{@srt_name}"
             $stdout.print "FOUND\n"
             e.srt << l
             @inspector.growl_episode(e, l)
@@ -43,7 +44,8 @@ private
             tempfile.close # tempfile need to be closed for unzip it
             Zip::ZipFile.open(tempfile.path) do |zip_file|
               zip_file.each do |f|
-                if good_episode?(episode, f.name) && f.name.include?(lang_version(lang)) && episode.good_format?(f.name) && no_tag?(f.name)
+                if f.size > 5000 && good_episode?(episode, f.name) && f.name.include?(lang_version(lang)) && episode.good_format?(f.name) && no_tag?(f.name)
+                  @srt_name = f.name
                   zip_file.extract(f, episode.srt_filename(lang))
                   found = true
                 end
@@ -51,6 +53,7 @@ private
             end
           end
         elsif episode.good_format?(text) && no_tag?(text) # just a single srt file
+          @srt_name = text
           Tempfile.open("tv.srt") do |tempfile|
             tempfile.write(open(el[:href]).read) # download the zip
             tempfile.close
@@ -60,6 +63,10 @@ private
         end
       end
     end
+  rescue => e
+    @inspector.log.fatal "SeriesSub // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
     found
   end
 

data/lib/sites/tvsubtitle.rb

@@ -12,15 +12,16 @@ class TVSubtitle
     episodes = @inspector.episodes.select { |e| e.need_srt?(@inspector.langs) }
     
     $stdout.print "--------------------------------------------\n"
-    $stdout.print "Searching fot srt in www.tvsubtitles.net RSS\n"
+    $stdout.print "Searching for srt in www.tvsubtitles.net RSS\n"
     $stdout.print "--------------------------------------------\n"
     @inspector.langs.each do |l|
       rss(l).items.each do |item|
         episodes.each do |e|
-          unless e.srt.include?(l) 
+          unless e.srt.include?(l)
             if episode_title(e, l) == item_title(item)
-              $stdout.print "FOUND: #{e.episode_name_with_format} [#{l}]\n"
               download_srt(item, e, l)
+              inspector.log.info "#{e.episode_name_with_format} [#{l}] // TVSubtitle // #{@srt_name}"
+              $stdout.print "FOUND: #{e.episode_name_with_format} [#{l}]\n"
               e.srt << l
               @inspector.growl_episode(e, l)
             end
@@ -33,6 +34,7 @@ class TVSubtitle
 private
   
   def download_srt(item, episode, lang)
+    found = false
     agent = WWW::Mechanize.new
     id = item_id(item)
     Tempfile.open("srt") do |tempfile|
@@ -41,10 +43,19 @@ private
       tempfile.close
       Zip::ZipFile.open(tempfile.path) do |zip_file|
         zip_file.each do |f|
-          zip_file.extract(f, episode.srt_filename(lang))
+          if f.size > 5000
+            @srt_name = f.name
+            zip_file.extract(f, episode.srt_filename(lang))
+            found = true
+          end
         end
       end
     end
+  rescue => e
+    @inspector.log.fatal "TVSubtitle // #{episode.episode_name_with_format} [#{lang}] // #{e}"
+    found = false
+  else
+    found
   end
   
   def rss(lang)

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification 
 name: pirate-autosub
 version: !ruby/object:Gem::Version 
-  version: 0.2.4
+  version: "0.3"
 platform: ruby
 authors: 
 - Pirate
@@ -9,11 +9,12 @@ autorequire:
 bindir: bin
 cert_chain: []
 
-date: 2009-01-17 00:00:00 -08:00
+date: 2009-03-15 00:00:00 -07:00
 default_executable: autosub
 dependencies: 
 - !ruby/object:Gem::Dependency 
   name: hpricot
+  type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
@@ -23,6 +24,7 @@ dependencies:
     version: 
 - !ruby/object:Gem::Dependency 
   name: rubyzip
+  type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
@@ -32,6 +34,7 @@ dependencies:
     version: 
 - !ruby/object:Gem::Dependency 
   name: optiflag
+  type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
@@ -41,6 +44,7 @@ dependencies:
     version: 
 - !ruby/object:Gem::Dependency 
   name: simple-rss
+  type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
@@ -50,15 +54,7 @@ dependencies:
     version: 
 - !ruby/object:Gem::Dependency 
   name: mechanize
-  version_requirement: 
-  version_requirements: !ruby/object:Gem::Requirement 
-    requirements: 
-    - - ">="
-      - !ruby/object:Gem::Version 
-        version: "0"
-    version: 
-- !ruby/object:Gem::Dependency 
-  name: echoe
+  type: :runtime
   version_requirement: 
   version_requirements: !ruby/object:Gem::Requirement 
     requirements: 
@@ -76,6 +72,7 @@ extra_rdoc_files:
 - bin/autosub
 - lib/episode.rb
 - lib/inspector.rb
+- lib/sites/podnapisi.rb
 - lib/sites/seriessub.rb
 - lib/sites/tvsubtitle.rb
 - README.markdown
@@ -86,11 +83,12 @@ files:
 - bin/autosub
 - lib/episode.rb
 - lib/inspector.rb
+- lib/sites/podnapisi.rb
 - lib/sites/seriessub.rb
 - lib/sites/tvsubtitle.rb
+- Manifest
 - Rakefile
 - README.markdown
-- Manifest
 has_rdoc: true
 homepage: http://github.com/pirate/autosub
 post_install_message: