mirrors/cpython
1
0
Fork 0
mirror of https://github.com/python/cpython.git synced 2025-08-31 14:07:50 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions 9

bpo-43901: Fix refleaks in test_module (GH-25754)

Browse source
This commit is contained in:
Pablo Galindo 2021-04-30 17:26:45 +01:00 • committed by GitHub
parent 7dcf0f6db3
commit e374a40afa
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 13 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

15
Objects/moduleobject.c
View file

@ -845,6 +845,7 @@ module_get_annotations(PyModuleObject *m, void *Py_UNUSED(ignored))
if ((dict == NULL) || !PyDict_Check(dict)) {
PyErr_Format(PyExc_TypeError, "<module>.__dict__ is not a dictionary");
Py_XDECREF(dict);
return NULL;
}
@ -876,25 +877,31 @@ module_get_annotations(PyModuleObject *m, void *Py_UNUSED(ignored))
static int
module_set_annotations(PyModuleObject *m, PyObject *value, void *Py_UNUSED(ignored))
{
int ret = -1;
PyObject *dict = _PyObject_GetAttrId((PyObject *)m, &PyId___dict__);
if ((dict == NULL) || !PyDict_Check(dict)) {
PyErr_Format(PyExc_TypeError, "<module>.__dict__ is not a dictionary");
return -1;
goto exit;
}
if (value != NULL) {
/* set */
return _PyDict_SetItemId(dict, &PyId___annotations__, value);
ret = _PyDict_SetItemId(dict, &PyId___annotations__, value);
goto exit;
}
/* delete */
if (!_PyDict_ContainsId(dict, &PyId___annotations__)) {
PyErr_Format(PyExc_AttributeError, "__annotations__");
return -1;
goto exit;
}
return _PyDict_DelItemId(dict, &PyId___annotations__);
ret = _PyDict_DelItemId(dict, &PyId___annotations__);
exit:
Py_XDECREF(dict);
return ret;
}