tbl: add Table.row_cells()

Steve Canny · Steve Canny · commit ce3ecf788aa7 · 2014-11-28T21:22:23.000-05:00
diff --git a/docx/table.py b/docx/table.py
@@ -71,7 +71,10 @@ def row_cells(self, row_idx):
         """
         Sequence of cells in the row at *row_idx* in this table.
         """
-        raise NotImplementedError
+        column_count = self._column_count
+        start = row_idx * column_count
+        end = start + column_count
+        return self._cells[start:end]
 
     @lazyproperty
     def rows(self):
@@ -103,6 +106,22 @@ def table(self):
         """
         return self
 
+    @property
+    def _cells(self):
+        """
+        A sequence of |_Cell| objects, one for each cell of the layout grid.
+        If the table contains a span, one or more |_Cell| object references
+        are repeated.
+        """
+        raise NotImplementedError
+
+    @property
+    def _column_count(self):
+        """
+        The number of grid columns in this table.
+        """
+        raise NotImplementedError
+
     @property
     def _tblPr(self):
         return self._tbl.tblPr
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -39,6 +39,11 @@ def it_provides_access_to_a_cell_by_row_and_col_indices(self, table):
                 tc = tr.tc_lst[col_idx]
                 assert tc is cell._tc
 
+    def it_provides_access_to_the_cells_in_a_row(self, row_cells_fixture):
+        table, row_idx, expected_cells = row_cells_fixture
+        row_cells = table.row_cells(row_idx)
+        assert row_cells == expected_cells
+
     def it_can_add_a_row(self, add_row_fixture):
         table, expected_xml = add_row_fixture
         row = table.add_row()
@@ -120,6 +125,15 @@ def autofit_set_fixture(self, request):
         expected_xml = xml(expected_tbl_cxml)
         return table, new_value, expected_xml
 
+    @pytest.fixture
+    def row_cells_fixture(self, _cells_, _column_count_):
+        table = Table(None, None)
+        _cells_.return_value = [0, 1, 2, 3, 4, 5, 6, 7, 8]
+        _column_count_.return_value = 3
+        row_idx = 1
+        expected_cells = [3, 4, 5]
+        return table, row_idx, expected_cells
+
     @pytest.fixture
     def table_fixture(self):
         table = Table(None, None)
@@ -152,6 +166,14 @@ def table_style_set_fixture(self, request):
 
     # fixture components ---------------------------------------------
 
+    @pytest.fixture
+    def _cells_(self, request):
+        return property_mock(request, Table, '_cells')
+
+    @pytest.fixture
+    def _column_count_(self, request):
+        return property_mock(request, Table, '_column_count')
+
     @pytest.fixture
     def table(self):
         tbl = _tbl_bldr(rows=2, cols=2).element
