wip

eupharis · eupharis · commit a460f3ba9299 · 2015-11-30T17:51:37.000-08:00
diff --git a/docx/document.py b/docx/document.py
@@ -8,7 +8,12 @@
     absolute_import, division, print_function, unicode_literals
 )
 
-from .opc.constants import RELATIONSHIP_TYPE as RT
+from .oxml import OxmlElement
+from .oxml.header import CT_Header
+from .oxml.ns import qn
+from .parts.header import HeaderPart
+from .opc.constants import RELATIONSHIP_TYPE as RT, CONTENT_TYPE as CT
+from .opc.packuri import PackURI
 from .blkcntnr import BlockItemContainer
 from .enum.section import WD_SECTION
 from .enum.text import WD_BREAK
@@ -208,7 +213,39 @@ def __init__(self, body_elm, parent):
         self._body = body_elm
 
     def add_header(self, text):
-        return text
+        rel_id = 'rId9'
+        target = 'header1.xml'
+
+        # make header_ref_elm
+        header_ref_elm_tag = 'w:headerReference'
+        header_attrs = {
+            qn('r:id'): rel_id,
+            qn('w:type'): "default"
+        }
+        header_ref_elm = OxmlElement(header_ref_elm_tag, attrs=header_attrs)
+
+        # make header_elm
+        header_elm_tag = 'w:hdr'
+        # WRITE THE NEW METHOD YO WITH header_elm_tag
+        header_elm = CT_Header.new(partname, content_type)
+
+        # make header instance (wrapper around elm)
+        header = BlockItemContainer(header_elm, self)
+        header.add_paragraph(text)
+
+        # make target part
+        partname = PackURI('/word/header1.xml')
+        content_type = CT.WML_HEADER
+        target = HeaderPart(partname, content_type, header_elm, self._parent._part.package)
+
+        # TODO figure out nicer way to get this
+        RELATIONSHIPS_SCHEMA = 'http://schemas.openxmlformats.org/officeDocument/2006/relationships'
+        reltype = '%s/header' % RELATIONSHIPS_SCHEMA
+        rel = self._parent.part.rels.add_relationship(reltype, target, rel_id)
+
+        sentinel_sectPr = self._body.get_or_add_sectPr()
+        sentinel_sectPr.append(header_ref_elm)
+        return rel
 
     def remove_headers(self):
         """
diff --git a/docx/header.py b/docx/header.py
@@ -0,0 +1,15 @@
+from .blkcntnr import BlockItemContainer
+
+
+# TODO figure out if this needed?
+
+
+class Header(BlockItemContainer):
+    """
+    Proxy object wrapping ``<w:p>`` element.
+    """
+    def __init__(self, header_elm, parent):
+        super(Header, self).__init__(header_elm, parent)
+
+    # add paragraph inherited from parent
+    # add image needs to be added I think?
diff --git a/docx/oxml/header.py b/docx/oxml/header.py
@@ -0,0 +1,23 @@
+from .xmlchemy import BaseOxmlElement, ZeroOrMore
+
+
+class CT_Header(BaseOxmlElement):
+    """
+    ``<w:body>``, the container element for the main document story in
+    ``document.xml``.
+    """
+    p = ZeroOrMore('w:p', successors=())
+
+    # TODO DO THIS METHOD!
+    @staticmethod
+    def new(partname, content_type):
+        """
+        Return a new ``<Override>`` element with attributes set to parameter
+        values.
+        """
+        pass
+        # xml = '<Override xmlns="%s"/>' % nsmap['ct']
+        # override = parse_xml(xml)
+        # override.set('PartName', partname)
+        # override.set('ContentType', content_type)
+        # return override
diff --git a/docx/oxml/section.py b/docx/oxml/section.py
@@ -13,6 +13,12 @@
 from .xmlchemy import BaseOxmlElement, OptionalAttribute, ZeroOrOne
 
 
+class CT_HeaderReference(BaseOxmlElement):
+    """
+    ``<w:headerReference>``, the element for the header reference
+    """
+
+
 class CT_PageMar(BaseOxmlElement):
     """
     ``<w:pgMar>`` element, defining page margins.
diff --git a/docx/oxml/xmlchemy.py b/docx/oxml/xmlchemy.py
@@ -14,8 +14,8 @@
 from . import OxmlElement
 from ..compat import Unicode
 from .exceptions import InvalidXmlError
-from .ns import NamespacePrefixedTag, nsmap, qn
 from ..shared import lazyproperty
+from .ns import NamespacePrefixedTag, nsmap, qn
 
 
 def serialize_for_reading(element):
diff --git a/docx/parts/header.py b/docx/parts/header.py
@@ -3,3 +3,6 @@
 
 class HeaderPart(XmlPart):
     pass
+    # def __init__(self, partname, content_type, blob, package):
+    # super(HeaderPart, self).__init__(partname, content_type, blob)
+    # self._header = header
diff --git a/tests/test_header.py b/tests/test_header.py
@@ -50,13 +50,16 @@ def it_adds_a_header(self):
         document.remove_headers()
 
         header = document.add_header('foobar')
+        header_elm_tag = 'w:headerReference'
+        sentinel_sectPr = document._body._body.get_or_add_sectPr()
+        header_elms = sentinel_sectPr.findall(qn(header_elm_tag))
+        assert len(header_elms) == 1
 
         assert header
-        assert header != 'foobar'
-        # import uuid
-        # random_name = uuid.uuid4().hex
-        # finish_path = '{}.docx'.format(random_name)
-        # document.save(finish_path)
-        # print 'file {} header removed!'.format(finish_path)
+        import uuid
+        random_name = uuid.uuid4().hex
+        finish_path = '{}.docx'.format(random_name)
+        document.save(finish_path)
+        print 'file {} header removed!'.format(finish_path)
 
 # class DescribeHeaderAdd(object):
