diff --git a/main.go b/main.go index 4d30a24..c316ef1 100644 --- a/main.go +++ b/main.go @@ -4,32 +4,43 @@ import ( "encoding/csv" "fmt" "os" + "strconv" "strings" ) /* Electoral votes */ type StateVotes struct { State string - Votes string + Votes int } func readElectoralVotes(filename string) ([]StateVotes, error) { + var stateVotes []StateVotes + file, err := os.Open(filename) if err != nil { return nil, fmt.Errorf("error opening the file: %v", err) } defer file.Close() - var stateVotes []StateVotes - reader := csv.NewReader(file) + // Skip the first row ('State, Votes' header) + if _, err := reader.Read(); err != nil { + return nil, fmt.Errorf("error reading header row: %v", err) + } + for { record, err := reader.Read() // End of file is expected, not an error in this context if err != nil { break } - stateVotes = append(stateVotes, StateVotes{State: record[0], Votes: record[1]}) + + votes, err := strconv.Atoi(record[1]) + if err != nil { + return nil, fmt.Errorf("error converting votes for %s to int: %v", record[0], err) + } + stateVotes = append(stateVotes, StateVotes{State: record[0], Votes: votes}) } if len(stateVotes) == 0 { @@ -39,17 +50,20 @@ func readElectoralVotes(filename string) ([]StateVotes, error) { return stateVotes, nil } -func findElectoralVotes(stateVotes []StateVotes, stateName string) (string, error) { +func findElectoralVotes(stateVotes []StateVotes, stateName string) (int, error) { for _, sv := range stateVotes { if strings.EqualFold(sv.State, stateName) { return sv.Votes, nil } } - return "", fmt.Errorf("state not found") + return 0, fmt.Errorf("state not found") } +/* Results for the elections */ + func main() { + // Get electoral votes filename := "data/electoral-college-votes.csv" stateVotes, err := readElectoralVotes(filename) if err != nil { @@ -57,6 +71,8 @@ func main() { return } + // Get baserates + // Example states to query states := []string{"California", "Texas", "Florida"} for _, state := range states { @@ -64,7 +80,7 @@ func main() { if err != nil { fmt.Printf("Error: %s\n", err) } else { - fmt.Printf("%s has %s electoral votes.\n", state, votes) + fmt.Printf("%s has %d electoral votes.\n", state, votes) } } }