Separate data generation from csv save for sections

main.go

@@ -562,7 +562,7 @@ func SaveGeneratedUsers(users []User) {
 	fmt.Printf("Generated %d users to users.csv\n", len(users))
 }
 
-func GenerateSectionData(config Config) {
+func GenerateSectionData(config Config) []Section {
 	// Generate unique instructor names: "LastName, X"
 	instructorNames := make([]string, 0, config.NumberOfInstructors)
 	seenInstructors := make(map[string]struct{}, config.NumberOfInstructors)
@@ -598,6 +598,11 @@ func GenerateSectionData(config Config) {
 
 	}
 
+	return sections
+}
+
+func SaveGeneratedSections(sections []Section) {
+
 	f, err := os.Create("sections.csv")
 	if err != nil {
 		log.Fatalf("failed to create sections.csv: %v", err)
@@ -618,7 +623,7 @@ func GenerateSectionData(config Config) {
 		log.Fatalf("csv flush error: %v", err)
 	}
 
-	fmt.Printf("Generated %d sections to sections.csv\n", config.NumberOfSections)
+	fmt.Printf("Generated %d sections to sections.csv\n", len(sections))
 }
 
 func main() {
@@ -643,7 +648,8 @@ func main() {
 	generated_users := GenerateUserData(config)
 	SaveGeneratedUsers(generated_users)
 
-	GenerateSectionData(config)
+	generated_sections := GenerateSectionData(config)
+	SaveGeneratedSections(generated_sections)
 
 }