paulin/bonobo
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge branch 'issue/302/CsvWriter-skip_header' of https://github.com/gozer/bonobo into 302_csv_skip_headers_option

Browse Source
This commit is contained in:
Romain Dorgueil
2019-06-01 09:24:06 +02:00
parent 751f4bcbfd 3ba86fe7e7
commit 94c5f1a45a
1 changed files with 10 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
bonobo/nodes/io/csv.py
View File

@ -96,6 +96,15 @@ class CsvReader(FileReader, CsvHandler):
@use_context @use_context
class CsvWriter(FileWriter, CsvHandler): class CsvWriter(FileWriter, CsvHandler):
skip_header = Option(
bool,
default=False,
__doc__="""
If true, the writer will not produce a file header
""",
)
@Method( @Method(
__doc__=""" __doc__="""
Builds the CSV writer, a.k.a an object we can pass a field collection to be written as one line in the Builds the CSV writer, a.k.a an object we can pass a field collection to be written as one line in the
@ -114,7 +123,7 @@ class CsvWriter(FileWriter, CsvHandler):
if not context.lineno: if not context.lineno:
context.writer = self.writer_factory(file) context.writer = self.writer_factory(file)
if fields: if fields and not self.skip_header:
context.writer(fields) context.writer(fields)
context.lineno += 1 context.lineno += 1