shr: add ElementProxy.part

Steve Canny · Steve Canny · commit 08152f9b58df · 2014-12-27T22:07:12.000-08:00
diff --git a/docx/shared.py b/docx/shared.py
@@ -172,10 +172,11 @@ class ElementProxy(object):
     type of class in python-docx other than custom element (oxml) classes.
     """
 
-    __slots__ = ('_element',)
+    __slots__ = ('_element', '_parent')
 
-    def __init__(self, element):
+    def __init__(self, element, parent=None):
         self._element = element
+        self._parent = parent
 
     def __eq__(self, other):
         """
@@ -201,6 +202,13 @@ def element(self):
         """
         return self._element
 
+    @property
+    def part(self):
+        """
+        The package part containing this object
+        """
+        return self._parent.part
+
 
 class Parented(object):
     """
diff --git a/tests/test_shared.py b/tests/test_shared.py
@@ -10,9 +10,11 @@
 
 import pytest
 
+from docx.opc.part import XmlPart
 from docx.shared import ElementProxy
 
 from .unitutil.cxml import element
+from .unitutil.mock import instance_mock
 
 
 class DescribeElementProxy(object):
@@ -32,6 +34,10 @@ def it_knows_its_element(self, element_fixture):
         proxy, element = element_fixture
         assert proxy.element is element
 
+    def it_knows_its_part(self, part_fixture):
+        proxy, part_ = part_fixture
+        assert proxy.part is part_
+
     # fixture --------------------------------------------------------
 
     @pytest.fixture
@@ -48,3 +54,19 @@ def eq_fixture(self):
         proxy_3 = ElementProxy(q)
         not_a_proxy = 'Foobar'
         return proxy, proxy_2, proxy_3, not_a_proxy
+
+    @pytest.fixture
+    def part_fixture(self, other_proxy_, part_):
+        other_proxy_.part = part_
+        proxy = ElementProxy(None, other_proxy_)
+        return proxy, part_
+
+    # fixture components ---------------------------------------------
+
+    @pytest.fixture
+    def other_proxy_(self, request):
+        return instance_mock(request, ElementProxy)
+
+    @pytest.fixture
+    def part_(self, request):
+        return instance_mock(request, XmlPart)
