blkct: introduce BlockItemContainer

Steve Canny · Steve Canny · commit 45f2c64ff552 · 2014-07-17T22:34:23.000-07:00
* base _Body on BlockItemContainer
* no substantial inheritence yet, just getting it wired up
diff --git a/docx/blkcntnr.py b/docx/blkcntnr.py
@@ -0,0 +1,23 @@
+# encoding: utf-8
+
+"""
+Block item container, used by body, cell, header, etc. Block level items are
+things like paragraph and table, although there are a few other specialized
+ones like structured document tags.
+"""
+
+from __future__ import absolute_import, print_function
+
+from .shared import Parented
+
+
+class BlockItemContainer(Parented):
+    """
+    Base class for proxy objects that can contain block items, such as _Body,
+    _Cell, header, footer, footnote, endnote, comment, and text box objects.
+    Provides the shared functionality to add a block item like a paragraph or
+    table.
+    """
+    def __init__(self, element, parent):
+        super(BlockItemContainer, self).__init__(parent)
+        self._element = element
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -10,6 +10,7 @@
 
 from collections import Sequence
 
+from ..blkcntnr import BlockItemContainer
 from ..enum.section import WD_SECTION
 from ..opc.constants import RELATIONSHIP_TYPE as RT
 from ..opc.package import XmlPart
@@ -113,13 +114,13 @@ def tables(self):
         return self.body.tables
 
 
-class _Body(Parented):
+class _Body(BlockItemContainer):
     """
     Proxy for ``<w:body>`` element in this document, having primarily a
     container role.
     """
     def __init__(self, body_elm, parent):
-        super(_Body, self).__init__(parent)
+        super(_Body, self).__init__(body_elm, parent)
         self._body = body_elm
 
     def add_paragraph(self, text='', style=None):
