summaryrefslogtreecommitdiffstats
path: root/Objects/setobject.c
diff options
context:
space:
mode:
Diffstat (limited to 'Objects/setobject.c')
-rw-r--r--Objects/setobject.c18
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 *