Read footnotes and store in Document

1 files changed, 24 insertions, 2 deletions

diff --git a/mammoth/documents.py b/mammoth/documents.py
index 326fbeb..ed98aaa 100644
--- a/mammoth/documents.py
+++ b/mammoth/documents.py
@@ -1,7 +1,7 @@
 import dodge
 
 
-Document = dodge.data_class("Document", ["children"])
+Document = dodge.data_class("Document", ["children", "footnotes"])
 Paragraph = dodge.data_class("Paragraph", ["children", "style_id", "style_name", "numbering"])
 Run = dodge.data_class("Run", ["children", "style_id", "style_name", "is_bold", "is_italic", "is_underline"])
 Text = dodge.data_class("Text", ["value"])
@@ -22,7 +22,10 @@ class Image(object):
         self.open = open
 
 
-document = Document
+def document(children, footnotes=None):
+    if footnotes is None:
+        footnotes = Footnotes({})
+    return Document(children, footnotes)
 
 def paragraph(children, style_id=None, style_name=None, numbering=None):
     return Paragraph(children, style_id, style_name, numbering)
@@ -49,3 +52,22 @@ def numbering_level(level_index, is_ordered):
     return _NumberingLevel(str(level_index), bool(is_ordered))
 
 _NumberingLevel = dodge.data_class("NumberingLevel", ["level_index", "is_ordered"])
+
+
+footnote = Footnote = dodge.data_class("Footnote", ["id", "body"])
+
+
+class Footnotes(object):
+    def __init__(self, footnotes):
+        self._footnotes = footnotes
+    
+    def find_footnote_by_id(self, id):
+        return self._footnotes[id]
+    
+    def __eq__(self, other):
+        return isinstance(other, Footnotes) and self._footnotes == other._footnotes
+
+    def __ne__(self, other):
+        return not (self == other)
+
+footnotes = Footnotes