Class: DNN::Downloader

Inherits:
Object
  • Object
show all
Defined in:
lib/dnn/datasets/downloader.rb

Class Method Summary collapse

Instance Method Summary collapse

Constructor Details

#initialize(url) ⇒ Downloader

Returns a new instance of Downloader.



23
24
25
26
 
# File 'lib/dnn/datasets/downloader.rb', line 23

def initialize(url)
  @url = url
  *, @fqdn, @path = *url.match(%r`https?://(.+?)(/.+)`)
end

Class Method Details

.download(url, dir_path = nil) ⇒ Object 



13
14
15
16
17
18
19
20
21
 
# File 'lib/dnn/datasets/downloader.rb', line 13

def self.download(url, dir_path = nil)
  unless dir_path
    Dir.mkdir("#{DOWNLOADS_PATH}/downloads") unless Dir.exist?("#{DOWNLOADS_PATH}/downloads")
    dir_path = "#{DOWNLOADS_PATH}/downloads"
  end
  Downloader.new(url).download(dir_path)
rescue => e
  raise DNN_DownloadError.new(e.message)
end

Instance Method Details

#download(dir_path) ⇒ Object 



28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
 
# File 'lib/dnn/datasets/downloader.rb', line 28

def download(dir_path)
  puts %`download "#{@url}"`
  buf = ""
  Net::HTTP.start(@fqdn) do |http|
    content_length = http.head(@path).content_length
    http.get(@path) do |body_segment|
      buf << body_segment
      log = "\r"
      40.times do |i|
        if i < buf.size * 40 / content_length
          log << "="
        elsif i == buf.size * 40 / content_length
          log << ">"
        else
          log << "_"
        end
      end
      log << "  #{buf.size}/#{content_length}"
      print log
    end
    puts ""
  end
  file_name = @path.match(%r`.*/(.+)`)[1]
  File.binwrite("#{dir_path}/#{file_name}", buf)
end