143 lines
4.0 KiB
PHP
143 lines
4.0 KiB
PHP
<?php
|
|
|
|
$options = getopt("", ["candidates-files:", "results-file:"]);
|
|
|
|
if (isset($options['candidates-files'])) {
|
|
$candidates_files = $options['candidates-files'];
|
|
} else {
|
|
error_log("Error: Missing required option '--candidates-files'.");
|
|
exit(1);
|
|
}
|
|
|
|
if (isset($options['results-file'])) {
|
|
$results_file = $options['results-file'];
|
|
$results_string = file_get_contents($results_file);
|
|
|
|
if ($results_string !== FALSE) {
|
|
$results = json_decode($results_string, true);
|
|
} else {
|
|
error_log("Error opening results.json.");
|
|
exit(1);
|
|
}
|
|
} else {
|
|
error_log("Error: Missing required option '--results-file'.");
|
|
exit(1);
|
|
}
|
|
|
|
function trim_sluggify($input) {
|
|
return strtolower(str_replace(' ', '-', trim($input)));
|
|
}
|
|
|
|
function match_words($words, $list) {
|
|
/* Match database names to VEC names */
|
|
$max_score = 0;
|
|
$best_match = "no match";
|
|
foreach ($list as $possible_match) {
|
|
$aa = preg_split("/[^a-z]/", strtolower($words));
|
|
$bb = preg_split("/[^a-z]/", strtolower($possible_match));
|
|
|
|
$score_sum = 0;
|
|
foreach ($aa as $a) {
|
|
foreach ($bb as $b) {
|
|
similar_text($a, $b, $score);
|
|
if ($score > 70) $score_sum += $score;
|
|
else $score_sum -= 1;
|
|
}
|
|
}
|
|
|
|
if ($score_sum > $max_score) {
|
|
$max_score = $score_sum;
|
|
$best_match = $possible_match;
|
|
}
|
|
}
|
|
return array($max_score, $best_match);
|
|
}
|
|
|
|
$candidates_files = explode(" ", $candidates_files);
|
|
|
|
/* Generate dictionary of candidates and LGAs */
|
|
$candidate_data = [];
|
|
foreach ($candidates_files as $file) {
|
|
$config_file = dirname($file)."/config.json";
|
|
$config_string = file_get_contents($config_file);
|
|
|
|
if ($config_string !== FALSE) {
|
|
$config = json_decode($config_string, true);
|
|
} else {
|
|
error_log("Error opening config.json.");
|
|
exit(1);
|
|
}
|
|
|
|
$candidate_data[$config['councilName']]['_filename'] = $file;
|
|
|
|
if (($handle = fopen($file, "r")) !== FALSE) {
|
|
$headers = fgetcsv($handle);
|
|
while (($data = fgetcsv($handle)) !== FALSE) {
|
|
$candidate = [];
|
|
foreach ($headers as $key => $value) {
|
|
$candidate[$value] = $data[$key];
|
|
}
|
|
$name_slug = trim_sluggify($candidate['Candidate Name']);
|
|
$candidate_data[$config['councilName']][$name_slug] = $candidate;
|
|
}
|
|
}
|
|
}
|
|
|
|
$vec_lga_names = [];
|
|
foreach ($results as $lga => $data) {
|
|
$vec_lga_names[] = $lga;
|
|
}
|
|
|
|
function was_elected($candidate, $vec_wards) {
|
|
foreach ($vec_wards as $vec_candidates) {
|
|
list($score, $match) = match_words($candidate, $vec_candidates);
|
|
if ($score > 180) return true;
|
|
}
|
|
return false;
|
|
}
|
|
|
|
$header = ["Ward", "Candidate Name", "Elected"];
|
|
|
|
foreach ($candidate_data as $lga => $db_candidates) {
|
|
/* Find LGA in results dict */
|
|
list($score, $vec_lga_name) = match_words($lga, $vec_lga_names);
|
|
$vec_wards = $results[$vec_lga_name];
|
|
|
|
$elected = [];
|
|
/* Go through database candidates and build list of elected candidates */
|
|
foreach ($db_candidates as $key => $value) {
|
|
if ($key === '_filename') {
|
|
$output_file = dirname($value)."/candidates-elected.csv";
|
|
continue;
|
|
}
|
|
if (was_elected($value['Candidate Name'], $vec_wards)) {
|
|
$elected[] = $value;
|
|
}
|
|
}
|
|
|
|
/* Don't create file if none were elected. */
|
|
if (count($elected) === 0) continue;
|
|
|
|
if (($handle = fopen($output_file, "w")) === FALSE) {
|
|
error_log('Error opening output file');
|
|
exit(1);
|
|
}
|
|
|
|
if (fputcsv($handle, $header) === FALSE) {
|
|
error_log('Error writing headers to output file');
|
|
exit(3);
|
|
}
|
|
|
|
foreach ($elected as $candidate) {
|
|
$line = array($candidate['Ward'], $candidate['Candidate Name'], "y");
|
|
if (fputcsv($handle, $line) === FALSE) {
|
|
error_log('Error writing candidate to output file');
|
|
exit(3);
|
|
}
|
|
}
|
|
|
|
fclose($handle);
|
|
}
|
|
|
|
exit(0);
|