style: add LatentStyles.load_count getter

Steve Canny · Steve Canny · commit f9cacf3cb880 · 2015-02-08T00:01:47.000-08:00
diff --git a/docx/oxml/styles.py b/docx/oxml/styles.py
@@ -38,6 +38,7 @@ class CT_LatentStyles(BaseOxmlElement):
     """
     lsdException = ZeroOrMore('w:lsdException', successors=())
 
+    count = OptionalAttribute('w:count', ST_DecimalNumber)
     defUIPriority = OptionalAttribute('w:defUIPriority', ST_DecimalNumber)
 
     def get_by_name(self, name):
diff --git a/docx/styles/latent.py b/docx/styles/latent.py
@@ -44,6 +44,17 @@ def default_priority(self):
         """
         return self._element.defUIPriority
 
+    @property
+    def load_count(self):
+        """
+        Integer specifying the number of built-in styles to initialize to the
+        defaults specified in this |LatentStyles| object. |None| if there is
+        no setting in the XML (very uncommon). The default Word 2011 template
+        sets this value to 276, accounting for the built-in styles in Word
+        2010.
+        """
+        return self._element.count
+
 
 class _LatentStyle(ElementProxy):
     """
diff --git a/tests/styles/test_latent.py b/tests/styles/test_latent.py
@@ -43,8 +43,21 @@ def it_knows_its_default_priority(self, priority_get_fixture):
         latent_styles, expected_value = priority_get_fixture
         assert latent_styles.default_priority == expected_value
 
+    def it_knows_its_load_count(self, count_get_fixture):
+        latent_styles, expected_value = count_get_fixture
+        assert latent_styles.load_count == expected_value
+
     # fixture --------------------------------------------------------
 
+    @pytest.fixture(params=[
+        ('w:latentStyles',             None),
+        ('w:latentStyles{w:count=42}', 42),
+    ])
+    def count_get_fixture(self, request):
+        latentStyles_cxml, expected_value = request.param
+        latent_styles = LatentStyles(element(latentStyles_cxml))
+        return latent_styles, expected_value
+
     @pytest.fixture(params=[
         ('w:lsdException{w:name=Ab},w:lsdException,w:lsdException', 'Ab', 0),
         ('w:lsdException,w:lsdException{w:name=Cd},w:lsdException', 'Cd', 1),
