add ZipPkgReader.rels_xml_for()

Steve Canny · Steve Canny · commit ed80ca672a4e · 2013-08-05T14:56:07.000-07:00
diff --git a/opc/phys_pkg.py b/opc/phys_pkg.py
@@ -44,3 +44,14 @@ def content_types_xml(self):
         Return the `[Content_Types].xml` blob from the zip package.
         """
         return self._zipf.read(self._CONTENT_TYPES_MEMBERNAME)
+
+    def rels_xml_for(self, source_uri):
+        """
+        Return rels item XML for source with *source_uri* or None if no rels
+        item is present.
+        """
+        try:
+            rels_xml = self._zipf.read(source_uri.rels_uri.membername)
+        except KeyError:
+            rels_xml = None
+        return rels_xml
diff --git a/tests/test_phys_pkg.py b/tests/test_phys_pkg.py
@@ -11,6 +11,7 @@
 
 import hashlib
 
+from opc.packuri import PACKAGE_URI, PackURI
 from opc.phys_pkg import PhysPkgReader, ZipPkgReader
 
 import pytest
@@ -69,3 +70,13 @@ def it_can_be_closed(self, ZipFile_):
     def it_has_the_content_types_xml(self, phys_reader):
         sha1 = hashlib.sha1(phys_reader.content_types_xml).hexdigest()
         assert sha1 == '9604a4fb3bf9626f5ad59a4e82029b3a501f106a'
+
+    def it_can_retrieve_rels_xml_for_source_uri(self, phys_reader):
+        rels_xml = phys_reader.rels_xml_for(PACKAGE_URI)
+        sha1 = hashlib.sha1(rels_xml).hexdigest()
+        assert sha1 == 'e31451d4bbe7d24adbe21454b8e9fdae92f50de5'
+
+    def it_returns_none_when_part_has_no_rels_xml(self, phys_reader):
+        partname = PackURI('/ppt/viewProps.xml')
+        rels_xml = phys_reader.rels_xml_for(partname)
+        assert rels_xml is None
