python-openxml
diff --git a/‎docx/opc/package.py‎
Lines changed: 22 additions & 0 deletions b/‎docx/opc/package.py‎
Lines changed: 22 additions & 0 deletions
diff --git a/‎docx/package.py‎
Lines changed: 37 additions & 1 deletion b/‎docx/package.py‎
Lines changed: 37 additions & 1 deletion
diff --git a/‎tests/test_files/having-images.docx‎
130 KB b/‎tests/test_files/having-images.docx‎
130 KB
diff --git a/‎tests/test_files/monty-truth.png‎
62.8 KB b/‎tests/test_files/monty-truth.png‎
62.8 KB
diff --git a/‎tests/test_files/python-icon.jpeg‎
3.2 KB b/‎tests/test_files/python-icon.jpeg‎
3.2 KB
diff --git a/‎tests/test_files/python-powered.png‎
5.97 KB b/‎tests/test_files/python-powered.png‎
5.97 KB
diff --git a/‎tests/test_files/python.bmp‎
44.2 KB b/‎tests/test_files/python.bmp‎
44.2 KB
diff --git a/‎tests/test_package.py‎
Lines changed: 18 additions & 0 deletions b/‎tests/test_package.py‎
Lines changed: 18 additions & 0 deletions
diff --git a/‎tests/unitutil.py‎
Lines changed: 12 additions & 0 deletions b/‎tests/unitutil.py‎
Lines changed: 12 additions & 0 deletions
@@ -33,6 +33,28 @@ def after_unmarshal(self):
         # subclass
         pass
 
+    def iter_rels(self):
+        """
+        Generate exactly one reference to each relationship in the package by
+        performing a depth-first traversal of the rels graph.
+        """
+        def walk_rels(source, visited=None):
+            visited = [] if visited is None else visited
+            for rel in source.rels.values():
+                yield rel
+                if rel.is_external:
+                    continue
+                part = rel.target_part
+                if part in visited:
+                    continue
+                visited.append(part)
+                new_source = part
+                for rel in walk_rels(new_source, visited):
+                    yield rel
+
+        for rel in walk_rels(self):
+            yield rel
+
     def iter_parts(self):
         """
         Generate exactly one reference to each of the parts in the package by
 
@@ -6,25 +6,61 @@
 
 from __future__ import absolute_import, print_function, unicode_literals
 
+from docx.opc.constants import RELATIONSHIP_TYPE as RT
 from docx.opc.package import OpcPackage
+from docx.shared import lazyproperty
 
 
 class Package(OpcPackage):
     """
     Customizations specific to a WordprocessingML package.
     """
-    @property
+    def after_unmarshal(self):
+        """
+        Called by loading code after all parts and relationships have been
+        loaded, to afford the opportunity for any required post-processing.
+        """
+        self._gather_image_parts()
+
+    @lazyproperty
     def image_parts(self):
         """
         Collection of all image parts in this package.
         """
+        return ImageParts()
+
+    def _gather_image_parts(self):
+        """
+        Load the image part collection with all the image parts in package.
+        """
+        for rel in self.iter_rels():
+            if rel.is_external:
+                continue
+            if rel.reltype != RT.IMAGE:
+                continue
+            if rel.target_part in self.image_parts:
+                continue
+            self.image_parts.append(rel.target_part)
 
 
 class ImageParts(object):
     """
     Collection of |ImagePart| instances containing all the image parts in the
     package.
     """
+    def __init__(self):
+        super(ImageParts, self).__init__()
+        self._image_parts = []
+
+    def __contains__(self, item):
+        return self._image_parts.__contains__(item)
+
+    def __len__(self):
+        return self._image_parts.__len__()
+
+    def append(self, item):
+        self._image_parts.append(item)
+
     def get_or_add_image_part(self, image_descriptor):
         """
         Return an |ImagePart| instance containing the image identified by
 
@@ -0,0 +1,18 @@
+# encoding: utf-8
+
+"""
+Test suite for docx.package module
+"""
+
+from __future__ import absolute_import, print_function, unicode_literals
+
+from docx.package import Package
+
+from .unitutil import docx_path
+
+
+class DescribePackage(object):
+
+    def it_gathers_package_image_parts_after_unmarshalling(self):
+        package = Package.open(docx_path('having-images'))
+        assert len(package.image_parts) == 3
@@ -28,6 +28,18 @@ def absjoin(*paths):
     return os.path.abspath(os.path.join(*paths))
 
 
+def docx_path(name):
+    """
+    Return the absolute path to test .docx file with root name *name*.
+    """
+    return absjoin(_thisdir, 'test_files', '%s.docx' % name)
+
+
+# ===========================================================================
+# pytest mocking helpers
+# ===========================================================================
+
+
 def class_mock(request, q_class_name, autospec=True, **kwargs):
     """
     Return a mock patching the class with qualified name *q_class_name*.