scinote-web/lib/tasks/db_users.rake

138 lines
3.9 KiB
Ruby
Raw Normal View History

2016-02-12 23:52:43 +08:00
require "#{Rails.root}/app/utilities/users_generator"
include UsersGenerator
namespace :db do
desc "Load users into database from the provided YAML file"
task :load_users, [ :file_path, :create_orgs ] => :environment do |task, args|
if args.blank? or args.empty? or args[:file_path].blank?
puts "No file provided"
return
end
create_orgs = false
if args[:create_orgs].present? and args[:create_orgs].downcase == "true"
create_orgs = true
end
# Parse file
yaml = YAML.load(File.read("#{args[:file_path]}"))
begin
ActiveRecord::Base.transaction do
# Parse user & organization hashes from YAML
orgs = yaml.select{ |k, v| /org_[0-9]+/ =~ k }
users = yaml.select{ |k, v| /user_[0-9]+/ =~ k }
# Create organizations
orgs.each do |k, org_hash|
org = Organization.order(created_at: :desc).where(name: org_hash["name"]).first
if org.blank?
org = Organization.create({
name: org_hash["name"][0..99]
})
end
org_hash["id"] = org.id
end
# Create users
puts "Created users"
users.each do |k, user_hash|
password = user_hash["password"]
if password.blank?
password = generate_user_password
end
user_orgs = user_hash["organizations"]
if user_orgs.blank?
user_orgs = ""
end
org_ids =
user_orgs
.split(",")
.collect{ |o| o.strip }
.uniq
.select{ |o| orgs.include? o }
.collect{ |o| orgs[o]["id"] }
user = create_user(
user_hash["full_name"],
user_hash["email"],
password,
true,
create_orgs ? DEFAULT_PRIVATE_ORG_NAME : nil,
org_ids
)
if user.id.present? then
puts ""
print_user(user, password)
end
end
puts ""
end
rescue ActiveRecord::ActiveRecordError, ArgumentError
puts "Error creating all users, transaction reverted"
end
end
desc "Add a single user to the database"
task :add_user => :environment do
puts "Type in user's full name (e.g. 'Steve Johnson')"
full_name = $stdin.gets.to_s.strip
puts "Type in user's email (e.g. 'steve.johnson@gmail.com')"
email = $stdin.gets.to_s.strip
puts "Type in user's password (e.g. 'password'), or leave blank to let Rails generate password"
password = $stdin.gets.to_s.strip
if password.empty?
password = generate_user_password
end
puts "Do you want Rails to create default user's organization? (T/F)"
create_org = $stdin.gets.to_s.strip == "T"
puts "Type names of any additional organizations you want the user to be admin of (delimited with ','), or leave blank"
org_names = $stdin.gets.to_s.strip
if org_names.empty?
org_names = []
else
org_names = org_names.split(",").collect { |n| n.strip }
end
begin
ActiveRecord::Base.transaction do
# Add/fetch organizations if needed
org_ids = []
org_names.each do |org_name|
org = Organization.order(created_at: :desc).where(name: org_name).first
if org.blank? then
org = Organization.create({ name: org_name[0..99] })
end
org_ids << org.id
end
user = create_user(
full_name,
email,
password,
true,
create_org ? DEFAULT_PRIVATE_ORG_NAME : nil,
org_ids
)
if user.id.present? then
puts "Successfully created user"
puts ""
print_user(user, password)
else
puts "Error creating new user"
end
puts ""
end
2016-07-21 19:11:15 +08:00
rescue ActiveRecord::ActiveRecordError, ArgumentError, ActiveRecord::RecordNotSaved => e
puts "Error creating user, transaction reverted: #{$!}"
2016-02-12 23:52:43 +08:00
end
end
end