diff options
Diffstat (limited to 'Objects')
| -rw-r--r-- | Objects/setobject.c | 18 |
1 files changed, 17 insertions, 1 deletions
diff --git a/Objects/setobject.c b/Objects/setobject.c index 88d5640..2d77c74 100644 --- a/Objects/setobject.c +++ b/Objects/setobject.c @@ -104,7 +104,23 @@ set_len(PySetObject *so) static int set_contains(PySetObject *so, PyObject *key) { - return PySequence_Contains(so->data, key); + PyObject *olddict; + PySetObject *tmp; + int result; + + result = PySequence_Contains(so->data, key); + if (result == -1 && PyType_IsSubtype(key->ob_type, &PySet_Type)) { + PyErr_Clear(); + tmp = (PySetObject *)make_new_set(&PyFrozenSet_Type, NULL); + if (tmp == NULL) + return -1; + olddict = tmp->data; + tmp->data = ((PySetObject *)(key))->data; + result = PySequence_Contains(so->data, (PyObject *)tmp); + tmp->data = olddict; + Py_DECREF(tmp); + } + return result; } static PyObject * |