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
|
import string
import datetime
from ..process.cell import Cell
from typing import Iterable
from itertools import chain
class Row:
"""
A Row object is populated with an iterable (list or other sequence), bound
to an openpyxl worksheet. It is used to populate a row of cells in an output
Excel file with the values from the iterable.
The ``anchor_column`` and ``anchor_row`` parameters represent the coordinates of
a cell which form the *leftmost* cell of the row, i.e. to set the row of data
to start at the very top left of a sheet, you'd create the ``Row()`` object this::
r = Row(1, 1, interable)
r.bind(ws)
"""
def __init__(self, anchor_column: int, anchor_row: int, seq: Iterable):
if isinstance(anchor_column, str):
if len(anchor_column) == 1:
enumerated_alphabet = list(enumerate(string.ascii_uppercase, start=1))
col_letter = [x for x in enumerated_alphabet if x[1] == anchor_column][0]
self._anchor_column = col_letter[0]
self._anchor_row = anchor_row
self._cell_map = []
elif len(anchor_column) == 2:
enumerated_alphabet = list(enumerate(list(chain(
string.ascii_uppercase, ["{}{}".format(x[0], x[1]) for x in list(zip(['A'] * 26, string.ascii_uppercase))])), start=1))
col_letter = [x for x in enumerated_alphabet if x[1] == anchor_column][0]
self._anchor_column = col_letter[0]
self._anchor_row = anchor_row
self._cell_map = []
else:
raise ValueError("You can only have a column up to AZ")
else:
self._anchor_column = anchor_column
self._anchor_row = anchor_row
self._cell_map = []
self._seq = seq
def _basic_bind(self, ws):
for x in list(enumerate(self._seq, start=self._anchor_column)):
self._ws.cell(row=self._anchor_row, column=x[0], value=x[1])
def _cell_bind(self, ws):
self._cell_map = []
for x in list(enumerate(self._seq, start=self._anchor_column)):
self._cell_map.append(
Cell(
cell_key="",
cell_value=x[1],
cell_reference=f"{self._anchor_column}{self._anchor_row}",
template_sheet=ws,
bg_colour=None,
fg_colour=None,
number_format=None,
verification_list=None,
r_idx=self._anchor_row,
c_idx=x[0]
)
)
for c in self._cell_map:
if not isinstance(c.cell_value, datetime.date) and not None:
self._ws.cell(row=c.r_idx, column=c.c_idx, value=c.cell_value).number_format = '0'
else:
self._ws.cell(row=c.r_idx, column=c.c_idx, value=c.cell_value)
def bind(self, worksheet):
"""Bind the Row to a particular worksheetl, which effectively does the
printing of data into cells. Must be done prior to saving the workbook.
"""
self._ws = worksheet
# self._basic_bind(self._ws)
self._cell_bind(self._ws)
|
