diff options
Diffstat (limited to 'defusedxml/ElementTree.py')
| -rw-r--r-- | defusedxml/ElementTree.py | 8 |
1 files changed, 5 insertions, 3 deletions
diff --git a/defusedxml/ElementTree.py b/defusedxml/ElementTree.py index 7209661..15c179f 100644 --- a/defusedxml/ElementTree.py +++ b/defusedxml/ElementTree.py @@ -9,6 +9,7 @@ from __future__ import print_function, absolute_import import sys import warnings +from xml.etree.ElementTree import ParseError from xml.etree.ElementTree import TreeBuilder as _TreeBuilder from xml.etree.ElementTree import parse as _parse from xml.etree.ElementTree import tostring @@ -51,12 +52,13 @@ def _get_py3_cls(): _XMLParser = pure_pymod.XMLParser _iterparse = pure_pymod.iterparse - ParseError = pure_pymod.ParseError + # patch pure module to use ParseError from C extension + pure_pymod.ParseError = ParseError - return _XMLParser, _iterparse, ParseError + return _XMLParser, _iterparse -_XMLParser, _iterparse, ParseError = _get_py3_cls() +_XMLParser, _iterparse = _get_py3_cls() _sentinel = object() |