mass importer command line script
Andrew Coleman
parent
df7e487041
commit
8fd1eb3567
|
|
@ -0,0 +1,193 @@
|
|||
#!/usr/bin/env ruby
|
||||
#
|
||||
# TuxBliki image file importer. Use locally, saves remotely.
|
||||
#
|
||||
# == Synopsis
|
||||
#
|
||||
# Automatically creates the album if not found, so no worries there. Requires
|
||||
# ImageMagick's `convert` program to function. Also automatically resizes images
|
||||
# to a maxiumum of 1280 x 1280. Parameters are required unless otherwise noted.
|
||||
# Also uses `identify` to determine if resizing is required.
|
||||
#
|
||||
# == Usage
|
||||
#
|
||||
# ./bin/import_images.rb <options>
|
||||
#
|
||||
# --help, -h
|
||||
# This help message.
|
||||
#
|
||||
# --file <filename>
|
||||
# The image you are going to upload. Must not be in /tmp for conversion.
|
||||
#
|
||||
# --directory <directory>
|
||||
# The directory of images to upload.
|
||||
#
|
||||
# --album <album_name>
|
||||
# The name of the Album to import the file into.
|
||||
#
|
||||
# --username <username>
|
||||
# The username in TuxBliki to use as the owner.
|
||||
#
|
||||
# --password <password>
|
||||
# The username's password in TuxBliki. Check your shell logs...
|
||||
#
|
||||
# --server <server address>
|
||||
# The server running TuxBliki, optional; defaults to http://penguincoder.org
|
||||
#
|
||||
# --port <server port>
|
||||
# The port to use, optional; defaults to 80.
|
||||
#
|
||||
require 'rdoc/usage'
|
||||
require 'getoptlong'
|
||||
require 'mechanize'
|
||||
require 'uri'
|
||||
|
||||
opts = GetoptLong.new(
|
||||
[ '--help', '-h', GetoptLong::NO_ARGUMENT ],
|
||||
[ '--file', GetoptLong::OPTIONAL_ARGUMENT ],
|
||||
[ '--directory', GetoptLong::OPTIONAL_ARGUMENT ],
|
||||
[ '--album', GetoptLong::REQUIRED_ARGUMENT ],
|
||||
[ '--username', GetoptLong::REQUIRED_ARGUMENT ],
|
||||
[ '--password', GetoptLong::REQUIRED_ARGUMENT ],
|
||||
[ '--server', GetoptLong::OPTIONAL_ARGUMENT ],
|
||||
[ '--port', GetoptLong::OPTIONAL_ARGUMENT ]
|
||||
)
|
||||
|
||||
filename = nil
|
||||
directory = nil
|
||||
album = nil
|
||||
username = nil
|
||||
password = nil
|
||||
server = 'http://penguincoder.org'
|
||||
port = 80
|
||||
|
||||
opts.each do |opt, arg|
|
||||
case opt
|
||||
when '--file'
|
||||
filename = arg
|
||||
when '--directory'
|
||||
directory = arg.gsub(/\/$/, '')
|
||||
when '--album'
|
||||
album = arg
|
||||
when '--username'
|
||||
username = arg
|
||||
when '--password'
|
||||
password = arg
|
||||
when '--server'
|
||||
server = arg
|
||||
when '--port'
|
||||
port = arg.to_i
|
||||
end
|
||||
end
|
||||
|
||||
# process argument sanity
|
||||
if filename.nil? and directory.nil?
|
||||
RDoc::usage
|
||||
exit 1
|
||||
elsif album.nil?
|
||||
RDoc::usage
|
||||
exit 1
|
||||
elsif filename and !File.readable?(filename)
|
||||
puts "File is not readable!"
|
||||
exit 1
|
||||
elsif directory and !File.directory?(directory)
|
||||
puts "Directory is not a directory!"
|
||||
exit 1
|
||||
end
|
||||
|
||||
# build request
|
||||
agent = WWW::Mechanize.new
|
||||
host_prefix = "#{server.gsub(/\/$/, '')}:#{port}"
|
||||
|
||||
# log into the software first
|
||||
page = agent.get("#{host_prefix}/sessions/new")
|
||||
login_form = page.forms.first
|
||||
login_form.username = username
|
||||
login_form.password = password
|
||||
page = agent.submit(login_form, login_form.buttons.first)
|
||||
|
||||
# redirected back, so it is a failure
|
||||
if page.uri.to_s =~ /sessions/
|
||||
puts "Login failed!"
|
||||
exit 1
|
||||
else
|
||||
puts "* Successfully logged in as '#{username}'"
|
||||
end
|
||||
|
||||
# make sure the album is there, create it otherwise...
|
||||
begin
|
||||
page = agent.get("#{host_prefix}/albums/#{album.gsub(/ /, '_')}")
|
||||
puts "* Found the album '#{album}'"
|
||||
rescue WWW::Mechanize::ResponseCodeError => exception
|
||||
if exception.response_code.to_i == 404
|
||||
puts "Album does not exist, creating it!"
|
||||
page = agent.get("#{host_prefix}/albums/new")
|
||||
album_form = page.forms.first
|
||||
album_form['album[name]'] = album
|
||||
page = agent.submit(album_form, album_form.buttons.first)
|
||||
if page.uri.to_s =~ /new/
|
||||
puts "Album could not be created!"
|
||||
exit 1
|
||||
else
|
||||
puts "* Album was successfully created"
|
||||
end
|
||||
else
|
||||
puts "There was a problem finding the album: #{exception}"
|
||||
exit 1
|
||||
end
|
||||
end
|
||||
|
||||
# figure out how many files to process
|
||||
files = if filename
|
||||
[ filename ]
|
||||
else
|
||||
Dir.glob("#{directory}/*.*")
|
||||
end
|
||||
puts "* Processing #{files.size} files"
|
||||
|
||||
# set up the mechanize page and form submission so that you only do this once
|
||||
page = agent.get("#{host_prefix}/photos/new")
|
||||
photo_form = page.forms.first
|
||||
photo_form['photo[album_id]'] = album
|
||||
|
||||
# process
|
||||
files.each_with_index do |fname, idx|
|
||||
puts "-> #{fname}"
|
||||
|
||||
# determine picture size
|
||||
width, height = nil, nil
|
||||
`identify #{fname}`.chomp.gsub(/ (\d+)x(\d+) /) do |match|
|
||||
width = $1.to_i
|
||||
height = $2.to_i
|
||||
end
|
||||
|
||||
# convert if necessary
|
||||
converted = false
|
||||
basefname = nil
|
||||
if width.to_i > 1280 or height.to_i > 1280
|
||||
converted = true
|
||||
print "Converting... "
|
||||
STDOUT.flush
|
||||
basefname = "/tmp/#{File.basename(fname)}"
|
||||
`convert "#{fname}" -resize "1280x1280" "#{basefname}"`
|
||||
print "Done. "
|
||||
STDOUT.flush
|
||||
end
|
||||
|
||||
# now upload it
|
||||
print "Uploading... "
|
||||
STDOUT.flush
|
||||
photo_form.file_uploads.first.file_name = (converted ? basefname : fname)
|
||||
page = agent.submit(photo_form, photo_form.buttons.first)
|
||||
if page.uri.to_s =~ /new/
|
||||
puts "Failed!"
|
||||
else
|
||||
puts "Success."
|
||||
end
|
||||
|
||||
# remove resized image
|
||||
File.unlink(basefname) if converted
|
||||
end
|
||||
|
||||
puts "Fin."
|
||||
|
||||
Reference in New Issue