From 8fd1eb3567b1541d16b1157ddb2b144c2853d128 Mon Sep 17 00:00:00 2001
From: Andrew Coleman <andrew@linux-a1p5.(none)>
Date: Sat, 21 Mar 2009 21:01:38 -0500
Subject: [PATCH] mass importer command line script

---
 bin/import_images.rb | 193 +++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 193 insertions(+)
 create mode 100755 bin/import_images.rb

diff --git a/bin/import_images.rb b/bin/import_images.rb
new file mode 100755
index 0000000..e196d7e
--- /dev/null
+++ b/bin/import_images.rb
@@ -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."
+