some code rearrangement without functionality changes

This commit is contained in:
Pablo Hoffman 2009-09-03 14:29:20 -03:00
parent 0e7b2a6da5
commit 0d493615ab
1 changed files with 11 additions and 11 deletions

View File

@ -105,24 +105,24 @@ class CsvItemExporter(BaseItemExporter):
self._configure(kwargs, dont_fail=True)
self.include_headers_line = include_headers_line
self.csv_writer = csv.writer(file, **kwargs)
self._header_written = False
self._headers_written = False
def export_item(self, item):
if not self._header_written:
self._write_header(item)
self._write_headers_and_set_fields_to_export(item)
fields = self._get_serialized_fields(item, default_value='', \
include_empty=True)
values = [x[1] for x in fields]
self.csv_writer.writerow(values)
def _write_header(self, item):
self._header_written = True
if self.include_headers_line:
if not self.fields_to_export:
self.fields_to_export = item.fields.keys()
self.csv_writer.writerow(self.fields_to_export)
def _write_headers_and_set_fields_to_export(self, item):
if self._headers_written:
return
self._headers_written = True
if not self.include_headers_line:
return
if not self.fields_to_export:
self.fields_to_export = item.fields.keys()
self.csv_writer.writerow(self.fields_to_export)
class PickleItemExporter(BaseItemExporter):