def write_data(self, archive, as_template=False):
"""Write the various xml files into the zip archive."""
# cleanup all worksheets
archive.writestr(ARC_ROOT_RELS, write_root_rels(self.workbook))
archive.writestr(ARC_WORKBOOK_RELS, write_workbook_rels(self.workbook))
archive.writestr(ARC_APP, write_properties_app(self.workbook))
archive.writestr(ARC_CORE, write_properties(self.workbook.properties))
if self.workbook.loaded_theme:
archive.writestr(ARC_THEME, self.workbook.loaded_theme)
else:
archive.writestr(ARC_THEME, write_theme())
archive.writestr(ARC_WORKBOOK, write_workbook(self.workbook))
if self.workbook.vba_archive:
vba_archive = self.workbook.vba_archive
for name in vba_archive.namelist():
for s in ARC_VBA:
if match(s, name):
archive.writestr(name, vba_archive.read(name))
break
self._write_charts(archive)
self._write_images(archive)
self._write_worksheets(archive)
self._write_chartsheets(archive)
self._write_string_table(archive)
self._write_external_links(archive)
archive.writestr(ARC_STYLE, self.style_writer.write_table())
manifest = write_content_types(self.workbook, as_template=as_template)
archive.writestr(ARC_CONTENT_TYPES, tostring(manifest.to_tree()))
def test_write_content_types(datadir):
datadir.join('reader').chdir()
wb = load_workbook('vba-test.xlsm', keep_vba=True)
manifest = write_content_types(wb)
datadir.chdir()
datadir.join('writer').chdir()
with open('Content_types_vba.xml') as expected:
diff = compare_xml(tostring(manifest.to_tree()), expected.read())
assert diff is None, diff
def test_write_content_types(datadir):
datadir.join('reader').chdir()
wb = load_workbook('vba-test.xlsm', keep_vba=True)
manifest = write_content_types(wb)
datadir.chdir()
datadir.join('writer').chdir()
with open('Content_types_vba.xml') as expected:
diff = compare_xml(tostring(manifest.to_tree()), expected.read())
assert diff is None, diff
def write_data(self, archive, as_template=False):
"""Write the various xml files into the zip archive."""
# cleanup all worksheets
archive.writestr(ARC_ROOT_RELS, write_root_rels(self.workbook))
props = ExtendedProperties()
archive.writestr(ARC_APP, tostring(props.to_tree()))
archive.writestr(ARC_CORE,
tostring(self.workbook.properties.to_tree()))
if self.workbook.loaded_theme:
archive.writestr(ARC_THEME, self.workbook.loaded_theme)
else:
archive.writestr(ARC_THEME, write_theme())
self._write_charts(archive)
self._write_images(archive)
self._write_worksheets(archive)
self._write_chartsheets(archive)
self._write_string_table(archive)
self._write_external_links(archive)
stylesheet = write_stylesheet(self.workbook)
archive.writestr(ARC_STYLE, tostring(stylesheet))
archive.writestr(ARC_WORKBOOK, write_workbook(self.workbook))
archive.writestr(ARC_WORKBOOK_RELS, write_workbook_rels(self.workbook))
if self.workbook.vba_archive:
vba_archive = self.workbook.vba_archive
for name in set(vba_archive.namelist()) - self.vba_modified:
for s in ARC_VBA:
if match(s, name):
archive.writestr(name, vba_archive.read(name))
break
exts = []
for n in archive.namelist():
if "media" in n:
exts.append(n)
manifest = write_content_types(self.workbook,
as_template=as_template,
exts=exts)
archive.writestr(ARC_CONTENT_TYPES, tostring(manifest.to_tree()))
def write_data(self, archive, as_template=False):
"""Write the various xml files into the zip archive."""
# cleanup all worksheets
archive.writestr(ARC_ROOT_RELS, write_root_rels(self.workbook))
props = ExtendedProperties()
archive.writestr(ARC_APP, tostring(props.to_tree()))
archive.writestr(ARC_CORE, tostring(self.workbook.properties.to_tree()))
if self.workbook.loaded_theme:
archive.writestr(ARC_THEME, self.workbook.loaded_theme)
else:
archive.writestr(ARC_THEME, write_theme())
self._write_charts(archive)
self._write_images(archive)
self._write_worksheets(archive)
self._write_chartsheets(archive)
self._write_string_table(archive)
self._write_external_links(archive)
stylesheet = write_stylesheet(self.workbook)
archive.writestr(ARC_STYLE, tostring(stylesheet))
archive.writestr(ARC_WORKBOOK, write_workbook(self.workbook))
archive.writestr(ARC_WORKBOOK_RELS, write_workbook_rels(self.workbook))
if self.workbook.vba_archive:
vba_archive = self.workbook.vba_archive
for name in set(vba_archive.namelist()) - self.vba_modified:
for s in ARC_VBA:
if match(s, name):
archive.writestr(name, vba_archive.read(name))
break
exts = []
for n in archive.namelist():
if "media" in n:
exts.append(n)
manifest = write_content_types(self.workbook, as_template=as_template, exts=exts)
archive.writestr(ARC_CONTENT_TYPES, tostring(manifest.to_tree()))
