def main(): writer = UnicodeWriter(sys.stdout) writer.writerow(["gemeente", "stembureau", "postcode", "stemmen"]) for file_path in get_file_paths(): rows = parse_eml_file(file_path) writer.writerows(rows) return 0
class SqlExportFileWriter(object): """Writes rows to a CSV file, optionally filtering on a predicate.""" def __init__(self, dest, predicate=None, use_unicode=False): if use_unicode: self._writer = UnicodeWriter(dest, delimiter=DELIMITER) else: self._writer = csv.writer(dest, delimiter=DELIMITER) self._predicate = predicate def write_header(self, keys): self._writer.writerow(keys) def write_rows(self, results): if self._predicate: results = [result for result in results if self._predicate(result)] if results: self._writer.writerows(results)