api: add _Document.body

Steve Canny · Steve Canny · commit 946ab5eb84dd · 2013-12-20T20:10:43.000-08:00
diff --git a/docx/api.py b/docx/api.py
@@ -46,3 +46,11 @@ class _Document(object):
     """
     def __init__(self, pkg, document_part):
         super(_Document, self).__init__()
+        self._document = document_part
+
+    @property
+    def body(self):
+        """
+        Return a reference to the |Body| instance for this document.
+        """
+        return self._document.body
diff --git a/docx/parts.py b/docx/parts.py
@@ -0,0 +1,25 @@
+# -*- coding: utf-8 -*-
+#
+# parts.py
+#
+# Copyright (C) 2013 Steve Canny scanny@cisco.com
+#
+# This module is part of python-docx and is released under the MIT License:
+# http://www.opensource.org/licenses/mit-license.php
+
+"""
+Document parts such as _Document, and closely related classes.
+"""
+
+from opc import Part
+
+
+class _Document(Part):
+    """
+    Main document part of a WordprocessingML (WML) package, aka a .docx file.
+    """
+    @property
+    def body(self):
+        """
+        The |_Body| instance containing the content for this document.
+        """
diff --git a/tests/test_api.py b/tests/test_api.py
@@ -11,8 +11,9 @@
 
 import pytest
 
-from mock import Mock, PropertyMock
+from mock import create_autospec, Mock, PropertyMock
 
+from docx import parts
 from docx.api import Document, _Document
 from opc.constants import CONTENT_TYPE as CT
 
@@ -69,3 +70,14 @@ def it_should_raise_if_not_a_docx_file(self, OpcPackage_mockery):
         # verify -----------------------
         with pytest.raises(ValueError):
             Document(docx)
+
+
+class Describe_Document(object):
+
+    def it_provides_access_to_the_document_body(self):
+        document_part = create_autospec(parts._Document)
+        doc = _Document(None, document_part)
+        # exercise ---------------------
+        body = doc.body
+        # verify -----------------------
+        assert body is document_part.body
