initial commit

1 files changed, 84 insertions, 0 deletions

diff --git a/datamaps/core/row.py b/datamaps/core/row.py
new file mode 100644
index 0000000..470b753
--- /dev/null
+++ b/datamaps/core/row.py
@@ -0,0 +1,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)