Пример #1
0
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
Пример #2
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)