use contains for ListValue and KX_GameObject types (has_key is deprecated by python)

eg.
 if 'prop' in gameOb: ...
 if 'GameOb' in sce.objects: ...
This commit is contained in:
Campbell Barton 2009-06-12 12:56:12 +00:00
parent 96c5f36eff
commit 1c2ce9535c
5 changed files with 86 additions and 28 deletions

@ -209,6 +209,30 @@ static PyObject *listvalue_buffer_concat(PyObject * self, PyObject * other)
return listval_new->NewProxy(true); /* python owns this list */ return listval_new->NewProxy(true); /* python owns this list */
} }
static int listvalue_buffer_contains(PyObject *self_v, PyObject *value)
{
CListValue *self= static_cast<CListValue *>(BGE_PROXY_REF(self_v));
if (self==NULL) {
PyErr_SetString(PyExc_SystemError, "val in CList, "BGE_PROXY_ERROR_MSG);
return -1;
}
if (PyString_Check(value)) {
if (self->FindValue((const char *)PyString_AsString(value))) {
return 1;
}
}
else if (BGE_PROXY_CHECK_TYPE(value)) { /* not dict like at all but this worked before __contains__ was used */
CValue *item= static_cast<CValue *>(BGE_PROXY_REF(value));
for (int i=0; i < self->GetCount(); i++)
if (self->GetValue(i) == item) // Com
return 1;
} // not using CheckEqual
return 0;
}
static PySequenceMethods listvalue_as_sequence = { static PySequenceMethods listvalue_as_sequence = {
@ -225,6 +249,7 @@ static PySequenceMethods listvalue_as_sequence = {
NULL, /*sq_ass_item*/ NULL, /*sq_ass_item*/
NULL, /*sq_ass_slice*/ NULL, /*sq_ass_slice*/
#endif #endif
(objobjproc)listvalue_buffer_contains, /* sq_contains */
}; };
@ -264,7 +289,9 @@ PyTypeObject CListValue::Type = {
0, 0,
py_base_getattro, py_base_getattro,
py_base_setattro, py_base_setattro,
0,0,0,0,0,0,0,0,0, 0,
Py_TPFLAGS_DEFAULT,
0,0,0,0,0,0,0,
Methods Methods
}; };
@ -499,7 +526,7 @@ PyObject* CListValue::Pyreverse()
bool CListValue::CheckEqual(CValue* first,CValue* second) bool CListValue::CheckEqual(CValue* first,CValue* second)
{ {
bool result = false; bool result = false;
CValue* eqval = ((CValue*)first)->Calc(VALUE_EQL_OPERATOR,(CValue*)second); CValue* eqval = ((CValue*)first)->Calc(VALUE_EQL_OPERATOR,(CValue*)second);
if (eqval==NULL) if (eqval==NULL)
@ -528,7 +555,7 @@ PyObject* CListValue::Pyindex(PyObject *value)
for (int i=0;i<numelem;i++) for (int i=0;i<numelem;i++)
{ {
CValue* elem = GetValue(i); CValue* elem = GetValue(i);
if (CheckEqual(checkobj,elem)) if (checkobj==elem || CheckEqual(checkobj,elem))
{ {
result = PyInt_FromLong(i); result = PyInt_FromLong(i);
break; break;
@ -560,7 +587,7 @@ PyObject* CListValue::Pycount(PyObject* value)
for (int i=0;i<numelem;i++) for (int i=0;i<numelem;i++)
{ {
CValue* elem = GetValue(i); CValue* elem = GetValue(i);
if (CheckEqual(checkobj,elem)) if (checkobj==elem || CheckEqual(checkobj,elem))
{ {
numfound ++; numfound ++;
} }

@ -547,12 +547,15 @@ PyTypeObject KX_Camera::Type = {
0, 0,
0, 0,
py_base_repr, py_base_repr,
0,0, 0,
&KX_GameObject::Sequence,
&KX_GameObject::Mapping, &KX_GameObject::Mapping,
0,0,0, 0,0,0,
py_base_getattro, py_base_getattro,
py_base_setattro, py_base_setattro,
0,0,0,0,0,0,0,0,0, 0,
Py_TPFLAGS_DEFAULT,
0,0,0,0,0,0,0,
Methods Methods
}; };

@ -1287,7 +1287,7 @@ PyObject* KX_GameObject::PyGetPosition()
return PyObjectFrom(NodeGetWorldPosition()); return PyObjectFrom(NodeGetWorldPosition());
} }
PyObject *KX_GameObject::Map_GetItem(PyObject *self_v, PyObject *item) static PyObject *Map_GetItem(PyObject *self_v, PyObject *item)
{ {
KX_GameObject* self= static_cast<KX_GameObject*>BGE_PROXY_REF(self_v); KX_GameObject* self= static_cast<KX_GameObject*>BGE_PROXY_REF(self_v);
const char *attr_str= PyString_AsString(item); const char *attr_str= PyString_AsString(item);
@ -1295,7 +1295,7 @@ PyObject *KX_GameObject::Map_GetItem(PyObject *self_v, PyObject *item)
PyObject* pyconvert; PyObject* pyconvert;
if (self==NULL) { if (self==NULL) {
PyErr_SetString(PyExc_SystemError, BGE_PROXY_ERROR_MSG); PyErr_SetString(PyExc_SystemError, "val = gameOb[key]: KX_GameObject, "BGE_PROXY_ERROR_MSG);
return NULL; return NULL;
} }
@ -1321,7 +1321,7 @@ PyObject *KX_GameObject::Map_GetItem(PyObject *self_v, PyObject *item)
} }
int KX_GameObject::Map_SetItem(PyObject *self_v, PyObject *key, PyObject *val) static int Map_SetItem(PyObject *self_v, PyObject *key, PyObject *val)
{ {
KX_GameObject* self= static_cast<KX_GameObject*>BGE_PROXY_REF(self_v); KX_GameObject* self= static_cast<KX_GameObject*>BGE_PROXY_REF(self_v);
const char *attr_str= PyString_AsString(key); const char *attr_str= PyString_AsString(key);
@ -1329,7 +1329,7 @@ int KX_GameObject::Map_SetItem(PyObject *self_v, PyObject *key, PyObject *val)
PyErr_Clear(); PyErr_Clear();
if (self==NULL) { if (self==NULL) {
PyErr_SetString(PyExc_SystemError, BGE_PROXY_ERROR_MSG); PyErr_SetString(PyExc_SystemError, "gameOb[key] = value: KX_GameObject, "BGE_PROXY_ERROR_MSG);
return -1; return -1;
} }
@ -1409,11 +1409,40 @@ int KX_GameObject::Map_SetItem(PyObject *self_v, PyObject *key, PyObject *val)
return 0; /* success */ return 0; /* success */
} }
/* Cant set the len otherwise it can evaluate as false */ static int Seq_Contains(PyObject *self_v, PyObject *value)
{
KX_GameObject* self= static_cast<KX_GameObject*>BGE_PROXY_REF(self_v);
if (self==NULL) {
PyErr_SetString(PyExc_SystemError, "val in gameOb: KX_GameObject, "BGE_PROXY_ERROR_MSG);
return -1;
}
if(PyString_Check(value) && self->GetProperty(PyString_AsString(value)))
return 1;
if (self->m_attr_dict && PyDict_GetItem(self->m_attr_dict, value))
return 1;
return 0;
}
PyMappingMethods KX_GameObject::Mapping = { PyMappingMethods KX_GameObject::Mapping = {
(lenfunc)NULL , /*inquiry mp_length */ (lenfunc)NULL , /*inquiry mp_length */
(binaryfunc)KX_GameObject::Map_GetItem, /*binaryfunc mp_subscript */ (binaryfunc)Map_GetItem, /*binaryfunc mp_subscript */
(objobjargproc)KX_GameObject::Map_SetItem, /*objobjargproc mp_ass_subscript */ (objobjargproc)Map_SetItem, /*objobjargproc mp_ass_subscript */
};
PySequenceMethods KX_GameObject::Sequence = {
NULL, /* Cant set the len otherwise it can evaluate as false */
NULL, /* sq_concat */
NULL, /* sq_repeat */
NULL, /* sq_item */
NULL, /* sq_slice */
NULL, /* sq_ass_item */
NULL, /* sq_ass_slice */
(objobjproc)Seq_Contains, /* sq_contains */
}; };
PyTypeObject KX_GameObject::Type = { PyTypeObject KX_GameObject::Type = {
@ -1433,12 +1462,15 @@ PyTypeObject KX_GameObject::Type = {
0, 0,
0, 0,
py_base_repr, py_base_repr,
0,0, 0,
&Sequence,
&Mapping, &Mapping,
0,0,0, 0,0,0,
py_base_getattro, py_base_getattro,
py_base_setattro, py_base_setattro,
0,0,0,0,0,0,0,0,0, 0,
Py_TPFLAGS_DEFAULT,
0,0,0,0,0,0,0,
Methods Methods
}; };
@ -2779,16 +2811,11 @@ PyObject* KX_GameObject::Pyget(PyObject *args)
/* Matches python dict.has_key() */ /* Matches python dict.has_key() */
PyObject* KX_GameObject::Pyhas_key(PyObject* value) PyObject* KX_GameObject::Pyhas_key(PyObject* value)
{ {
if(PyString_Check(value) && GetProperty(PyString_AsString(value))) // the ONLY error case is invalid data, this is checked by the macro'd static function
Py_RETURN_TRUE; // that calls this one. but make sure Seq_Contains doesnt add extra errors later on.
return PyBool_FromLong(Seq_Contains((PyObject *)this, value));
if (m_attr_dict && PyDict_GetItem(m_attr_dict, value))
Py_RETURN_TRUE;
Py_RETURN_FALSE;
} }
/* --------------------------------------------------------------------- /* ---------------------------------------------------------------------
* Some stuff taken from the header * Some stuff taken from the header
* --------------------------------------------------------------------- */ * --------------------------------------------------------------------- */

@ -917,10 +917,8 @@ public:
static PyObject* pyattr_get_actuators(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef); static PyObject* pyattr_get_actuators(void *self_v, const KX_PYATTRIBUTE_DEF *attrdef);
/* getitem/setitem */ /* getitem/setitem */
static Py_ssize_t Map_Len(PyObject* self);
static PyMappingMethods Mapping; static PyMappingMethods Mapping;
static PyObject* Map_GetItem(PyObject *self_v, PyObject *item); static PySequenceMethods Sequence;
static int Map_SetItem(PyObject *self_v, PyObject *key, PyObject *val);
private : private :

@ -293,12 +293,15 @@ PyTypeObject KX_LightObject::Type = {
0, 0,
0, 0,
py_base_repr, py_base_repr,
0,0, 0,
&KX_GameObject::Sequence,
&KX_GameObject::Mapping, &KX_GameObject::Mapping,
0,0,0, 0,0,0,
py_base_getattro, py_base_getattro,
py_base_setattro, py_base_setattro,
0,0,0,0,0,0,0,0,0, 0,
Py_TPFLAGS_DEFAULT,
0,0,0,0,0,0,0,
Methods Methods
}; };