tbl: base _Row on Parented

Steve Canny · Steve Canny · commit ce779e5a799b · 2014-07-11T17:11:19.000-07:00
diff --git a/docx/table.py b/docx/table.py
@@ -36,7 +36,7 @@ def add_row(self):
         tr = tbl.add_tr()
         for gridCol in tbl.tblGrid.gridCol_lst:
             tr.add_tc()
-        return _Row(tr)
+        return _Row(tr, self)
 
     def cell(self, row_idx, col_idx):
         """
@@ -216,12 +216,12 @@ def _gridCol_lst(self):
         return tblGrid.gridCol_lst
 
 
-class _Row(object):
+class _Row(Parented):
     """
     Table row
     """
-    def __init__(self, tr):
-        super(_Row, self).__init__()
+    def __init__(self, tr, parent):
+        super(_Row, self).__init__(parent)
         self._tr = tr
 
     @lazyproperty
@@ -277,10 +277,10 @@ def __getitem__(self, idx):
         except IndexError:
             msg = "row index [%d] out of range" % idx
             raise IndexError(msg)
-        return _Row(tr)
+        return _Row(tr, self)
 
     def __iter__(self):
-        return (_Row(tr) for tr in self._tbl.tr_lst)
+        return (_Row(tr, self) for tr in self._tbl.tr_lst)
 
     def __len__(self):
         return len(self._tbl.tr_lst)
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -278,7 +278,7 @@ def columns_fixture(self):
 class Describe_Row(object):
 
     def it_provides_access_to_the_row_cells(self):
-        row = _Row(element('w:tr'))
+        row = _Row(element('w:tr'), None)
         cells = row.cells
         assert isinstance(cells, _RowCells)
 
