Issue #17901: fix TreeBuilder construction for an explicit element_factory=None

eliben · eliben · commit 08231a9c6aeb · 2013-05-18T15:47:16.000-07:00
Based on report and patch by Aaron Oakley.
diff --git a/Lib/test/test_xml_etree.py b/Lib/test/test_xml_etree.py
@@ -1762,6 +1762,12 @@ class DummyBuilder(BaseDummyBuilder):
         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
@@ -887,6 +887,7 @@ John O'Connor
 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
@@ -2381,7 +2381,7 @@ treebuilder_handle_start(TreeBuilderObject* self, PyObject* tag,
         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);

Original file line number	Diff line number	Diff line change
`@@ -2381,7 +2381,7 @@ treebuilder_handle_start(TreeBuilderObject* self, PyObject* tag,`
`2381`	`2381`	`self->data = NULL;`
`2382`	`2382`	`}`
`2383`	`2383`
`2384`		`- if (self->element_factory) {`
	`2384`	`+ if (self->element_factory && self->element_factory != Py_None) {`
`2385`	`2385`	`node = PyObject_CallFunction(self->element_factory, "OO", tag, attrib);`
`2386`	`2386`	`} else {`
`2387`	`2387`	`node = create_new_element(tag, attrib);`