opc: add CorePropertiesPart.default()

Steve Canny · Steve Canny · commit da9a4688e004 · 2014-12-13T23:31:00.000-08:00
diff --git a/docx/opc/parts/coreprops.py b/docx/opc/parts/coreprops.py
@@ -8,7 +8,12 @@
     absolute_import, division, print_function, unicode_literals
 )
 
+from datetime import datetime
+
+from ..constants import CONTENT_TYPE as CT
 from ..coreprops import CoreProperties
+from ...oxml.parts.coreprops import CT_CoreProperties
+from ..packuri import PackURI
 from ..part import XmlPart
 
 
@@ -23,7 +28,13 @@ def default(cls, package):
         Return a new |CorePropertiesPart| object initialized with default
         values for its base properties.
         """
-        raise NotImplementedError
+        core_properties_part = cls._new(package)
+        core_properties = core_properties_part.core_properties
+        core_properties.title = 'Word Document'
+        core_properties.last_modified_by = 'python-docx'
+        core_properties.revision = 1
+        core_properties.modified = datetime.utcnow()
+        return core_properties_part
 
     @property
     def core_properties(self):
@@ -32,3 +43,12 @@ def core_properties(self):
         properties contained in this core properties part.
         """
         return CoreProperties(self.element)
+
+    @classmethod
+    def _new(cls, package):
+        partname = PackURI('/docProps/core.xml')
+        content_type = CT.OPC_CORE_PROPERTIES
+        coreProperties = CT_CoreProperties.new()
+        return CorePropertiesPart(
+            partname, content_type, coreProperties, package
+        )
diff --git a/docx/oxml/parts/coreprops.py b/docx/oxml/parts/coreprops.py
@@ -12,7 +12,8 @@
 
 from datetime import datetime, timedelta
 
-from ..ns import qn
+from .. import parse_xml
+from ..ns import nsdecls, qn
 from ..xmlchemy import BaseOxmlElement, ZeroOrOne
 
 
@@ -40,6 +41,19 @@ class CT_CoreProperties(BaseOxmlElement):
     title = ZeroOrOne('dc:title', successors=())
     version = ZeroOrOne('cp:version', successors=())
 
+    _coreProperties_tmpl = (
+        '<cp:coreProperties %s/>\n' % nsdecls('cp', 'dc', 'dcterms')
+    )
+
+    @classmethod
+    def new(cls):
+        """
+        Return a new ``<cp:coreProperties>`` element
+        """
+        xml = cls._coreProperties_tmpl
+        coreProperties = parse_xml(xml)
+        return coreProperties
+
     @property
     def author_text(self):
         """
diff --git a/features/doc-coreprops.feature b/features/doc-coreprops.feature
@@ -16,7 +16,6 @@ Feature: Read and write core document properties
       Then the core property values match the new values
 
 
-  @wip
   Scenario: a default core properties part is added if doc doesn't have one
      Given a document having no core properties part
       When I access the core properties object
diff --git a/tests/opc/parts/test_coreprops.py b/tests/opc/parts/test_coreprops.py
@@ -8,6 +8,8 @@
     absolute_import, division, print_function, unicode_literals
 )
 
+from datetime import datetime, timedelta
+
 import pytest
 
 from docx.opc.coreprops import CoreProperties
@@ -25,6 +27,17 @@ def it_provides_access_to_its_core_props_object(self, coreprops_fixture):
         CoreProperties_.assert_called_once_with(core_properties_part.element)
         assert isinstance(core_properties, CoreProperties)
 
+    def it_can_create_a_default_core_properties_part(self):
+        core_properties_part = CorePropertiesPart.default(None)
+        assert isinstance(core_properties_part, CorePropertiesPart)
+        core_properties = core_properties_part.core_properties
+        assert core_properties.title == 'Word Document'
+        assert core_properties.last_modified_by == 'python-docx'
+        assert core_properties.revision == 1
+        delta = datetime.utcnow() - core_properties.modified
+        max_expected_delta = timedelta(seconds=2)
+        assert delta < max_expected_delta
+
     # fixtures ---------------------------------------------
 
     @pytest.fixture
