Add parsing of footnotes XML

2 files changed, 37 insertions, 0 deletions

diff --git a/mammoth/docx/footnotes_xml.py b/mammoth/docx/footnotes_xml.py
new file mode 100644
index 0000000..c5ea746
--- /dev/null
+++ b/mammoth/docx/footnotes_xml.py
@@ -0,0 +1,21 @@
+#~ from .. import documents
+#~ from .. import results
+from .. import lists
+#~ from .xmlparser import node_types
+#~ 
+
+def read_footnotes_xml_element(element):
+    return lists.map(
+        _read_footnote_element,
+        element.find_children("w:footnote"),
+    )
+        
+
+def _read_footnote_element(element):
+    return FootnoteElement(element.attributes["w:id"], element.children)
+
+
+class FootnoteElement(object):
+    def __init__(self, id, body):
+        self.id = id
+        self.body = body
diff --git a/tests/docx/footnotes_xml_tests.py b/tests/docx/footnotes_xml_tests.py
new file mode 100644
index 0000000..56caff0
--- /dev/null
+++ b/tests/docx/footnotes_xml_tests.py
@@ -0,0 +1,16 @@
+from nose.tools import istest, assert_equal, assert_is
+
+from mammoth.docx.xmlparser import element as xml_element
+from mammoth.docx.footnotes_xml import read_footnotes_xml_element
+
+
+@istest
+def id_and_body_of_footnote_are_read():
+    footnote_body = [xml_element("w:p")]
+    footnotes = read_footnotes_xml_element(xml_element("w:footnotes", {}, [
+        xml_element("w:footnote", {"w:id": "1"}, footnote_body),
+    ]))
+    assert_equal(1, len(footnotes))
+    assert_is(footnote_body, footnotes[0].body)
+    assert_equal("1", footnotes[0].id)
+