tbl: base _Cell on Parented

Steve Canny · Steve Canny · commit 54d619f1bdc1 · 2014-07-11T17:22:51.000-07:00
diff --git a/docx/table.py b/docx/table.py
@@ -79,12 +79,12 @@ def _tblPr(self):
         return self._tbl.tblPr
 
 
-class _Cell(object):
+class _Cell(Parented):
     """
     Table cell
     """
-    def __init__(self, tc):
-        super(_Cell, self).__init__()
+    def __init__(self, tc, parent):
+        super(_Cell, self).__init__(parent)
         self._tc = tc
 
     @property
@@ -159,12 +159,12 @@ def __getitem__(self, idx):
             msg = "cell index [%d] is out of range" % idx
             raise IndexError(msg)
         tc = tr.tc_lst[self._col_idx]
-        return _Cell(tc)
+        return _Cell(tc, self)
 
     def __iter__(self):
         for tr in self._tr_lst:
             tc = tr.tc_lst[self._col_idx]
-            yield _Cell(tc)
+            yield _Cell(tc, self)
 
     def __len__(self):
         return len(self._tr_lst)
@@ -250,10 +250,10 @@ def __getitem__(self, idx):
         except IndexError:
             msg = "cell index [%d] is out of range" % idx
             raise IndexError(msg)
-        return _Cell(tc)
+        return _Cell(tc, self)
 
     def __iter__(self):
-        return (_Cell(tc) for tc in self._tr.tc_lst)
+        return (_Cell(tc, self) for tc in self._tr.tc_lst)
 
     def __len__(self):
         return len(self._tr.tc_lst)
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -142,13 +142,13 @@ def it_can_replace_its_content_with_a_string_of_text(
     ])
     def text_set_fixture(self, request):
         tc_cxml, new_text, expected_cxml = request.param
-        cell = _Cell(element(tc_cxml))
+        cell = _Cell(element(tc_cxml), None)
         expected_xml = xml(expected_cxml)
         return cell, new_text, expected_xml
 
     @pytest.fixture
     def paragraphs_fixture(self):
-        return _Cell(element('w:tc/(w:p, w:p)'))
+        return _Cell(element('w:tc/(w:p, w:p)'), None)
 
 
 class Describe_Column(object):
