From 5118358b4368d33611940708de59ca61333dcdd0 Mon Sep 17 00:00:00 2001
From: rebuilt <memoryman51@hotmail.com>
Date: Mon, 16 Jan 2023 17:04:12 -0800
Subject: [PATCH] Also run student demographic loader when loading survey data

---
 lib/tasks/data.rake | 15 ++++++++++++++-
 1 file changed, 14 insertions(+), 1 deletion(-)

diff --git a/lib/tasks/data.rake b/lib/tasks/data.rake
index ed88aea4..c03ccda3 100644
--- a/lib/tasks/data.rake
+++ b/lib/tasks/data.rake
@@ -3,11 +3,24 @@ require 'csv'
 namespace :data do
   desc 'load survey responses'
   task load_survey_responses: :environment do
+    survey_item_response_count = SurveyItemResponse.count
+    student_count = Student.count
+
     Dir.glob(Rails.root.join('data', 'survey_responses', '*.csv')).each do |filepath|
       puts "=====================> Loading data from csv at path: #{filepath}"
       SurveyResponsesDataLoader.load_data filepath:
     end
-    puts "=====================> Completed loading #{SurveyItemResponse.count} survey responses"
+    puts "=====================> Completed loading #{SurveyItemResponse.count - survey_item_response_count} survey responses. #{SurveyItemResponse.count} total responses in the database"
+
+    SurveyItemResponse.update_all(student_id: nil)
+    StudentRace.delete_all
+    Student.delete_all
+
+    Dir.glob(Rails.root.join('data', 'survey_responses', '*student*.csv')).each do |file|
+      puts "=====================> Loading student data from csv at path: #{file}"
+      StudentLoader.load_data filepath: file
+    end
+    puts "=====================> Completed loading #{Student.count - student_count} students. #{Student.count} total students"
 
     puts 'Resetting response rates'
     ResponseRateLoader.reset