Change data download methods to allow multiple exercise ids

Author: ayaankazerouni

app/admin/course_offering.rb

@@ -8,7 +8,7 @@
   index do
     id_column
     column :course, sortable: 'courses.number' do |c|
-      link_to c.course.number_and_org, admin_course_path(c.course)
+      link_to c.course.number_and_org, admin_course_path(c.course.id)
     end
     column :term, sortable: 'term.ends_on' do |c|
       link_to c.term.display_name, admin_term_path(c.term)

app/controllers/exercises_controller.rb

@@ -56,10 +56,14 @@ def query_data
 
   # -------------------------------------------------------------
   def download_attempt_data
-    @exercise = Exercise.find params[:id]
+    exercise_id = params[:id] # may be one or more ids
+    course_offering_id = params[:course_offering_id]
+    workout_id = params[:workout_id]
+
+    time = Time.now.utc.strftime("%Y%m%d%H%M%S")
 
     if params[:progsnap].to_b
-      main_events, code_states = @exercise.progsnap2_attempt_csv
+      main_events, code_states = Exercise.progsnap2_attempt_csv(exercise_id, course_offering_id, workout_id)
       compressed_filestream = Zip::OutputStream.write_buffer do |zos|
         main_events_file = "MainTable.csv"
         zos.put_next_entry main_events_file
@@ -73,15 +77,15 @@ def download_attempt_data
       compressed_filestream.rewind
       respond_to do |format|
         format.zip do
-          send_data compressed_filestream.read, filename: "X#{params[:id]}-progsnap.zip", type: 'application/zip'
+          send_data compressed_filestream.read, filename: "CW-progsnap-#{time}.zip", type: 'application/zip'
         end
       end
     else
-      result = @exercise.denormalized_attempt_csv
+      result = Exercise.denormalized_attempt_csv(exercise_id)
 
       respond_to do |format|
         format.csv do
-          send_data result, filename: "X#{params[:id]}-submissions.csv"
+          send_data result, filename: "CW-#{time}-submissions.csv"
         end
       end
     end

app/models/exercise.rb

@@ -310,15 +310,15 @@ def is_publicly_available?
     end
   end
 
-  def progsnap2_attempt_csv
-    denormalized = self.denormalized_attempt_data
-    main_events = progsnap2_main_events_csv(denormalized)
-    code_states = progsnap2_code_states_csv(denormalized)
+  def self.progsnap2_attempt_csv(exercise_id, course_offering_id = nil, workout_id = nil) 
+    denormalized = Exercise.denormalized_attempt_data(exercise_id, course_offering_id, workout_id)
+    main_events = Exercise.progsnap2_main_events_csv(denormalized)
+    code_states = Exercise.progsnap2_code_states_csv(denormalized)
     return main_events, code_states
   end
 
-  def denormalized_attempt_csv
-    denormalized_data = self.denormalized_attempt_data
+  def self.denormalized_attempt_csv(exercise_id)
+    denormalized_data = Exercise.denormalized_attempt_data(exercise_id)
     exercise_attributes = %w{ exercise_id exercise_name }
     attempt_attributes = %w{
       user_id
@@ -356,15 +356,23 @@ def denormalized_attempt_csv
   # All relationship fields are in the same table, so null values
   # are possible for workout_id, workout_offering_id, course_id,
   # course_offering_id, etc.
-  def denormalized_attempt_data(workout_id = nil)
-    result = exercise_versions.joins{ attempts.prompt_answers }
+  def self.denormalized_attempt_data(exercise_id, course_offering_id = nil, workout_id = nil)
+    course_offering_filter = course_offering_id ? 
+      "AND course_offerings.id = #{course_offering_id}" :
+      ""
+    workout_filter = workout_id ?
+      "AND workouts.id = #{workout_id}" :
+      ""
+
+    result = Exercise.where(:id, exercise_id)
+      .joins(exercise_versions: { attempts: :prompt_answers })
       .joins('LEFT JOIN workout_scores ON
         workout_scores.id = attempts.workout_score_id')
       .joins('LEFT JOIN workout_offerings ON
         workout_offerings.id = workout_scores.workout_offering_id')
-      .joins('LEFT JOIN workouts ON workouts.id = workout_scores.workout_id')
-      .joins('LEFT JOIN course_offerings ON
-        course_offerings.id = workout_offerings.course_offering_id')
+      .joins("LEFT JOIN workouts ON workouts.id = workout_scores.workout_id #{workout_filter}")
+      .joins("LEFT JOIN course_offerings ON
+        course_offerings.id = workout_offerings.course_offering_id #{course_offering_filter}")
       .joins('LEFT JOIN terms ON terms.id = course_offerings.term_id')
       .joins('LEFT JOIN courses ON courses.id = course_offerings.course_id')
       .joins('LEFT JOIN coding_prompt_answers ON
@@ -389,9 +397,9 @@ def denormalized_attempt_data(workout_id = nil)
         courses.number as course_number,
         courses.name as course_name,
         terms.slug as term')
-    if workout_id
-      result = result.where("workouts.id = #{workout_id}")
-    end
+    # if workout_id
+    #   result = result.where("workouts.id = #{workout_id}")
+    # end
 
     return result
   end
@@ -412,7 +420,7 @@ def set_defaults
     self.experience ||= 10
   end
 
-  def progsnap2_main_events_csv(denormalized_data)
+  def self.progsnap2_main_events_csv(denormalized_data)
     # MainTable
     main_attributes = %w{
       SubjectID
@@ -520,7 +528,7 @@ def progsnap2_main_events_csv(denormalized_data)
     return data
   end
 
-  def progsnap2_code_states_csv(denormalized_data)
+  def self.progsnap2_code_states_csv(denormalized_data)
     code_state_attributes = %w{
       CodeStateID
       Code

config/routes.rb

@@ -64,7 +64,7 @@
     get 'exercises/search' => 'exercises#search', as: :exercises_search
     get 'exercises/query_data' => 'exercises#query_data',
       as: :exercises_query_data
-    get 'exercises/:id/download_attempt_data' =>
+    get 'exercises/download_attempt_data' =>
       'exercises#download_attempt_data', as: :download_exercise_attempt_data
     # At the bottom, so the routes above take precedence over existing ids
     resources :exercises