pkg: add .main_document property

Steve Canny · Steve Canny · commit 59088bda25c6 · 2013-10-20T14:18:06.000-07:00
Locates the main document part, e.g. word/document.xml for a Word
package, ppt/presentation.xml for a PowerPoint document.
diff --git a/opc/package.py b/opc/package.py
@@ -11,6 +11,7 @@
 Provides an API for manipulating Open Packaging Convention (OPC) packages.
 """
 
+from opc.constants import RELATIONSHIP_TYPE as RT
 from opc.oxml import CT_Relationships
 from opc.packuri import PACKAGE_URI
 from opc.pkgreader import PackageReader
@@ -27,6 +28,17 @@ def __init__(self):
         super(OpcPackage, self).__init__()
         self._rels = RelationshipCollection(PACKAGE_URI.baseURI)
 
+    @property
+    def main_document(self):
+        """
+        Return a reference to the main document part for this package.
+        Examples include a document part for a WordprocessingML package, a
+        presentation part for a PresentationML package, or a workbook part
+        for a SpreadsheetML package.
+        """
+        rel = self._rels.get_rel_of_type(RT.OFFICE_DOCUMENT)
+        return rel.target_part
+
     @staticmethod
     def open(pkg_file):
         """
