Fixed issue where XML comments caused exceptions to be thrown. Fixes #64

.
STIXProject · Feb 25, 2014 · 40ec36e · 40ec36e
1 parent 3b03c61
commit 40ec36e
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/stix/utils/parser.py b/stix/utils/parser.py
@@ -68,9 +68,8 @@ def parse_xml_to_obj(self, xml_file, check_version=True, check_root=True):
         check_root -- Inspect the root element before parsing.
 
         """
-        tree = None
-        if check_version or check_root:
-            tree = etree.parse(xml_file)
+        parser = etree.ETCompatXMLParser(huge_tree=True)
+        tree = etree.parse(xml_file, parser=parser)
 
         if check_version:
             self._check_version(tree)
@@ -93,9 +92,8 @@ def parse_xml(self, xml_file, check_version=True, check_root=True):
         check_root -- Inspect the root element before parsing.
 
         """
-        tree = None
-        if check_version or check_root:
-            tree = etree.parse(xml_file)
+        parser = etree.ETCompatXMLParser(huge_tree=True)
+        tree = etree.parse(xml_file, parser=parser)
 
         if check_version:
             self._check_version(tree)