You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
sqm-dashboards/lib/tasks/data.rake

87 lines
2.8 KiB

namespace(:data) do
desc("load survey responses")
task(load_survey_responses: :environment) do
survey_item_response_count = SurveyItemResponse.count
student_count = Student.count
path = "/data/survey_responses/clean/"
Sftp::Directory.open(path:) do |file|
SurveyResponsesDataLoader.new.from_file(file:)
end
puts(
"=====================> Completed loading #{SurveyItemResponse.count - survey_item_response_count} survey responses. #{SurveyItemResponse.count} total responses in the database"
)
Rails.cache.clear
end
desc("load survey responses from a specific directory")
task(load_survey_responses_from_path: :environment) do
survey_item_response_count = SurveyItemResponse.count
student_count = Student.count
path = "#{ENV["SFTP_PATH"]}"
Sftp::Directory.open(path:) do |file|
SurveyResponsesDataLoader.new.from_file(file:)
end
puts(
"=====================> Completed loading #{SurveyItemResponse.count - survey_item_response_count} survey responses. #{SurveyItemResponse.count} total responses in the database"
)
Rails.cache.clear
end
desc("load admin_data")
task(load_admin_data: :environment) do
original_count = AdminDataValue.count
pool_size = 2
jobs = Queue.new
Dir.glob(Rails.root.join("data", "admin_data", "dese", "*.csv")).each { |filepath| jobs << filepath }
Dir.glob(Rails.root.join("data", "admin_data", "out_of_state", "*.csv")).each { |filepath| jobs << filepath }
workers = pool_size.times.map do
Thread.new do
while filepath = jobs.pop(true)
puts("=====================> Loading data from csv at path: #{filepath}")
Dese::Loader.load_data(filepath:)
end
rescue ThreadError
end
end
workers.each(&:join)
puts("=====================> Completed loading #{AdminDataValue.count - original_count} admin data values")
end
desc("reset all cache counters")
task(reset_cache_counters: :environment) do
puts("=====================> Resetting Category counters")
Category.all.each do |category|
Category.reset_counters(category.id, :subcategories)
end
puts("=====================> Resetting Subcategory counters")
Subcategory.all.each do |subcategory|
Subcategory.reset_counters(subcategory.id, :measures)
end
puts("=====================> Resetting Measure counters")
Measure.all.each do |measure|
Measure.reset_counters(measure.id, :scales)
end
puts("=====================> Resetting Scale counters")
Scale.all.each do |scale|
Scale.reset_counters(scale.id, :survey_items)
end
puts("=====================> Resetting SurveyItem counters")
SurveyItem.all.each do |survey_item|
SurveyItem.reset_counters(survey_item.id, :survey_item_responses)
end
end
end