diff --git a/bonobo/io/csv.py b/bonobo/io/csv.py index 8640633..30ad3a4 100644 --- a/bonobo/io/csv.py +++ b/bonobo/io/csv.py @@ -2,6 +2,7 @@ import csv from bonobo.config import Option from bonobo.config.processors import ContextProcessor, contextual +from bonobo.constants import NOT_MODIFIED from bonobo.util.objects import ValueHolder from .file import FileHandler, FileReader, FileWriter @@ -54,7 +55,7 @@ class CsvReader(CsvHandler, FileReader): for row in reader: if len(row) != field_count: - raise ValueError('Got a line with %d fields, expecting %d.' % (len(row), field_count, )) + raise ValueError('Got a line with %d fields, expecting %d.' % (len(row), field_count,)) yield dict(zip(headers.value, row)) @@ -72,3 +73,6 @@ class CsvWriter(CsvHandler, FileWriter): writer.writerow(headers.value) writer.writerow(row[header] for header in headers.value) lineno.value += 1 + return NOT_MODIFIED + + diff --git a/bonobo/io/file.py b/bonobo/io/file.py index 8337490..3298fd9 100644 --- a/bonobo/io/file.py +++ b/bonobo/io/file.py @@ -1,6 +1,7 @@ from bonobo.config import Option, Service from bonobo.config.configurables import Configurable from bonobo.config.processors import ContextProcessor, contextual +from bonobo.constants import NOT_MODIFIED from bonobo.util.objects import ValueHolder __all__ = [ @@ -94,6 +95,7 @@ class FileWriter(Writer): """ self._write_line(file, (self.eol if lineno.value else '') + row) lineno.value += 1 + return NOT_MODIFIED def _write_line(self, file, line): return file.write(line)