parts: add _Body.add_paragraph()

Steve Canny · Steve Canny · commit 63f32a431e1d · 2013-12-20T20:10:44.000-08:00
diff --git a/docx/parts.py b/docx/parts.py
@@ -14,6 +14,7 @@
 from opc import Part
 
 from docx.oxml.base import oxml_fromstring
+from docx.text import Paragraph
 
 
 class _Document(Part):
@@ -43,3 +44,10 @@ class _Body(object):
     Proxy for ``<w:body>`` element in this document, having primarily a
     container role.
     """
+    def __init__(self, body_elm):
+        super(_Body, self).__init__()
+        self._body = body_elm
+
+    def add_paragraph(self):
+        p = self._body.add_p()
+        return Paragraph(p)
diff --git a/docx/text.py b/docx/text.py
@@ -0,0 +1,18 @@
+# -*- coding: utf-8 -*-
+#
+# text.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
+
+"""
+Text-related proxy types for python-docx, such as Paragraph and Run.
+"""
+
+
+class Paragraph(object):
+    """
+    Proxy object wrapping ``<w:p>`` element.
+    """
diff --git a/tests/test_parts.py b/tests/test_parts.py
@@ -9,7 +9,7 @@
 
 """Test suite for the docx.parts module."""
 
-from docx.parts import _Document
+from docx.parts import _Body, _Document
 
 import pytest
 
@@ -56,3 +56,22 @@ def it_has_a_body(self, init, _Body_):
         # verify -----------------------
         _Body_.assert_called_once_with(doc._element.body)
         assert body is _Body_.return_value
+
+
+class Describe_Body(object):
+
+    @pytest.fixture
+    def Paragraph_(self, request):
+        return class_mock('docx.parts.Paragraph', request)
+
+    def it_can_add_a_paragraph_to_itself(self, Paragraph_):
+        # mockery ----------------------
+        body_elm = Mock(name='body_elm')
+        body_elm.add_p.return_value = p_elm = Mock(name='p_elm')
+        body = _Body(body_elm)
+        # exercise ---------------------
+        p = body.add_paragraph()
+        # verify -----------------------
+        body_elm.add_p.assert_called_once_with()
+        Paragraph_.assert_called_once_with(p_elm)
+        assert p is Paragraph_.return_value
