From 6dddb42899370996314c958fab92fe0d982729bd Mon Sep 17 00:00:00 2001
From: NunoSempere <nuno.semperelh@protonmail.com>
Date: Sun, 14 Apr 2024 10:09:46 -0400
Subject: [PATCH] filter recent polls

---
 main.go | 53 ++++++++++++++++++++++++++++++-----------------------
 1 file changed, 30 insertions(+), 23 deletions(-)

diff --git a/main.go b/main.go
index ad03f5e..4c558b3 100644
--- a/main.go
+++ b/main.go
@@ -187,15 +187,21 @@ func readStates() ([]State, error) {
 }
 
 func sampleFromState(state State) VotesForEachParty {
+	fmt.Printf("\n\nState: %s", state.Name)
+	fmt.Printf("\n\tVotes: %d", state.Votes)
+	fmt.Printf("\n\tHistory: %s", state.PresidentialElectoralHistory)
+	fmt.Printf("\n\tPolls: %s", state.Polls)
 
 	switch state.Name {
 	case "Nebraska":
-		// 2000: R
-		// 2004: R
-		// 2008: Split, 1 D, 4 R
-		// 2012: R
-		// 2016: R
-		// 2020: Split, 1 D, 4 R
+		/*
+			2000: R
+			2004: R
+			2008: Split, 1 D, 4 R
+			2012: R
+			2016: R
+			2020: Split, 1 D, 4 R
+		*/
 		p_split := 2.0 / 6.0
 		if r.Float64() < p_split {
 			return VotesForEachParty{Democrats: 1, Republicans: 4}
@@ -203,12 +209,14 @@ func sampleFromState(state State) VotesForEachParty {
 			return VotesForEachParty{Democrats: 0, Republicans: 5}
 		}
 	case "Maine":
-		// 2000: D
-		// 2004: D
-		// 2008: D
-		// 2012: D
-		// 2016: Split: 3 D, 1 R
-		// 2020: Split, 3 D, 1 R
+		/*
+			2000: D
+			2004: D
+			2008: D
+			2012: D
+			2016: Split: 3 D, 1 R
+			2020: Split, 3 D, 1 R
+		*/
 		p_split := 2.0 / 6.0
 		if r.Float64() < p_split {
 			return VotesForEachParty{Democrats: 3, Republicans: 1}
@@ -217,6 +225,7 @@ func sampleFromState(state State) VotesForEachParty {
 		}
 	default:
 		{
+			/* Just considering the base rate for the state */
 			p_republican := 0.0
 			for _, party := range state.PresidentialElectoralHistory {
 				if party == "R" {
@@ -229,6 +238,15 @@ func sampleFromState(state State) VotesForEachParty {
 			} else {
 				return VotesForEachParty{Democrats: state.Votes, Republicans: 0}
 			}
+
+			/* Considering polls */
+			var recent_polls []Poll
+			for _, poll := range state.Polls {
+				if poll.Date.After(time.Now().AddDate(0, 0, -30)) {
+					recent_polls = append(recent_polls, poll)
+				}
+			}
+			return VotesForEachParty{Democrats: 1, Republicans: 1}
 		}
 	}
 }
@@ -237,21 +255,10 @@ func simulateElection(states []State) int {
 
 	republican_seats := 0
 	for _, state := range states {
-		fmt.Printf("\n\nState: %s\n\tVotes: %d\n\tHistory: %s\n\tPolls: %s", state.Name, state.Votes, state.PresidentialElectoralHistory, state.Polls)
 		election_sample := sampleFromState(state)
 		republican_seats += election_sample.Republicans
-		/*
-			fmt.Printf("%s: Votes: %d,\n\tWinners: ", state.Name, state.Votes)
-			for year, party := range state.PresidentialElectoralHistory {
-				fmt.Printf("[%s: %s] ", year, party)
-			}
-
-			fmt.Printf("\n\tSample: Democrat seats: %d, Republican seats: %d", election_sample.Democrats, election_sample.Republicans)
-			fmt.Println()
-		*/
 	}
 
-	// fmt.Printf("\nDemocrat seats: %d\n", republican_seats)
 	fmt.Printf(" (%d) ", republican_seats)
 	if republican_seats >= 270 {
 		return 1