add CT_Default.new()

Steve Canny · Steve Canny · commit 791e48e813b6 · 2013-08-05T15:08:41.000-07:00
diff --git a/opc/oxml.py b/opc/oxml.py
@@ -87,6 +87,19 @@ def extension(self):
         """
         return self.get('Extension')
 
+    @staticmethod
+    def new(ext, content_type):
+        """
+        Return a new ``<Default>`` element with attributes set to parameter
+        values.
+        """
+        xml = '<Default xmlns="%s"/>' % nsmap['ct']
+        default = oxml_fromstring(xml)
+        default.set('Extension', ext[1:])
+        default.set('ContentType', content_type)
+        objectify.deannotate(default, cleanup_namespaces=True)
+        return default
+
 
 class CT_Override(OxmlBaseElement):
     """
diff --git a/tests/test_oxml.py b/tests/test_oxml.py
@@ -22,6 +22,11 @@ def it_provides_read_access_to_xml_values(self):
         assert default.extension == 'xml'
         assert default.content_type == 'application/xml'
 
+    def it_can_construct_a_new_default_element(self):
+        default = CT_Default.new('.xml', 'application/xml')
+        expected_xml = a_Default().xml
+        assert default.xml == expected_xml
+
 
 class DescribeCT_Override(object):
 
