add CT_Override.new()

Steve Canny · Steve Canny · commit 9e8315dbc80c · 2013-08-05T15:08:42.000-07:00
diff --git a/opc/oxml.py b/opc/oxml.py
@@ -114,6 +114,19 @@ def content_type(self):
         """
         return self.get('ContentType')
 
+    @staticmethod
+    def new(partname, content_type):
+        """
+        Return a new ``<Override>`` element with attributes set to parameter
+        values.
+        """
+        xml = '<Override xmlns="%s"/>' % nsmap['ct']
+        override = oxml_fromstring(xml)
+        override.set('PartName', partname)
+        override.set('ContentType', content_type)
+        objectify.deannotate(override, cleanup_namespaces=True)
+        return override
+
     @property
     def partname(self):
         """
diff --git a/tests/test_oxml.py b/tests/test_oxml.py
@@ -35,6 +35,11 @@ def it_provides_read_access_to_xml_values(self):
         assert override.partname == '/part/name.xml'
         assert override.content_type == 'app/vnd.type'
 
+    def it_can_construct_a_new_override_element(self):
+        override = CT_Override.new('/part/name.xml', 'app/vnd.type')
+        expected_xml = an_Override().xml
+        assert override.xml == expected_xml
+
 
 class DescribeCT_Relationship(object):
 
