tbl: add _Row.table

Steve Canny · Steve Canny · commit acf76f46640a · 2014-11-28T21:22:23.000-05:00
diff --git a/docx/table.py b/docx/table.py
@@ -93,6 +93,16 @@ def style(self):
     def style(self, value):
         self._tblPr.style = value
 
+    @property
+    def table(self):
+        """
+        Provide child objects with reference to the |Table| object they
+        belong to, without them having to know their direct parent is
+        a |Table| object. This is the terminus of a series of `parent._table`
+        calls from an arbitrary child through its ancestors.
+        """
+        raise NotImplementedError
+
     @property
     def _tblPr(self):
         return self._tbl.tblPr
@@ -315,7 +325,7 @@ def table(self):
         """
         Reference to the |Table| object this row belongs to.
         """
-        raise NotImplementedError
+        return self._parent.table
 
     @property
     def _index(self):
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -441,6 +441,10 @@ def it_provides_access_to_the_row_cells(self):
         cells = row.cells
         assert isinstance(cells, _RowCells)
 
+    def it_provides_access_to_the_table_it_belongs_to(self, table_fixture):
+        row, table_ = table_fixture
+        assert row.table is table_
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -451,12 +455,22 @@ 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 table_fixture(self, parent_, table_):
+        row = _Row(None, parent_)
+        parent_.table = table_
+        return row, table_
+
     # fixture components ---------------------------------------------
 
     @pytest.fixture
     def _index_(self, request):
         return property_mock(request, _Row, '_index')
 
+    @pytest.fixture
+    def parent_(self, request):
+        return instance_mock(request, Table)
+
     @pytest.fixture
     def table_(self, request):
         return instance_mock(request, Table)
