def write_rows(xf, worksheet, string_table, style_table=None):
"""Write worksheet data to xml."""
# Ensure a blank cell exists if it has a style
for styleCoord in iterkeys(worksheet._styles):
if isinstance(styleCoord, str) and COORD_RE.search(styleCoord):
worksheet.cell(styleCoord)
# create rows of cells
cells_by_row = {}
for cell in itervalues(worksheet._cells):
cells_by_row.setdefault(cell.row, []).append(cell)
with xf.element("sheetData"):
for row_idx in sorted(cells_by_row):
# row meta data
row_dimension = worksheet.row_dimensions[row_idx]
row_dimension.style = worksheet._styles.get(row_idx)
attrs = {
'r': '%d' % row_idx,
'spans': '1:%d' % worksheet.max_column
}
attrs.update(dict(row_dimension))
with xf.element("row", attrs):
row_cells = cells_by_row[row_idx]
for cell in sorted(row_cells, key=row_sort):
write_cell(xf, worksheet, cell, string_table)
def get_rows_to_write(worksheet):
"""Return all rows, and any cells that they contain"""
# Ensure a blank cell exists if it has a style
for styleCoord in iterkeys(worksheet._styles):
if isinstance(styleCoord, str) and COORD_RE.search(styleCoord):
worksheet.cell(styleCoord)
# create rows of cells
cells_by_row = {}
for cell in itervalues(worksheet._cells):
cells_by_row.setdefault(cell.row, []).append(cell)
# make sure rows that only have a height set are returned
for row_idx in worksheet.row_dimensions:
if row_idx not in cells_by_row:
cells_by_row[row_idx] = []
return cells_by_row
def compare_complex(a, b):
if isinstance(a, list):
if not isinstance(b, list):
return False
else:
for i, v in enumerate(a):
if not compare_complex(v, b[i]):
return False
elif isinstance(a, dict):
if not isinstance(b, dict):
return False
else:
for k in iterkeys(a):
if isinstance(a[k], (list, dict)):
if not compare_complex(a[k], b[k]):
return False
elif a[k] != b[k]:
return False
elif isinstance(a, HashableObject) or isinstance(b, HashableObject):
if repr(a) != repr(b):
return False
elif a != b:
return False
return True
def compare_complex(a, b):
if isinstance(a, list):
if not isinstance(b, list):
return False
else:
for i, v in enumerate(a):
if not compare_complex(v, b[i]):
return False
elif isinstance(a, dict):
if not isinstance(b, dict):
return False
else:
for k in iterkeys(a):
if isinstance(a[k], (list, dict)):
if not compare_complex(a[k], b[k]):
return False
elif a[k] != b[k]:
return False
elif isinstance(a, HashableObject) or isinstance(b, HashableObject):
if repr(a) != repr(b):
return False
elif a != b:
return False
return True
def test_iterkeys(dictionary):
from openpyxl.compat import iterkeys
d = dictionary
assert set(iterkeys(d)) == set(["1", "a", 3])
def test_iterkeys(dictionary):
from openpyxl.compat import iterkeys
d = dictionary
assert set(iterkeys(d)) == set(['1', 'a', 3])
