From c4e64b2cbea4b08f55fb6b4eee23229339ab2c60 Mon Sep 17 00:00:00 2001 From: Joshua Herring Date: Sun, 5 Apr 2026 20:37:56 -0400 Subject: [PATCH] Separate data generation from csv save for sections --- main.go | 12 +++++++++--- 1 file changed, 9 insertions(+), 3 deletions(-) diff --git a/main.go b/main.go index 29d8b3c..bc933fa 100644 --- a/main.go +++ b/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) }