Issue #17901: fix TreeBuilder construction for an explicit element_factory=None
Based on report and patch by Aaron Oakley.
diff --git a/Lib/test/test_xml_etree.py b/Lib/test/test_xml_etree.py
index f3683f2..7e0f25a 100644
--- a/Lib/test/test_xml_etree.py
+++ b/Lib/test/test_xml_etree.py
@@ -1762,6 +1762,12 @@
parser.feed(self.sample1)
self.assertIsNone(parser.close())
+ def test_treebuilder_elementfactory_none(self):
+ parser = ET.XMLParser(target=ET.TreeBuilder(element_factory=None))
+ parser.feed(self.sample1)
+ e = parser.close()
+ self._check_sample1_element(e)
+
def test_subclass(self):
class MyTreeBuilder(ET.TreeBuilder):
def foobar(self, x):
diff --git a/Misc/ACKS b/Misc/ACKS
index b47accf..4eb8ef6 100644
--- a/Misc/ACKS
+++ b/Misc/ACKS
@@ -887,6 +887,7 @@
Kevin O'Connor
Tim O'Malley
Zooko O'Whielacronx
+Aaron Oakley
James Oakley
Jon Oberheide
Pascal Oberndoerfer
diff --git a/Modules/_elementtree.c b/Modules/_elementtree.c
index 9caef99..a568cd3 100644
--- a/Modules/_elementtree.c
+++ b/Modules/_elementtree.c
@@ -2381,7 +2381,7 @@
self->data = NULL;
}
- if (self->element_factory) {
+ if (self->element_factory && self->element_factory != Py_None) {
node = PyObject_CallFunction(self->element_factory, "OO", tag, attrib);
} else {
node = create_new_element(tag, attrib);