Add parsing for numbering.xml

2 files changed, 65 insertions, 0 deletions

diff --git a/mammoth/docx/numbering_xml.py b/mammoth/docx/numbering_xml.py
new file mode 100644
index 0000000..77400c1
--- /dev/null
+++ b/mammoth/docx/numbering_xml.py
@@ -0,0 +1,62 @@
+def read_numbering_xml_element(element):
+    abstract_nums = _read_abstract_nums(element)
+    nums = _read_nums(element, abstract_nums)
+    return Numbering(nums)
+
+
+def _read_abstract_nums(element):
+    abstract_num_elements = element.find_children("w:abstractNum")
+    return dict(map(_read_abstract_num, abstract_num_elements))
+
+
+def _read_abstract_num(element):
+    abstract_num_id = element.attributes.get("w:abstractNumId")
+    levels = _read_abstract_num_levels(element)
+    return abstract_num_id, levels
+
+
+def _read_abstract_num_levels(element):
+    levels = map(_read_abstract_num_level, element.find_children("w:lvl"))
+    return dict(
+        (level.level_index, level)
+        for level in levels
+    )
+
+
+def _read_abstract_num_level(element):
+    level_index = element.attributes["w:ilvl"]
+    num_fmt = element.find_child_or_null("w:numFmt").attributes.get("w:val")
+    is_ordered = num_fmt != "bullet"
+    return NumberingLevel(level_index, is_ordered)
+
+
+def _read_nums(element, abstract_nums):
+    num_elements = element.find_children("w:num")
+    return dict(
+        _read_num(num_element, abstract_nums)
+        for num_element in num_elements
+    )
+
+
+def _read_num(element, abstract_nums):
+    num_id = element.attributes.get("w:numId")
+    abstract_num_id = element.find_child_or_null("w:abstractNumId").attributes["w:val"]
+    return num_id, abstract_nums[abstract_num_id]
+
+
+class Numbering(object):
+    def __init__(self, nums):
+        self._nums = nums
+    
+    def find_level(self, num_id, level):
+        num = self._nums.get(num_id)
+        if num is None:
+            return None
+        else:
+            return num[level]
+
+
+class NumberingLevel(object):
+    def __init__(self, level_index, is_ordered):
+        self.level_index = level_index
+        self.is_ordered = is_ordered
diff --git a/mammoth/docx/xmlparser.py b/mammoth/docx/xmlparser.py
index d58ea75..efeafcf 100644
--- a/mammoth/docx/xmlparser.py
+++ b/mammoth/docx/xmlparser.py
@@ -9,6 +9,9 @@ class XmlElement(XmlElementBase):
             if child.name == name:
                 return child
         return _null_xml_element
+    
+    def find_children(self, name):
+        return filter(lambda child: child.name == name, self.children)
 
 
 class NullXmlElement(object):