diff options
Diffstat (limited to 'Modules/_elementtree.c')
-rw-r--r-- | Modules/_elementtree.c | 14 |
1 files changed, 5 insertions, 9 deletions
diff --git a/Modules/_elementtree.c b/Modules/_elementtree.c index b05a39f..5a1ed8c 100644 --- a/Modules/_elementtree.c +++ b/Modules/_elementtree.c @@ -278,29 +278,25 @@ element_new(PyTypeObject *type, PyObject *args, PyObject *kwds) static PyObject* get_attrib_from_keywords(PyObject *kwds) { - PyObject *attrib_str = PyUnicode_FromString("attrib"); - PyObject *attrib = PyDict_GetItem(kwds, attrib_str); + const char* ATTRIB_KEY = "attrib"; + PyObject *attrib = PyDict_GetItemString(kwds, ATTRIB_KEY); if (attrib) { /* If attrib was found in kwds, copy its value and remove it from * kwds */ if (!PyDict_Check(attrib)) { - Py_DECREF(attrib_str); PyErr_Format(PyExc_TypeError, "attrib must be dict, not %.100s", Py_TYPE(attrib)->tp_name); return NULL; } attrib = PyDict_Copy(attrib); - PyDict_DelItem(kwds, attrib_str); + PyDict_DelItemString(kwds, ATTRIB_KEY); } else { attrib = PyDict_New(); } - - Py_DECREF(attrib_str); - - if (attrib) - PyDict_Update(attrib, kwds); + assert(attrib); + PyDict_Update(attrib, kwds); return attrib; } |