style: add DocumentPart.styles

Steve Canny · Steve Canny · commit 5b25f14411aa · 2014-12-27T22:51:55.000-08:00
diff --git a/docx/parts/document.py b/docx/parts/document.py
@@ -108,7 +108,7 @@ def styles(self):
         A |Styles| object providing access to the styles in the styles part
         of this document.
         """
-        raise NotImplementedError
+        return self._styles_part.styles
 
     @property
     def tables(self):
@@ -119,6 +119,14 @@ def tables(self):
         """
         return self.body.tables
 
+    @property
+    def _styles_part(self):
+        """
+        Instance of |StylesPart| for this document. Creates an empty styles
+        part if one is not present.
+        """
+        raise NotImplementedError
+
 
 class _Body(BlockItemContainer):
     """
diff --git a/tests/parts/test_document.py b/tests/parts/test_document.py
@@ -15,8 +15,10 @@
 from docx.package import ImageParts, Package
 from docx.parts.document import _Body, DocumentPart, InlineShapes, Sections
 from docx.parts.image import ImagePart
+from docx.parts.styles import StylesPart
 from docx.section import Section
 from docx.shape import InlineShape
+from docx.styles.styles import Styles
 from docx.table import Table
 from docx.text.paragraph import Paragraph
 from docx.text.run import Run
@@ -54,6 +56,11 @@ def it_provides_access_to_the_document_tables(self, tables_fixture):
         tables = document_part.tables
         assert tables is tables_
 
+    def it_provides_access_to_the_document_styles(self, styles_fixture):
+        document_part, styles_ = styles_fixture
+        styles = document_part.styles
+        assert styles is styles_
+
     def it_provides_access_to_the_inline_shapes_in_the_document(
             self, inline_shapes_fixture):
         document, InlineShapes_, body_elm = inline_shapes_fixture
@@ -162,11 +169,18 @@ def paragraphs_fixture(self, document_part_body_, body_, paragraphs_):
         return document_part, paragraphs_
 
     @pytest.fixture
-    def sections_fixture(self, request, Sections_):
+    def sections_fixture(self, Sections_):
         document_elm = a_document().with_nsdecls().element
         document = DocumentPart(None, None, document_elm, None)
         return document, document_elm, Sections_
 
+    @pytest.fixture
+    def styles_fixture(self, _styles_part_prop_, styles_part_, styles_):
+        document_part = DocumentPart(None, None, None, None)
+        _styles_part_prop_.return_value = styles_part_
+        styles_part_.styles = styles_
+        return document_part, styles_
+
     @pytest.fixture
     def tables_fixture(self, document_part_body_, body_, tables_):
         document_part = DocumentPart(None, None, None, None)
@@ -275,6 +289,18 @@ def sectPr_(self, request):
     def start_type_(self, request):
         return instance_mock(request, int)
 
+    @pytest.fixture
+    def styles_(self, request):
+        return instance_mock(request, Styles)
+
+    @pytest.fixture
+    def styles_part_(self, request):
+        return instance_mock(request, StylesPart)
+
+    @pytest.fixture
+    def _styles_part_prop_(self, request):
+        return property_mock(request, DocumentPart, '_styles_part')
+
     @pytest.fixture
     def table_(self, request):
         return instance_mock(request, Table)
