sect: add Section.page_width setter

Steve Canny · Steve Canny · commit d5073ee877a4 · 2014-06-20T23:31:17.000-07:00
diff --git a/docx/oxml/section.py b/docx/oxml/section.py
@@ -57,6 +57,11 @@ def page_width(self):
             return None
         return pgSz.w
 
+    @page_width.setter
+    def page_width(self, value):
+        pgSz = self.get_or_add_pgSz()
+        pgSz.w = value
+
     @property
     def start_type(self):
         """
diff --git a/docx/section.py b/docx/section.py
@@ -35,6 +35,10 @@ def page_width(self):
         """
         return self._sectPr.page_width
 
+    @page_width.setter
+    def page_width(self, value):
+        self._sectPr.page_width = value
+
     @property
     def start_type(self):
         """
diff --git a/tests/test_section.py b/tests/test_section.py
@@ -30,6 +30,11 @@ def it_knows_its_page_width(self, page_width_get_fixture):
         section, expected_page_width = page_width_get_fixture
         assert section.page_width == expected_page_width
 
+    def it_can_change_its_page_width(self, page_width_set_fixture):
+        section, new_page_width, expected_xml = page_width_set_fixture
+        section.page_width = new_page_width
+        assert section._sectPr.xml == expected_xml
+
     def it_knows_its_page_height(self, page_height_get_fixture):
         section, expected_page_height = page_height_get_fixture
         assert section.page_height == expected_page_height
@@ -60,6 +65,20 @@ def page_width_get_fixture(self, request):
         section = Section(sectPr)
         return section, expected_page_width
 
+    @pytest.fixture(params=[
+        (None,      None),
+        (Inches(1), 1440),
+    ])
+    def page_width_set_fixture(self, request):
+        new_page_width, expected_w_val = request.param
+        # section ----------------------
+        sectPr = self.sectPr_bldr().element
+        section = Section(sectPr)
+        # expected_xml -----------------
+        pgSz_bldr = self.pgSz_bldr(w=expected_w_val)
+        expected_xml = self.sectPr_bldr(pgSz_bldr).xml()
+        return section, new_page_width, expected_xml
+
     @pytest.fixture(params=[
         (False, None,         WD_SECTION.NEW_PAGE),
         (True,  None,         WD_SECTION.NEW_PAGE),
@@ -98,7 +117,7 @@ def start_type_set_fixture(self, request):
 
     # fixture components ---------------------------------------------
 
-    def pgSz_bldr(self, has_pgSz, w=None, h=None):
+    def pgSz_bldr(self, has_pgSz=True, w=None, h=None):
         if not has_pgSz:
             return None
         pgSz_bldr = a_pgSz()
