|
11 | 11 |
|
12 | 12 | import xlrd
|
13 | 13 | from xlwt import XFStyle, Workbook
|
14 |
| -from pyexcel_io.sheet import SheetWriter |
| 14 | +from pyexcel_io import constants |
| 15 | +from pyexcel_io.plugin_api.abstract_sheet import ISheetWriter |
15 | 16 | from pyexcel_io.plugin_api.abstract_writer import IWriter
|
16 | 17 |
|
17 | 18 | DEFAULT_DATE_FORMAT = "DD/MM/YY"
|
|
20 | 21 | EMPTY_SHEET_NOT_ALLOWED = "xlwt does not support a book without any sheets"
|
21 | 22 |
|
22 | 23 |
|
23 |
| -class XLSheetWriter(SheetWriter): |
| 24 | +class XLSheetWriter(ISheetWriter): |
24 | 25 | """
|
25 | 26 | xls sheet writer
|
26 | 27 | """
|
27 | 28 |
|
28 |
| - def set_sheet_name(self, name): |
29 |
| - """Create a sheet""" |
30 |
| - self._native_sheet = self._native_book.add_sheet(name) |
| 29 | + def __init__(self, xls_book, xls_sheet, sheet_name, **keywords): |
| 30 | + if sheet_name is None: |
| 31 | + sheet_name = constants.DEFAULT_SHEET_NAME |
| 32 | + self._xls_book = xls_book |
| 33 | + self._xls_sheet = xls_sheet |
| 34 | + self._keywords = keywords |
| 35 | + self._xls_sheet = self._xls_book.add_sheet(sheet_name) |
31 | 36 | self.current_row = 0
|
32 | 37 |
|
33 | 38 | def write_row(self, array):
|
@@ -60,11 +65,14 @@ def write_row(self, array):
|
60 | 65 | style = XFStyle()
|
61 | 66 | style.num_format_str = DEFAULT_TIME_FORMAT
|
62 | 67 | if style:
|
63 |
| - self._native_sheet.write(self.current_row, i, value, style) |
| 68 | + self._xls_sheet.write(self.current_row, i, value, style) |
64 | 69 | else:
|
65 |
| - self._native_sheet.write(self.current_row, i, value) |
| 70 | + self._xls_sheet.write(self.current_row, i, value) |
66 | 71 | self.current_row += 1
|
67 | 72 |
|
| 73 | + def close(self): |
| 74 | + pass |
| 75 | + |
68 | 76 |
|
69 | 77 | class XLSWriter(IWriter):
|
70 | 78 | """
|
|
0 commit comments