parfmt: add ParaFormat.page_break_before setter

Steve Canny · Steve Canny · commit 3832dc117f2d · 2015-02-08T00:01:45.000-08:00
diff --git a/docx/oxml/text/paragraph.py b/docx/oxml/text/paragraph.py
@@ -254,6 +254,13 @@ def pageBreakBefore_val(self):
             return None
         return pageBreakBefore.val
 
+    @pageBreakBefore_val.setter
+    def pageBreakBefore_val(self, value):
+        if value is None:
+            self._remove_pageBreakBefore()
+        else:
+            self.get_or_add_pageBreakBefore().val = value
+
     @property
     def spacing_after(self):
         """
diff --git a/docx/text/paragraph.py b/docx/text/paragraph.py
@@ -303,6 +303,10 @@ def page_break_before(self):
             return None
         return pPr.pageBreakBefore_val
 
+    @page_break_before.setter
+    def page_break_before(self, value):
+        self._element.get_or_add_pPr().pageBreakBefore_val = value
+
     @property
     def right_indent(self):
         """
diff --git a/tests/text/test_paragraph.py b/tests/text/test_paragraph.py
@@ -554,14 +554,19 @@ def on_off_get_fixture(self, request):
     @pytest.fixture(params=[
         ('w:p', 'keep_together',     True,  'w:p/w:pPr/w:keepLines'),
         ('w:p', 'keep_with_next',    True,  'w:p/w:pPr/w:keepNext'),
+        ('w:p', 'page_break_before', True,  'w:p/w:pPr/w:pageBreakBefore'),
         ('w:p/w:pPr/w:keepLines',                 'keep_together',     False,
          'w:p/w:pPr/w:keepLines{w:val=0}'),
         ('w:p/w:pPr/w:keepNext',                  'keep_with_next',    False,
          'w:p/w:pPr/w:keepNext{w:val=0}'),
+        ('w:p/w:pPr/w:pageBreakBefore',           'page_break_before', False,
+         'w:p/w:pPr/w:pageBreakBefore{w:val=0}'),
         ('w:p/w:pPr/w:keepLines{w:val=0}',        'keep_together',     None,
          'w:p/w:pPr'),
         ('w:p/w:pPr/w:keepNext{w:val=0}',         'keep_with_next',    None,
          'w:p/w:pPr'),
+        ('w:p/w:pPr/w:pageBreakBefore{w:val=0}',  'page_break_before', None,
+         'w:p/w:pPr'),
     ])
     def on_off_set_fixture(self, request):
         p_cxml, prop_name, value, expected_cxml = request.param
