From 3ba86fe7e7078d351b06d05ca88317ea233bd41c Mon Sep 17 00:00:00 2001 From: "Philippe M. Chiasson" Date: Fri, 30 Nov 2018 10:14:38 -0500 Subject: [PATCH] Add a skip_header option to bonobo.CsvWriter Fixes #302 --- bonobo/nodes/io/csv.py | 11 ++++++++++- 1 file changed, 10 insertions(+), 1 deletion(-) diff --git a/bonobo/nodes/io/csv.py b/bonobo/nodes/io/csv.py index c3a9b5d..49d8a3d 100644 --- a/bonobo/nodes/io/csv.py +++ b/bonobo/nodes/io/csv.py @@ -96,6 +96,15 @@ class CsvReader(FileReader, CsvHandler): @use_context class CsvWriter(FileWriter, CsvHandler): + + skip_header = Option( + bool, + default=False, + __doc__=""" + If true, the writer will not produce a file header + """, + ) + @Method( __doc__=""" 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: context.writer = self.writer_factory(file) - if fields: + if fields and not self.skip_header: context.writer(fields) context.lineno += 1