oxml: add CT_Body.clear_content()

Steve Canny · Steve Canny · commit 953919b73362 · 2013-12-20T21:05:20.000-08:00
diff --git a/docx/oxml/parts.py b/docx/oxml/parts.py
@@ -38,6 +38,16 @@ def add_p(self):
             self.append(p)
         return p
 
+    def clear_content(self):
+        """
+        Remove all content child elements from this <w:body> element. Leave
+        the <w:sectPr> element if it is present.
+        """
+        children = self.getchildren()
+        content_elms = children[:-1] if self._has_sectPr else children
+        for content_elm in content_elms:
+            self.remove(content_elm)
+
     @property
     def _has_sectPr(self):
         """
diff --git a/tests/oxml/test_parts.py b/tests/oxml/test_parts.py
@@ -34,6 +34,23 @@ def it_can_add_a_p_to_itself(self):
             assert body.xml == after_body_bldr.xml
             assert isinstance(p, CT_P)
 
+    def it_can_clear_all_the_content_it_holds(self):
+        """
+        Remove all content child elements from this <w:body> element.
+        """
+        cases = (
+            (a_body(), a_body()),
+            (a_body().with_p(), a_body()),
+            (a_body().with_sectPr(), a_body().with_sectPr()),
+            (a_body().with_p().with_sectPr(), a_body().with_sectPr()),
+        )
+        for before_body_bldr, after_body_bldr in cases:
+            body = before_body_bldr.element
+            # exercise -----------------
+            body.clear_content()
+            # verify -------------------
+            assert body.xml == after_body_bldr.xml
+
 
 class DescribeCT_Document(object):
 
