diff options
Diffstat (limited to 'defusedxml/common.py')
| -rw-r--r-- | defusedxml/common.py | 51 |
1 files changed, 30 insertions, 21 deletions
diff --git a/defusedxml/common.py b/defusedxml/common.py index 1995d61..9dcdc5a 100644 --- a/defusedxml/common.py +++ b/defusedxml/common.py @@ -12,7 +12,7 @@ PY3 = sys.version_info[0] == 3 # Fail early when pyexpat is not installed correctly if not hasattr(xml.parsers.expat, "ParserCreate"): - raise ImportError('pyexpat') + raise ImportError("pyexpat") class DefusedXmlException(ValueError): @@ -90,35 +90,44 @@ def _apply_defusing(defused_mod): return stdlib_mod -def _generate_etree_functions(DefusedXMLParser, _TreeBuilder, - _parse, _iterparse): +def _generate_etree_functions(DefusedXMLParser, _TreeBuilder, _parse, _iterparse): """Factory for functions needed by etree, dependent on whether cElementTree or ElementTree is used.""" - def parse(source, parser=None, forbid_dtd=False, forbid_entities=True, - forbid_external=True): + def parse(source, parser=None, forbid_dtd=False, forbid_entities=True, forbid_external=True): if parser is None: - parser = DefusedXMLParser(target=_TreeBuilder(), - forbid_dtd=forbid_dtd, - forbid_entities=forbid_entities, - forbid_external=forbid_external) + parser = DefusedXMLParser( + target=_TreeBuilder(), + forbid_dtd=forbid_dtd, + forbid_entities=forbid_entities, + forbid_external=forbid_external, + ) return _parse(source, parser) - def iterparse(source, events=None, parser=None, forbid_dtd=False, - forbid_entities=True, forbid_external=True): + def iterparse( + source, + events=None, + parser=None, + forbid_dtd=False, + forbid_entities=True, + forbid_external=True, + ): if parser is None: - parser = DefusedXMLParser(target=_TreeBuilder(), - forbid_dtd=forbid_dtd, - forbid_entities=forbid_entities, - forbid_external=forbid_external) + parser = DefusedXMLParser( + target=_TreeBuilder(), + forbid_dtd=forbid_dtd, + forbid_entities=forbid_entities, + forbid_external=forbid_external, + ) return _iterparse(source, events, parser) - def fromstring(text, forbid_dtd=False, forbid_entities=True, - forbid_external=True): - parser = DefusedXMLParser(target=_TreeBuilder(), - forbid_dtd=forbid_dtd, - forbid_entities=forbid_entities, - forbid_external=forbid_external) + def fromstring(text, forbid_dtd=False, forbid_entities=True, forbid_external=True): + parser = DefusedXMLParser( + target=_TreeBuilder(), + forbid_dtd=forbid_dtd, + forbid_entities=forbid_entities, + forbid_external=forbid_external, + ) parser.feed(text) return parser.close() |