diff options
Diffstat (limited to 'Objects/frameobject.c')
| -rw-r--r-- | Objects/frameobject.c | 9 |
1 files changed, 3 insertions, 6 deletions
diff --git a/Objects/frameobject.c b/Objects/frameobject.c index 8e78f0f..6e66d23 100644 --- a/Objects/frameobject.c +++ b/Objects/frameobject.c @@ -283,12 +283,9 @@ dict_to_map(PyObject *map, int nmap, PyObject *dict, PyObject **values, PyObject *value = PyDict_GetItem(dict, key); Py_XINCREF(value); if (deref) { - if (value) { + if (value || clear) { if (PyCell_Set(values[j], value) < 0) PyErr_Clear(); - } else if (clear) { - Py_XDECREF(values[j]); - values[j] = value; } } else if (value != NULL || clear) { Py_XDECREF(values[j]); @@ -370,10 +367,10 @@ PyFrame_LocalsToFast(PyFrameObject *f, int clear) return; dict_to_map(f->f_code->co_cellvars, PyTuple_GET_SIZE(f->f_code->co_cellvars), - locals, fast, 1, clear); + locals, fast + f->f_nlocals, 1, clear); dict_to_map(f->f_code->co_freevars, PyTuple_GET_SIZE(f->f_code->co_freevars), - locals, fast, 1, clear); + locals, fast + f->f_nlocals + f->f_ncells, 1, clear); } PyErr_Restore(error_type, error_value, error_traceback); } |