add v1 HeaderPart

eupharis · eupharis · commit 9c0544a02a46 · 2015-11-30T09:55:22.000-08:00
diff --git a/.gitignore b/.gitignore
@@ -6,3 +6,4 @@
 _scratch/
 Session.vim
 /.tox/
+tags
diff --git a/docx/__init__.py b/docx/__init__.py
@@ -15,6 +15,7 @@
 from docx.parts.image import ImagePart
 from docx.parts.numbering import NumberingPart
 from docx.parts.styles import StylesPart
+from docx.parts.header import HeaderPart
 
 
 def part_class_selector(content_type, reltype):
@@ -28,6 +29,8 @@ def part_class_selector(content_type, reltype):
 PartFactory.part_type_for[CT.WML_DOCUMENT_MAIN] = DocumentPart
 PartFactory.part_type_for[CT.WML_NUMBERING] = NumberingPart
 PartFactory.part_type_for[CT.WML_STYLES] = StylesPart
+PartFactory.part_type_for[CT.WML_HEADER] = HeaderPart
+# PartFactory.part_type_for[CT.WML_FOOTER] = FooterPart
 
 del (
     CT, CorePropertiesPart, DocumentPart, NumberingPart, PartFactory,
diff --git a/docx/parts/header.py b/docx/parts/header.py
@@ -0,0 +1,5 @@
+from ..opc.part import XmlPart
+
+
+class HeaderPart(XmlPart):
+    pass
diff --git a/tests/test_header.py b/tests/test_header.py
@@ -0,0 +1,16 @@
+# import pytest
+from unitutil.file import absjoin, test_file_dir
+from docx.api import Document
+from docx.opc.constants import CONTENT_TYPE as CT
+from docx.parts.header import HeaderPart
+
+
+dir_pkg_path = absjoin(test_file_dir, 'expanded_docx')
+
+
+class DescribeHeader(object):
+    def it_loads_header_as_header_part(self):
+        document = Document(dir_pkg_path)
+        for rel_id, part in document.part.related_parts.items():
+            if part.content_type == CT.WML_HEADER:
+                assert isinstance(part, HeaderPart)

-Original file line number
+Diff line change
@@ @@ -0,0 +1,5 @@ @@
 +from ..opc.part import XmlPart
++
++
 +class HeaderPart(XmlPart):
 +    pass