Skip to content

Commit 9089cca

Browse files
committed
bug fix on multiplesheet in csv memory stream where all custom fields will be encoded in the same way
1 parent 9481067 commit 9089cca

File tree

1 file changed

+6
-6
lines changed

1 file changed

+6
-6
lines changed

pyexcel_io/fileformat/_csv.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -216,10 +216,6 @@ def __init__(self, filename, name,
216216
encoding=encoding,
217217
single_sheet_in_book=single_sheet_in_book,
218218
sheet_index=sheet_index, **keywords)
219-
if not single_sheet_in_book:
220-
self.native_book.write(DEFAULT_CSV_STREAM_FILE_FORMATTER % (
221-
self.sheet_name,
222-
self.line_terminator))
223219

224220
def set_sheet_name(self, name):
225221
if PY2:
@@ -229,6 +225,10 @@ def set_sheet_name(self, name):
229225
else:
230226
self.f = self.native_book
231227
self.writer = csv.writer(self.f, **self.keywords)
228+
if not self.single_sheet_in_book:
229+
self.writer.writerow([DEFAULT_CSV_STREAM_FILE_FORMATTER % (
230+
self.sheet_name,
231+
"")])
232232

233233
def close(self):
234234
if self.single_sheet_in_book:
@@ -237,8 +237,8 @@ def close(self):
237237
# self.f.close()
238238
pass
239239
else:
240-
self.f.write(
241-
DEFAULT_SHEET_SEPARATOR_FORMATTER % self.line_terminator)
240+
self.writer.writerow(
241+
[DEFAULT_SHEET_SEPARATOR_FORMATTER % ""])
242242

243243

244244
class CSVBookReader(BookReader):

0 commit comments

Comments
 (0)