sect: add Sections.__len__()

Steve Canny · Steve Canny · commit 6469b5a8e874 · 2014-06-19T00:08:21.000-07:00
diff --git a/docx/oxml/parts/document.py b/docx/oxml/parts/document.py
@@ -16,6 +16,14 @@ class CT_Document(BaseOxmlElement):
     """
     body = ZeroOrOne('w:body')
 
+    @property
+    def sectPr_lst(self):
+        """
+        Return a list containing a reference to each ``<w:sectPr>`` element
+        in the document, in the order encountered.
+        """
+        return self.xpath('.//w:sectPr')
+
 
 class CT_Body(BaseOxmlElement):
     """
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -8,6 +8,8 @@
     absolute_import, division, print_function, unicode_literals
 )
 
+from collections import Sequence
+
 from ..opc.constants import RELATIONSHIP_TYPE as RT
 from ..opc.oxml import serialize_part_xml
 from ..opc.package import Part
@@ -223,11 +225,17 @@ def _inline_lst(self):
         return body.xpath(xpath)
 
 
-class Sections(object):
+class Sections(Sequence):
     """
     Sequence of |Section| objects corresponding to the sections in the
     document.
     """
     def __init__(self, document_elm):
         super(Sections, self).__init__()
         self._document_elm = document_elm
+
+    def __getitem__(self, key):
+        pass
+
+    def __len__(self):
+        return len(self._document_elm.sectPr_lst)
diff --git a/features/doc-access-sections.feature b/features/doc-access-sections.feature
@@ -4,7 +4,6 @@ Feature: Access document sections
   I need a way to access document sections
 
 
-  @wip
   Scenario: Access section collection of a document
      Given a document having three sections
       Then I can access the section collection of the document
diff --git a/tests/parts/test_document.py b/tests/parts/test_document.py
@@ -16,7 +16,7 @@
 from docx.oxml.parts.document import CT_Body, CT_Document
 from docx.oxml.text import CT_R
 from docx.package import ImageParts, Package
-from docx.parts.document import _Body, DocumentPart, InlineShapes
+from docx.parts.document import _Body, DocumentPart, InlineShapes, Sections
 from docx.parts.image import ImagePart
 from docx.shape import InlineShape
 from docx.table import Table
@@ -27,7 +27,7 @@
 from ..oxml.unitdata.table import (
     a_gridCol, a_tbl, a_tblGrid, a_tblPr, a_tblW, a_tc, a_tr
 )
-from ..oxml.unitdata.text import a_p, a_sectPr, an_r
+from ..oxml.unitdata.text import a_p, a_pPr, a_sectPr, an_r
 from ..unitutil import (
     function_mock, class_mock, initializer_mock, instance_mock, loose_mock,
     method_mock, property_mock
@@ -617,3 +617,26 @@ def rId_(self, request):
     @pytest.fixture
     def shape_id_(self, request):
         return instance_mock(request, int)
+
+
+class DescribeSections(object):
+
+    def it_knows_how_many_sections_it_contains(self, len_fixture):
+        sections, expected_len = len_fixture
+        print(sections._document_elm.xml)
+        assert len(sections) == expected_len
+
+    # fixtures -------------------------------------------------------
+
+    @pytest.fixture
+    def len_fixture(self):
+        document_elm = (
+            a_document().with_nsdecls().with_child(
+                a_body().with_child(
+                    a_p().with_child(
+                        a_pPr().with_child(
+                            a_sectPr()))).with_child(
+                    a_sectPr()))
+        ).element
+        sections = Sections(document_elm)
+        return sections, 2
