Enforce checks

service/pipeline.py

@@ -24,7 +24,7 @@ while True:
     start_time = time()
 
     # Get emails, student IDs, and usernames of students (excludes default users and instructors)
-    query = ("SELECT id, username, email FROM auth_user WHERE last_login IS NOT NULL AND is_superuser = 0")
+    query = ("SELECT id, username, email FROM auth_user WHERE last_login IS NOT NULL AND is_superuser = 0 AND is_staff = 0")
     mysql_cursor.execute(query)
     student_dict = {}
     for (id, username, email) in mysql_cursor:
@@ -112,28 +112,29 @@ while True:
     json_data = []
     index = 0
     for student_id in student_dict:
-        json_data.append({})
-        json_data[index]["Email"] = student_dict[student_id][1]
-        json_data[index]["Username"] = student_dict[student_id][0]
-        json_data[index]["Overall Grade"] = str(student_dict[student_id][2])
-        grade_data = student_dict[student_id][3]
-        quiz_ids_ordered = []
-        for quiz_id in grade_data:
-            quiz_name = quiz_id_to_quiz_name[quiz_id]
-            grades = grade_data[quiz_id]
-            json_data[index]["Grade on " + quiz_name] = str(grades[0])
-            quiz_ids_ordered.append(quiz_id)
-        for quiz_id in quiz_ids_ordered:
-            quiz_name = quiz_id_to_quiz_name[quiz_id]
-            question_grades = grade_data[quiz_id][1]
-            question_grades_length = len(question_grades)
-            question_index = 0
-            while question_index < question_grades_length:
-                question_index_string = str(int(question_index / 2) + 1)
-                json_data[index][quiz_name + " - Q" + question_index_string + " (Earned)"] = str(question_grades[question_index])
-                json_data[index][quiz_name + " - Q" + question_index_string + " (Possible)"] = str(question_grades[question_index + 1])
-                question_index += 2
-        index += 1
+        if len(student_dict[student_id]) == 4: # Ensures that we can access all list elements below
+            json_data.append({})
+            json_data[index]["Email"] = student_dict[student_id][1]
+            json_data[index]["Username"] = student_dict[student_id][0]
+            json_data[index]["Overall Grade"] = str(student_dict[student_id][2])
+            grade_data = student_dict[student_id][3]
+            quiz_ids_ordered = []
+            for quiz_id in grade_data:
+                quiz_name = quiz_id_to_quiz_name[quiz_id]
+                grades = grade_data[quiz_id]
+                json_data[index]["Grade on " + quiz_name] = str(grades[0])
+                quiz_ids_ordered.append(quiz_id)
+            for quiz_id in quiz_ids_ordered:
+                quiz_name = quiz_id_to_quiz_name[quiz_id]
+                question_grades = grade_data[quiz_id][1]
+                question_grades_length = len(question_grades)
+                question_index = 0
+                while question_index < question_grades_length:
+                    question_index_string = str(int(question_index / 2) + 1)
+                    json_data[index][quiz_name + " - Q" + question_index_string + " (Earned)"] = str(question_grades[question_index])
+                    json_data[index][quiz_name + " - Q" + question_index_string + " (Possible)"] = str(question_grades[question_index + 1])
+                    question_index += 2
+            index += 1
     with open("csvToPy.json", "w") as json_file:
         dump(json_data, json_file, indent=2)