150 lines
4.2 KiB
PHP
150 lines
4.2 KiB
PHP
<?php
|
|
$options = getopt("", ["input:", "output:", "media:"]);
|
|
|
|
if (isset($options['input'])) {
|
|
$inputFile = $options['input'];
|
|
} else {
|
|
error_log("Error: Missing required option '--input'.");
|
|
exit(1);
|
|
}
|
|
|
|
if (isset($options['output'])) {
|
|
$outputFile = $options['output'];
|
|
} else {
|
|
error_log("Error: Missing required option '--output'.");
|
|
exit(1);
|
|
}
|
|
|
|
if (isset($options['media'])) {
|
|
$mediaFolder = $options['media'];
|
|
} else {
|
|
error_log("Error: Missing required option '--media'.");
|
|
exit(1);
|
|
}
|
|
|
|
$mediaFiles = scandir($mediaFolder);
|
|
if ($mediaFiles === FALSE) {
|
|
error_log("Failed to list files in media folder");
|
|
exit(1);
|
|
}
|
|
|
|
$candidates = [];
|
|
if (($handle = fopen($inputFile, "r")) !== FALSE) {
|
|
$currentWard = null;
|
|
$currentLine = 0;
|
|
while (($data = fgetcsv($handle)) !== FALSE) {
|
|
$currentLine++;
|
|
//echo var_dump($data);
|
|
if ($data[0] == "Ward") {
|
|
// CSV contains ward names in uppercase, convert them to a more readable form
|
|
$currentWard = ucwords(strtolower($data[1]));
|
|
|
|
// Handle some special cases where the above logic doesn't match the expected names
|
|
// Note that we cannot just convert every letter after a '-' character to uppercase
|
|
// because there are some ward names like "Bulleke-bek"
|
|
if ($currentWard == "Warrk-warrk") {
|
|
$currentWard = "Warrk-Warrk";
|
|
}
|
|
if ($currentWard == "Djirri-djirri") {
|
|
$currentWard = "Djirri-Djirri";
|
|
}
|
|
if ($currentWard == "Coastal-promontory") {
|
|
$currentWard = "Coastal-Promontory";
|
|
}
|
|
}
|
|
if ($data[0] == "Candidate" || $data[0] == "") {
|
|
if ($currentWard == null) {
|
|
error_log("No ward found, skipping data on line " . $currentLine);
|
|
continue;
|
|
}
|
|
|
|
$candidateName = trim($data[1]);
|
|
|
|
if ($candidateName == " example name") {
|
|
error_log("Skipping line ". $currentLine);
|
|
continue;
|
|
}
|
|
|
|
print("Adding candidate " . $candidateName . " to ". $currentWard . "\n");
|
|
|
|
$name_split = explode(" ", $data[1]);
|
|
|
|
$name_patterns = [
|
|
implode(".*", $name_split),
|
|
implode(".*", array_reverse($name_split)),
|
|
];
|
|
|
|
$regex_groups = array_map(function($x) { return "(?:.*" . $x . ".*)"; }, $name_patterns);
|
|
|
|
$regex_pattern = "/" . implode("|", $regex_groups) . "/i";
|
|
|
|
$picture = "";
|
|
foreach ($mediaFiles as $mediaFile) {
|
|
if (preg_match($regex_pattern, $mediaFile)) {
|
|
$picture = $mediaFile;
|
|
break;
|
|
}
|
|
}
|
|
if ($picture === "") {
|
|
print("Failed to identify picture for " . $candidateName . "\n");
|
|
}
|
|
|
|
$rating = $data[2];
|
|
if ($rating == "score" || $rating == "") {
|
|
$rating = 0;
|
|
}
|
|
|
|
array_push(
|
|
$candidates,
|
|
[
|
|
"Ward" => $currentWard,
|
|
"Candidate Name" => $candidateName,
|
|
"Rating" => $rating,
|
|
"Picture" => $picture
|
|
]
|
|
);
|
|
}
|
|
}
|
|
fclose($handle);
|
|
} else {
|
|
error_log('Error opening input file');
|
|
exit(1);
|
|
}
|
|
|
|
if (empty($candidates)) {
|
|
error_log("Failed to find any candidates");
|
|
exit(2);
|
|
}
|
|
|
|
if (($handle = fopen($outputFile, "w")) !== FALSE) {
|
|
$headers = array(
|
|
"Ward",
|
|
"Candidate Name",
|
|
"Rating",
|
|
"Picture"
|
|
);
|
|
if (fputcsv($handle, $headers) === FALSE) {
|
|
error_log('Error writing headers to output file');
|
|
exit(3);
|
|
}
|
|
foreach ($candidates as $candidate) {
|
|
$fields = array(
|
|
$candidate["Ward"],
|
|
$candidate["Candidate Name"],
|
|
$candidate["Rating"],
|
|
$candidate["Picture"]
|
|
);
|
|
if (fputcsv($handle, $fields) === FALSE) {
|
|
error_log('Error writing candidate to output file');
|
|
exit(3);
|
|
}
|
|
}
|
|
} else {
|
|
error_log('Error opening output file');
|
|
exit(1);
|
|
}
|
|
|
|
print("Data written to " . $outputFile);
|
|
|
|
exit(0);
|