tbl: base Table on Parented

Steve Canny · Steve Canny · commit bf42778316b5 · 2014-07-11T16:51:04.000-07:00
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -135,7 +135,7 @@ def add_table(self, rows, cols):
         the main document story.
         """
         tbl = self._body.add_tbl()
-        table = Table(tbl)
+        table = Table(tbl, self)
         for i in range(cols):
             table.add_column()
         for i in range(rows):
@@ -161,7 +161,7 @@ def tables(self):
         A sequence containing all the tables in the document, in the order
         they appear.
         """
-        return [Table(tbl) for tbl in self._body.tbl_lst]
+        return [Table(tbl, self) for tbl in self._body.tbl_lst]
 
 
 class InlineShapes(Parented):
diff --git a/docx/table.py b/docx/table.py
@@ -6,16 +6,16 @@
 
 from __future__ import absolute_import, print_function, unicode_literals
 
-from .shared import lazyproperty, write_only_property
+from .shared import lazyproperty, Parented, write_only_property
 from .text import Paragraph
 
 
-class Table(object):
+class Table(Parented):
     """
     Proxy class for a WordprocessingML ``<w:tbl>`` element.
     """
-    def __init__(self, tbl):
-        super(Table, self).__init__()
+    def __init__(self, tbl, parent):
+        super(Table, self).__init__(parent)
         self._tbl = tbl
 
     def add_column(self):
diff --git a/tests/test_table.py b/tests/test_table.py
@@ -65,14 +65,14 @@ def it_can_apply_a_table_style_by_name(self, table_style_set_fixture):
     @pytest.fixture
     def add_column_fixture(self):
         tbl = _tbl_bldr(2, 1).element
-        table = Table(tbl)
+        table = Table(tbl, None)
         expected_xml = _tbl_bldr(2, 2).xml()
         return table, expected_xml
 
     @pytest.fixture
     def add_row_fixture(self):
         tbl = _tbl_bldr(rows=1, cols=2).element
-        table = Table(tbl)
+        table = Table(tbl, None)
         expected_xml = _tbl_bldr(rows=2, cols=2).xml()
         return table, expected_xml
 
@@ -82,7 +82,7 @@ def add_row_fixture(self):
     ])
     def table_style_get_fixture(self, request):
         tbl_cxml, expected_style = request.param
-        table = Table(element(tbl_cxml))
+        table = Table(element(tbl_cxml), None)
         return table, expected_style
 
     @pytest.fixture(params=[
@@ -97,7 +97,7 @@ def table_style_get_fixture(self, request):
     ])
     def table_style_set_fixture(self, request):
         tbl_cxml, new_style, expected_cxml = request.param
-        table = Table(element(tbl_cxml))
+        table = Table(element(tbl_cxml), None)
         expected_xml = xml(expected_cxml)
         return table, new_style, expected_xml
 
@@ -106,7 +106,7 @@ def table_style_set_fixture(self, request):
     @pytest.fixture
     def table(self):
         tbl = _tbl_bldr(rows=2, cols=2).element
-        table = Table(tbl)
+        table = Table(tbl, None)
         return table
 
 
