MySQL import for split files!

mysql-split-import

#!/usr/bin/env ruby

# MySQL import for split files!
#
# Usage:
#   mysql-split-import <USERNAME> <PASSWORD> <DATABASE> <BACKUP_DIR>
#   mysql-split-import a_user their_password some_database /path/to/backup

username = ARGV[0]
password = ARGV[1]
db_name = ARGV[2]
directory = ARGV[3]

exit unless username && password && db_name && directory

directory = File.absolute_path(directory)
Dir.entries(directory).select { |f| f =~ /\.sql$/ }.each do |file|
  sql_file = File.join(directory, file)
  txt_file = sql_file.gsub(/\.sql$/, '.txt')
  puts "Processing #{file}."
  `mysql -u #{username} -p#{password} #{db_name} < #{sql_file}`
  if File.exists?(txt_file)
    `mysqlimport -u #{username} -p#{password} #{db_name} #{txt_file}`
  end
end