PyPoll.py

#The data we need to retrieve
#1. The total number of votes cast
#2. A complete list of candidates who received votes
#3. The percentage of votes each candidate won
#4. The total number of votes each candidate won
#5. The winner of the elction based on popular vote.

#Add our depdencies.
import csv
import os

# Assign a variable for the file to load and the path.
#Windows only file_to_load below
#file_to_load = "D:\Class Docs\Data Analysis Projects\Election_Analysis\Resources\election_results.csv"
#Mac only file_to_load below
#file_to_load = "/Users/dmccu/Documents/Class Docs/Analysis Projects/Election_Analysis/Resources/election_results.csv"
file_to_load = os.path.join("Resources", "election_results.csv")

#Assign a variable to save the file to a path.
file_to_save = os.path.join("election_analysis.txt")


#Initialize a total vote counter.
total_votes = 0
#Candidate optioins and candidate votes
candidate_options = []
candidate_votes = {}

#Track the winning candidate, vote count and percentage.
winning_candidate = ""
winning_count = 0
winning_percentage = 0


# Open the election results and read the file.
with open(file_to_load, 'r') as election_data:
    
    #Read the file object with the reader function.
    file_reader = csv.reader(election_data)
    #Read the header row.
    headers = next(file_reader)
    #Print each row in the CSV file.
    for row in file_reader:
        #Add to the total vote count.
        total_votes += 1
        #Get the candidate name for each row.
        candidate_name = row[2]

        #If the candidate does not match any existing cnadidate add it to the candidate list.
        if candidate_name not in candidate_options:
            #Add the candidate name to the candidate list.
            candidate_options.append(candidate_name)
            #And begin tracking that candidate's voter count.
            candidate_votes[candidate_name] = 0
        #Add a vote to that candidate's count
        candidate_votes[candidate_name] += 1

with open(file_to_save, "w") as txt_file:
    #Print the final vote count to the terminal.
    election_results = (
        f"\nElection Results\n"
        f"-------------------------------\n"
        f"Total Votes: {total_votes:,}\n"
        f"-------------------------------\n")
    print(election_results, end="")
    #Save the final vote count to the text file.
    txt_file.write(election_results)

    for candidate_name in candidate_votes:
        #Retrieve vote count and percentage.
        votes = candidate_votes[candidate_name]
        vote_percentage = float(votes) / float(total_votes) * 100
        #Print each candidate, their voter count, and percentage to the terminal.
        candidate_results = (f"{candidate_name}: {vote_percentage:.1f}% ({votes:,} votes)\n")
        print(candidate_results)
        #Save the candidate results to our text file.
        txt_file.write(candidate_results)

        #Determine winning vote count, winning percentage and candidate.
        if (votes > winning_count) and (vote_percentage > winning_percentage):
            winning_count = votes
            winning_percentage = vote_percentage
            winning_candidate = candidate_name

    #Print the winning candidates' results to the terminal.
    winning_candidate_summary =(
            f"---------------------------\n"
            f"Winner: {winning_candidate}\n"
            f"Winning Vote Count: {winning_count:,}\n"
            f"Winning Percentage: {winning_percentage:.1f}%\n"
            f"---------------------------\n")
    print(winning_candidate_summary)
    #Save the winning candidate's name to the text file.
    txt_file.write(winning_candidate_summary)