Separate User generation from csv generation

main.go

@@ -509,7 +509,7 @@ type Config struct {
 
 var statuses = []string{"JD", "LLM", "MCL", "PHD", "SJD", "MLS", "SPEC", "MNR", "CERT"}
 
-func GenerateUserData(config Config) {
+func GenerateUserData(config Config) []User {
 	// Build a weighted list of statuses
 	weightedStatuses := make([]string, 0, 100)
 	for status, pct := range config.PercentStatus {
@@ -533,6 +533,12 @@ func GenerateUserData(config Config) {
 		}
 	}
 
+	return users
+
+}
+
+func SaveGeneratedUsers(users []User) {
+
 	f, err := os.Create("users.csv")
 	if err != nil {
 		log.Fatalf("failed to create users.csv: %v", err)
@@ -553,7 +559,7 @@ func GenerateUserData(config Config) {
 		log.Fatalf("csv flush error: %v", err)
 	}
 
-	fmt.Printf("Generated %d users to users.csv\n", config.NumberOfUsers)
+	fmt.Printf("Generated %d users to users.csv\n", len(users))
 }
 
 func GenerateSectionData(config Config) {
@@ -634,7 +640,9 @@ func main() {
 		log.Fatalf("percent_status values must total 100, got %d", total)
 	}
 
-	GenerateUserData(config)
+	generated_users := GenerateUserData(config)
+	SaveGeneratedUsers(generated_users)
+
 	GenerateSectionData(config)
 
 }