/usr/lib/python3/dist-packages/openpyxl/worksheet/read_only.py is in python3-openpyxl 2.3.0-1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126 127 128 129 130 131 132 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174 175 176 177 178 179 180 181 182 183 184 185 186 187 188 189 190 191 192 193 194 195 196 197 198 199 200 201 202 203 204 205 206 207 208 209 210 211 212 213 214 215 216 217 218 219 220 221 222 223 224 225 226 227 228 229 230 231 232 233 234 235 236 237 238 239 240 241 242 243 244 245 246 247 248 249 250 251 252 | from __future__ import absolute_import
# Copyright (c) 2010-2015 openpyxl
""" Iterators-based worksheet reader
*Still very raw*
"""
# compatibility
from openpyxl.compat import range
# package
from openpyxl.xml.functions import iterparse
from openpyxl.xml.functions import safe_iterator
from openpyxl.xml.constants import SHEET_MAIN_NS
from openpyxl.worksheet import Worksheet
from openpyxl.utils import (
ABSOLUTE_RE,
column_index_from_string,
get_column_letter,
coordinate_to_tuple,
)
from openpyxl.cell.read_only import ReadOnlyCell, EMPTY_CELL
def read_dimension(source):
if hasattr(source, "encode"):
return
min_row = min_col = max_row = max_col = None
DIMENSION_TAG = '{%s}dimension' % SHEET_MAIN_NS
DATA_TAG = '{%s}sheetData' % SHEET_MAIN_NS
it = iterparse(source, tag=[DIMENSION_TAG, DATA_TAG])
for _event, element in it:
if element.tag == DIMENSION_TAG:
dim = element.get("ref")
m = ABSOLUTE_RE.match(dim.upper())
if m is None:
return
min_col, min_row, sep, max_col, max_row = m.groups()
min_row = int(min_row)
if max_col is None or max_row is None:
max_col = min_col
max_row = min_row
else:
max_row = int(max_row)
return (
column_index_from_string(min_col),
min_row,
column_index_from_string(max_col),
max_row
)
elif element.tag == DATA_TAG:
# Dimensions missing
break
element.clear()
ROW_TAG = '{%s}row' % SHEET_MAIN_NS
CELL_TAG = '{%s}c' % SHEET_MAIN_NS
VALUE_TAG = '{%s}v' % SHEET_MAIN_NS
FORMULA_TAG = '{%s}f' % SHEET_MAIN_NS
DIMENSION_TAG = '{%s}dimension' % SHEET_MAIN_NS
CELL_TAGS = (CELL_TAG, VALUE_TAG, FORMULA_TAG)
class ReadOnlyWorksheet(Worksheet):
_xml = None
_min_column = 1
_min_row = 1
_max_column = _max_row = None
def __init__(self, parent_workbook, title, worksheet_path,
xml_source, shared_strings):
Worksheet.__init__(self, parent_workbook, title)
self.worksheet_path = worksheet_path
self.shared_strings = shared_strings
self.base_date = parent_workbook.excel_base_date
self.xml_source = xml_source
dimensions = read_dimension(self.xml_source)
if dimensions is not None:
self.min_column, self.min_row, self.max_column, self.max_row = dimensions
@property
def xml_source(self):
"""Parse xml source on demand, default to Excel archive"""
if self._xml is None:
return self.parent._archive.open(self.worksheet_path)
return self._xml
@xml_source.setter
def xml_source(self, value):
self._xml = value
def get_squared_range(self, min_col, min_row, max_col, max_row):
"""
The source worksheet file may have columns or rows missing.
Missing cells will be created.
"""
if max_col is not None:
empty_row = tuple(EMPTY_CELL for column in range(min_col, max_col + 1))
else:
empty_row = []
row_counter = min_row
p = iterparse(self.xml_source, tag=[ROW_TAG], remove_blank_text=True)
for _event, element in p:
if element.tag == ROW_TAG:
row_id = int(element.get("r"))
# got all the rows we need
if max_row is not None and row_id > max_row:
break
# some rows are missing
for row_counter in range(row_counter, row_id):
row_counter += 1
yield empty_row
# return cells from a row
if min_row <= row_id:
yield tuple(self._get_row(element, min_col, max_col))
row_counter += 1
if element.tag in CELL_TAGS:
# sub-elements of rows should be skipped as handled within a cell
continue
element.clear()
def _get_row(self, element, min_col=1, max_col=None):
"""Return cells from a particular row"""
col_counter = min_col
data_only = getattr(self.parent, 'data_only', False)
for cell in safe_iterator(element, CELL_TAG):
coordinate = cell.get('r')
row, column = coordinate_to_tuple(coordinate)
if max_col is not None and column > max_col:
break
if min_col <= column:
if col_counter < column:
for col_counter in range(max(col_counter, min_col), column):
# pad row with missing cells
yield EMPTY_CELL
data_type = cell.get('t', 'n')
style_id = int(cell.get('s', 0))
value = None
formula = cell.findtext(FORMULA_TAG)
if formula is not None and not data_only:
data_type = 'f'
value = "=%s" % formula
else:
value = cell.findtext(VALUE_TAG) or None
yield ReadOnlyCell(self, row, column,
value, data_type, style_id)
col_counter = column + 1
if max_col is not None:
for _ in range(col_counter, max_col+1):
yield EMPTY_CELL
def _get_cell(self, row, column):
"""Cells are returned by a generator which can be empty"""
cell = tuple(self.get_squared_range(column, row, column, row))[0]
if cell:
return cell[0]
return EMPTY_CELL
@property
def rows(self):
return self.iter_rows()
@property
def columns(self):
if self.max_column is None:
self.calculate_dimension()
return super(IterableWorksheet, self).columns
def calculate_dimension(self, force=False):
if not all([self.max_column, self.max_row]):
if force:
self._calculate_dimension()
else:
raise ValueError("Worksheet is unsized, use calculate_dimension(force=True)")
return '%s%d:%s%d' % (
get_column_letter(self.min_column), self.min_row,
get_column_letter(self.max_column), self.max_row
)
def _calculate_dimension(self):
"""
Loop through all the cells to get the size of a worksheet.
Do this only if it is explicitly requested.
"""
max_col = 0
for r in self.rows:
cell = r[-1]
max_col = max(max_col, cell.column)
self.max_row = cell.row
self.max_column = max_col
@property
def min_row(self):
return self._min_row
@min_row.setter
def min_row(self, value):
self._min_row = value
@property
def max_row(self):
return self._max_row
@max_row.setter
def max_row(self, value):
self._max_row = value
@property
def min_column(self):
return self._min_column
@min_column.setter
def min_column(self, value):
self._min_column = value
@property
def max_column(self):
return self._max_column
@max_column.setter
def max_column(self, value):
self._max_column = value
|