Skip to content

Commit fb82cf2

Browse files
committed
🔨 code refactoring
1 parent 24813b2 commit fb82cf2

File tree

2 files changed

+11
-23
lines changed

2 files changed

+11
-23
lines changed

pyexcel_ods/odsr.py

+4-10
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030
from odf.teletype import extractText
3131
from odf.namespaces import OFFICENS
3232
from odf.opendocument import load
33-
from pyexcel_io.plugin_api.abstract_sheet import ISheet
34-
from pyexcel_io.plugin_api.abstract_reader import IReader
33+
from pyexcel_io.plugin_api import ISheet
34+
from pyexcel_io.plugin_api import IReader, NamedContent
3535

3636

3737
class ODSSheet(ISheet):
@@ -107,13 +107,13 @@ def __init__(self, file_alike_object, _, **keywords):
107107
self._native_book = load(file_alike_object)
108108
self._keywords = keywords
109109
self.content_array = [
110-
NameObject(table.getAttribute("name"), table)
110+
NamedContent(table.getAttribute("name"), table)
111111
for table in self._native_book.spreadsheet.getElementsByType(Table)
112112
]
113113

114114
def read_sheet(self, sheet_index):
115115
"""read a sheet at a specified index"""
116-
table = self.content_array[sheet_index].sheet
116+
table = self.content_array[sheet_index].payload
117117
sheet = ODSSheet(table, **self._keywords)
118118
return sheet
119119

@@ -129,9 +129,3 @@ class ODSBookInContent(ODSBook):
129129
def __init__(self, file_content, file_type, **keywords):
130130
io = BytesIO(file_content)
131131
super().__init__(io, file_type, **keywords)
132-
133-
134-
class NameObject(object):
135-
def __init__(self, name, sheet):
136-
self.name = name
137-
self.sheet = sheet

pyexcel_ods/odsw.py

+7-13
Original file line numberDiff line numberDiff line change
@@ -7,27 +7,21 @@
77
:copyright: (c) 2014-2020 by Onni Software Ltd.
88
:license: New BSD License, see LICENSE for more details
99
"""
10-
import sys
11-
1210
import pyexcel_io.service as converter
1311
from odf.text import P
1412
from odf.table import Table, TableRow, TableCell
1513
from odf.namespaces import OFFICENS
1614
from odf.opendocument import OpenDocumentSpreadsheet
17-
from pyexcel_io.plugin_api.abstract_sheet import ISheetWriter
18-
from pyexcel_io.plugin_api.abstract_writer import IWriter
19-
20-
PY2 = sys.version_info[0] == 2
21-
22-
PY27_BELOW = PY2 and sys.version_info[1] < 7
15+
from pyexcel_io.plugin_api import ISheetWriter
16+
from pyexcel_io.plugin_api import IWriter
2317

2418

2519
class ODSSheetWriter(ISheetWriter):
2620
"""
2721
ODS sheet writer
2822
"""
2923

30-
def __init__(self, ods_book, ods_sheet, sheet_name, **keywords):
24+
def __init__(self, ods_book, sheet_name):
3125
self._native_book = ods_book
3226
self._native_sheet = Table(name=sheet_name)
3327

@@ -79,20 +73,20 @@ class ODSWriter(IWriter):
7973
8074
"""
8175

82-
def __init__(self, file_alike_object, file_type, **keywords):
83-
self._file_alike_object = file_alike_object
76+
def __init__(self, file_alike_object, file_type, **_):
77+
self.file_alike_object = file_alike_object
8478
self._native_book = OpenDocumentSpreadsheet()
8579

8680
def create_sheet(self, name):
8781
"""
8882
write a row into the file
8983
"""
90-
return ODSSheetWriter(self._native_book, None, name)
84+
return ODSSheetWriter(self._native_book, name)
9185

9286
def close(self):
9387
"""
9488
This call writes file
9589
9690
"""
97-
self._native_book.write(self._file_alike_object)
91+
self._native_book.write(self.file_alike_object)
9892
self._native_book = None

0 commit comments

Comments
 (0)