From 39580bfdb80465d78d96ef4d712a0f8c13d7e3d8 Mon Sep 17 00:00:00 2001
From: Joshua Herring <jwherrin@iu.edu>
Date: Sat, 18 Apr 2026 12:21:23 -0400
Subject: [PATCH] add Stage to Section and generate only stage prelim for now

---
 grading_data_generator.go | 6 ++++--
 1 file changed, 4 insertions(+), 2 deletions(-)

diff --git a/grading_data_generator.go b/grading_data_generator.go
index fa951d4..7417899 100644
--- a/grading_data_generator.go
+++ b/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)
 		}
 	}