parfmt: add ParagraphFormat.keep_together setter

Steve Canny · Steve Canny · commit ea6a195c458a · 2015-02-08T00:01:44.000-08:00
diff --git a/docx/oxml/text/paragraph.py b/docx/oxml/text/paragraph.py
@@ -220,6 +220,13 @@ def keepLines_val(self):
             return None
         return keepLines.val
 
+    @keepLines_val.setter
+    def keepLines_val(self, value):
+        if value is None:
+            self._remove_keepLines()
+        else:
+            self.get_or_add_keepLines().val = value
+
     @property
     def keepNext_val(self):
         """
diff --git a/docx/text/paragraph.py b/docx/text/paragraph.py
@@ -189,6 +189,10 @@ def keep_together(self):
             return None
         return pPr.keepLines_val
 
+    @keep_together.setter
+    def keep_together(self, value):
+        self._element.get_or_add_pPr().keepLines_val = value
+
     @property
     def keep_with_next(self):
         """
diff --git a/tests/text/test_paragraph.py b/tests/text/test_paragraph.py
@@ -365,6 +365,11 @@ def it_knows_its_on_off_prop_values(self, on_off_get_fixture):
         paragraph_format, prop_name, expected_value = on_off_get_fixture
         assert getattr(paragraph_format, prop_name) == expected_value
 
+    def it_can_change_its_on_off_props(self, on_off_set_fixture):
+        paragraph_format, prop_name, value, expected_xml = on_off_set_fixture
+        setattr(paragraph_format, prop_name, value)
+        assert paragraph_format._element.xml == expected_xml
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture(params=[
@@ -546,6 +551,19 @@ def on_off_get_fixture(self, request):
         paragraph_format = ParagraphFormat(element(p_cxml))
         return paragraph_format, prop_name, expected_value
 
+    @pytest.fixture(params=[
+        ('w:p', 'keep_together',     True,  'w:p/w:pPr/w:keepLines'),
+        ('w:p/w:pPr/w:keepLines',                 'keep_together',     False,
+         'w:p/w:pPr/w:keepLines{w:val=0}'),
+        ('w:p/w:pPr/w:keepLines{w:val=0}',        'keep_together',     None,
+         'w:p/w:pPr'),
+    ])
+    def on_off_set_fixture(self, request):
+        p_cxml, prop_name, value, expected_cxml = request.param
+        paragraph_format = ParagraphFormat(element(p_cxml))
+        expected_xml = xml(expected_cxml)
+        return paragraph_format, prop_name, value, expected_xml
+
     @pytest.fixture(params=[
         ('w:p',                             None),
         ('w:p/w:pPr',                       None),
