tbl: add _Row._index

Steve Canny · Steve Canny · commit f9197850641f · 2014-11-28T21:22:23.000-05:00
diff --git a/docx/oxml/table.py b/docx/oxml/table.py
@@ -24,6 +24,14 @@ class CT_Row(BaseOxmlElement):
     """
     tc = ZeroOrMore('w:tc')
 
+    @property
+    def tr_idx(self):
+        """
+        The index of this ``<w:tr>`` element within its parent ``<w:tbl>``
+        element.
+        """
+        return self.getparent().tr_lst.index(self)
+
     def _new_tc(self):
         return CT_Tc.new()
 
diff --git a/docx/table.py b/docx/table.py
@@ -332,7 +332,7 @@ def _index(self):
         """
         Index of this row in its table, starting from zero.
         """
-        raise NotImplementedError
+        return self._tr.tr_idx
 
 
 class _RowCells(Parented):
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -454,6 +454,10 @@ def it_provides_access_to_the_table_it_belongs_to(self, table_fixture):
         row, table_ = table_fixture
         assert row.table is table_
 
+    def it_knows_its_index_in_table_to_help(self, idx_fixture):
+        row, expected_idx = idx_fixture
+        assert row._index == expected_idx
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -464,6 +468,13 @@ def cells_fixture(self, _index_, table_prop_, table_):
         table_.row_cells.return_value = list(expected_cells)
         return row, row_idx, expected_cells
 
+    @pytest.fixture
+    def idx_fixture(self):
+        tbl = element('w:tbl/(w:tr,w:tr,w:tr)')
+        tr, expected_idx = tbl[1], 1
+        row = _Row(tr, None)
+        return row, expected_idx
+
     @pytest.fixture
     def table_fixture(self, parent_, table_):
         row = _Row(None, parent_)
