def get_xml(xml_node):
io = BytesIO()
if version_info[0] >= 3 and version_info[1] >= 2:
ElementTree(xml_node).write(io, encoding='UTF-8', xml_declaration=True)
ret = str(io.getvalue(), 'utf-8')
ret = ret.replace('utf-8', 'UTF-8', 1)
else:
ElementTree(xml_node).write(io, encoding='UTF-8')
ret = io.getvalue()
io.close()
return ret.replace('\n', '')
def get_xml(xml_node):
io = BytesIO()
if version_info[0] >= 3 and version_info[1] >= 2:
ElementTree(xml_node).write(io, encoding="UTF-8", xml_declaration=False)
ret = str(io.getvalue(), "utf-8")
ret = ret.replace("utf-8", "UTF-8", 1)
else:
ElementTree(xml_node).write(io, encoding="UTF-8")
ret = io.getvalue()
io.close()
return ret.replace("\n", "")
def _write_images(self, images, archive, image_id):
for img in images:
buf = BytesIO()
img.image.save(buf, format= 'PNG')
archive.writestr(PACKAGE_IMAGES + '/image%d.png' % image_id, buf.getvalue())
image_id += 1
return image_id
def save_virtual_workbook(workbook):
"""Return an in-memory workbook, suitable for a Django response."""
writer = ExcelWriter(workbook)
temp_buffer = BytesIO()
try:
archive = ZipFile(temp_buffer, "w", ZIP_DEFLATED)
writer.write_data(archive)
finally:
archive.close()
virtual_workbook = temp_buffer.getvalue()
temp_buffer.close()
return virtual_workbook
def save_virtual_workbook(workbook):
"""Return an in-memory workbook, suitable for a Django response."""
writer = ExcelWriter(workbook)
temp_buffer = BytesIO()
try:
archive = ZipFile(temp_buffer, 'w', ZIP_DEFLATED)
writer.write_data(archive)
finally:
archive.close()
virtual_workbook = temp_buffer.getvalue()
temp_buffer.close()
return virtual_workbook
def assert_equals_file_content(reference_file, fixture, filetype = 'xml'):
if os.path.isfile(fixture):
fixture_file = open(fixture)
try:
fixture_content = fixture_file.read()
finally:
fixture_file.close()
else:
fixture_content = fixture
expected_file = open(reference_file)
try:
expected_content = expected_file.read()
finally:
expected_file.close()
if filetype == 'xml':
fixture_content = fromstring(fixture_content)
pretty_indent(fixture_content)
temp = BytesIO()
ElementTree(fixture_content).write(temp)
fixture_content = temp.getvalue()
expected_content = fromstring(expected_content)
pretty_indent(expected_content)
temp = BytesIO()
ElementTree(expected_content).write(temp)
expected_content = temp.getvalue()
fixture_lines = unicode(fixture_content).split('\n')
expected_lines = unicode(expected_content).split('\n')
differences = list(difflib.unified_diff(expected_lines, fixture_lines))
if differences:
temp = BytesIO()
pprint(differences, stream = temp)
assert False, 'Differences found : %s' % temp.getvalue()
