reset_run_data.rb

require 'rubygems'
require 'mongo'
require 'json'

$: << "agents/sail.rb/lib"

DB = "evoroom"
RUN = "michelle-fall-2011"

require 'sail/rollcall/user'
require 'sail/rollcall/group'

json = File.read("config.json")
config = JSON.parse(json, :symbolize_names => true)

mongo = Mongo::Connection.new.db(DB)

puts "Wiping data in mongo..."
mongo.collection(:location_tracking).remove()
mongo.collection(:interviews).remove()
mongo.collection(:events).remove()
mongo.collection(:rainforest_guesses).remove()
mongo.collection(:organism_presence).remove()

keep_metadata_keys = ["assigned_organism_1", "assigned_organism_2", "key"]

Rollcall::User.site = config[:rollcall][:url]

users_url = "/runs/#{RUN}/users.xml"
puts "Pulling users from #{users_url}"
Rollcall::User.format = ActiveResource::Formats::XmlFormat
Rollcall::User.find(:all, :from => users_url).each do |u|
  puts "Wiping metadata for #{u.account.login.inspect}..."
  (u.metadata.attributes.keys).each do |key|
    unless keep_metadata_keys.include?(key)
      puts " - #{key}"
      u.metadata.send("#{key}=", nil)
    end
    begin
      g = Rollcall::Group.find(u.groups.first.id)
      if g
        puts "Clearing metadata for group #{g.name}"
        g.metadata.assigned_location_for_guess = nil
        g.save
      end
    rescue NoMethodError
      puts "Not clearing group metadata for #{u.account.login.inspect}..."
    end
  end
  u.save
end