style: add StylesPart.styles

Steve Canny · Steve Canny · commit d6a46fc137ac · 2014-02-02T22:19:41.000-08:00
diff --git a/docx/parts/styles.py b/docx/parts/styles.py
@@ -10,6 +10,7 @@
 
 from ..opc.package import Part
 from ..oxml.shared import oxml_fromstring
+from ..shared import lazyproperty
 
 
 class StylesPart(Part):
@@ -39,3 +40,21 @@ def new(cls):
         ``<w:styles>`` element.
         """
         raise NotImplementedError
+
+    @lazyproperty
+    def styles(self):
+        """
+        The |_Styles| instance containing the styles (<w:style> element
+        proxies) for this styles part.
+        """
+        return _Styles(self._element)
+
+
+class _Styles(object):
+    """
+    Collection of |_Style| instances corresponding to the ``<w:style>``
+    elements in a styles part.
+    """
+    def __init__(self, styles_elm):
+        super(_Styles, self).__init__()
+        self._styles_elm = styles_elm
diff --git a/tests/parts/test_styles.py b/tests/parts/test_styles.py
@@ -13,10 +13,10 @@
 from docx.opc.packuri import PackURI
 from docx.oxml.parts.styles import CT_Styles
 from docx.package import Package
-from docx.parts.styles import StylesPart
+from docx.parts.styles import StylesPart, _Styles
 
 from ..unitutil import (
-    function_mock, initializer_mock, instance_mock, method_mock
+    function_mock, class_mock, initializer_mock, instance_mock, method_mock
 )
 
 
@@ -51,6 +51,12 @@ def it_can_be_constructed_by_opc_part_factory(self, construct_fixture):
         )
         assert isinstance(styles_part, StylesPart)
 
+    def it_provides_access_to_the_styles(self, styles_fixture):
+        styles_part, _Styles_, styles_elm_, styles_ = styles_fixture
+        styles = styles_part.styles
+        _Styles_.assert_called_once_with(styles_elm_)
+        assert styles is styles_
+
     # fixtures -------------------------------------------------------
 
     @pytest.fixture
@@ -98,10 +104,25 @@ def package_(self, request):
     def partname_(self, request):
         return instance_mock(request, PackURI)
 
+    @pytest.fixture
+    def _Styles_(self, request, styles_):
+        return class_mock(
+            request, 'docx.parts.styles._Styles', return_value=styles_
+        )
+
+    @pytest.fixture
+    def styles_(self, request):
+        return instance_mock(request, _Styles)
+
     @pytest.fixture
     def styles_elm_(self, request):
         return instance_mock(request, CT_Styles)
 
+    @pytest.fixture
+    def styles_fixture(self, _Styles_, styles_elm_, styles_):
+        styles_part = StylesPart(None, None, styles_elm_, None)
+        return styles_part, _Styles_, styles_elm_, styles_
+
     @pytest.fixture
     def styles_part_(self, request):
         return instance_mock(request, StylesPart)
