run_all_elections.py

#######################################
## Run all scottish elections
## Save winners for each method in file
#######################################
from elections_class import mw_elections as mwe
import os
import csv
import sys


##################################
## Chose which elections to run ##
##################################
run_scottish = True
run_adapt_quota = False
run_meek = True
run_cham_cour_OM = True
run_cham_cour_PM = True
run_greedy_cham_cour = False
run_expand_app = True
run_cpo_stv = False
###################################
###################################

which_methods = [run_scottish, run_adapt_quota, run_meek, run_cham_cour_OM, run_cham_cour_PM, run_greedy_cham_cour, run_expand_app, run_cpo_stv]
headers = ['Scottish STV Winners', 'Adaptive Quota STV Winners', 'Meek STV Winners', 'Chamberlin-Courant (OM) Winners', 'Chamberlin-Courant (PM) Winners', 'Greedy Chamberlin-Courant Winners', 'Expanding Approvals Winners', 'CPO-STV Winners']
csv_headers = ['Election']
for i in range(len(which_methods)):
    if which_methods[i]:
        csv_headers.append(headers[i])
        
## create list of file names for all scottish elections
file_names = {}
for folder_name in os.listdir('../full_scot_data'):
    # print(folder_name)
    if 'cands' in folder_name:
        for name in os.listdir(f'../full_scot_data/{folder_name}'):
            # print(file_name)
            file_name = '../full_scot_data/'+folder_name+'/'+name
            file_names[name] = file_name
 
## Run all the chosen methods on all the scottish elections
all_cand_elections = []

election_results = []
for election_num, name in enumerate(file_names):
    
    sys.stdout.write('\r')
    sys.stdout.write(f'Election {election_num+1}/{len(file_names)}' + '    ' + name + '                                     ')
    sys.stdout.flush()
    
    # print(name)
    election_dict = {}
    election_dict['Election'] = name
    election = mwe(file_names[name])
    
    method_functions = [election.scot_stv, election.aq_stv, election.meek_stv, election.cham_cour, election.cham_cour, election.greedy_cham_cour, election.expanding_approvals, election.cpo_stv]
    
    unelected_cands = list(range(1, election.cand_num+1))
    
    for i in range(8):
        if which_methods[i]:
            if i == 4:
                election.model = 'PM'
            else:
                election.model = 'OM'
            election_dict[headers[i]] = method_functions[i]()[0]
            for cand in method_functions[i]()[0]:
                if cand in unelected_cands:
                    unelected_cands.remove(cand)
            
    if not unelected_cands:
        all_cand_elections.append(name)

    election_results.append(election_dict)

## Write results to csv file
with open('election_results_new.csv', 'w', newline='') as csvfile:
    writer = csv.DictWriter(csvfile, fieldnames=csv_headers)
    writer.writeheader()
    for election in election_results:
        writer.writerow(election)