add Stage to Section and generate only stage prelim for now

grading_data_generator.go

@@ -504,6 +504,7 @@ type Section struct {
 	CourseTitle  string `json:"course_title"`
 	CourseType   string `json:"course_type"`
 	GradesDue    string `json:"grades_due"`
+	Stage    string `json:"stage"`
 }
 
 type Config struct {
@@ -615,6 +616,7 @@ func GenerateSectionData(config Config, users []User) []Section {
 		my_section.Instructor = instructorUIDs[rand.Intn(len(instructorUIDs))]
 		my_section.CourseType = "default"
 		my_section.GradesDue = dueDates[rand.Intn(len(dueDates))]
+		my_section.Stage = "prelim"
 		sections[i] = my_section
 
 	}
@@ -631,11 +633,11 @@ func SaveGeneratedSections(sections []Section) {
 	defer f.Close()
 
 	w := csv.NewWriter(f)
-	if err := w.Write([]string{"section_id", "course_number", "course_instructor", "course_title", "course_type", "grades_due"}); err != nil {
+	if err := w.Write([]string{"section_id", "course_number", "course_instructor", "course_title", "course_type", "grades_due", "stage"}); err != nil {
 		log.Fatalf("failed to write csv header: %v", err)
 	}
 	for _, s := range sections {
-		if err := w.Write([]string{s.SectionID, s.CourseNumber, s.Instructor, s.CourseTitle, s.CourseType, s.GradesDue}); err != nil {
+		if err := w.Write([]string{s.SectionID, s.CourseNumber, s.Instructor, s.CourseTitle, s.CourseType, s.GradesDue, s.Stage}); err != nil {
 			log.Fatalf("failed to write csv row: %v", err)
 		}
 	}