[3.14] gh-133778: Fix setting __annotations__ under PEP 563 (GH-133794) (#134655)

gh-133778: Fix setting `__annotations__` under PEP 563 (GH-133794)
(cherry picked from commit 4443110c34)

Co-authored-by: Jelle Zijlstra <jelle.zijlstra@gmail.com>
This commit is contained in:
Miss Islington (bot) 2025-05-25 18:04:22 +02:00 committed by GitHub
parent 93aee568c0
commit d82d445b18
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
3 changed files with 60 additions and 9 deletions

View file

@ -2065,19 +2065,46 @@ type_set_annotations(PyObject *tp, PyObject *value, void *Py_UNUSED(closure))
return -1;
}
int result;
PyObject *dict = PyType_GetDict(type);
if (value != NULL) {
/* set */
result = PyDict_SetItem(dict, &_Py_ID(__annotations_cache__), value);
} else {
/* delete */
result = PyDict_Pop(dict, &_Py_ID(__annotations_cache__), NULL);
if (result == 0) {
PyErr_SetString(PyExc_AttributeError, "__annotations__");
int result = PyDict_ContainsString(dict, "__annotations__");
if (result < 0) {
Py_DECREF(dict);
return -1;
}
if (result) {
// If __annotations__ is currently in the dict, we update it,
if (value != NULL) {
result = PyDict_SetItem(dict, &_Py_ID(__annotations__), value);
} else {
result = PyDict_Pop(dict, &_Py_ID(__annotations__), NULL);
if (result == 0) {
// Somebody else just deleted it?
PyErr_SetString(PyExc_AttributeError, "__annotations__");
Py_DECREF(dict);
return -1;
}
}
if (result < 0) {
Py_DECREF(dict);
return -1;
}
// Also clear __annotations_cache__ just in case.
result = PyDict_Pop(dict, &_Py_ID(__annotations_cache__), NULL);
}
else {
// Else we update only __annotations_cache__.
if (value != NULL) {
/* set */
result = PyDict_SetItem(dict, &_Py_ID(__annotations_cache__), value);
} else {
/* delete */
result = PyDict_Pop(dict, &_Py_ID(__annotations_cache__), NULL);
if (result == 0) {
PyErr_SetString(PyExc_AttributeError, "__annotations__");
Py_DECREF(dict);
return -1;
}
}
}
if (result < 0) {
Py_DECREF(dict);