And the gdbm module's test now passes again.

This commit is contained in:
Guido van Rossum 2006-08-19 23:18:48 +00:00
parent ef67111634
commit 0da7e03e12
1 changed files with 33 additions and 13 deletions

View File

@ -241,21 +241,41 @@ dbm_keys(register dbmobject *dp, PyObject *unused)
return v; return v;
} }
PyDoc_STRVAR(dbm_contains__doc__, static int
"__contains__(key) -> bool\n\ dbm_contains(PyObject *self, PyObject *arg)
Find out whether or not the database contains a given key.");
static PyObject *
dbm_contains(register dbmobject *dp, PyObject *args)
{ {
dbmobject *dp = (dbmobject *)self;
datum key; datum key;
if (!PyArg_ParseTuple(args, "s#:contains", &key.dptr, &key.dsize)) if ((dp)->di_dbm == NULL) {
return NULL; PyErr_SetString(DbmError,
check_dbmobject_open(dp); "GDBM object has already been closed");
return PyInt_FromLong((long) gdbm_exists(dp->di_dbm, key)); return -1;
}
if (!PyString_Check(arg)) {
PyErr_Format(PyExc_TypeError,
"gdbm key must be string, not %.100s",
arg->ob_type->tp_name);
return -1;
}
key.dptr = PyString_AS_STRING(arg);
key.dsize = PyString_GET_SIZE(arg);
return gdbm_exists(dp->di_dbm, key);
} }
static PySequenceMethods dbm_as_sequence = {
0, /* sq_length */
0, /* sq_concat */
0, /* sq_repeat */
0, /* sq_item */
0, /* sq_slice */
0, /* sq_ass_item */
0, /* sq_ass_slice */
dbm_contains, /* sq_contains */
0, /* sq_inplace_concat */
0, /* sq_inplace_repeat */
};
PyDoc_STRVAR(dbm_firstkey__doc__, PyDoc_STRVAR(dbm_firstkey__doc__,
"firstkey() -> key\n\ "firstkey() -> key\n\
It's possible to loop over every key in the database using this method\n\ It's possible to loop over every key in the database using this method\n\
@ -355,7 +375,6 @@ dbm_sync(register dbmobject *dp, PyObject *unused)
static PyMethodDef dbm_methods[] = { static PyMethodDef dbm_methods[] = {
{"close", (PyCFunction)dbm_close, METH_NOARGS, dbm_close__doc__}, {"close", (PyCFunction)dbm_close, METH_NOARGS, dbm_close__doc__},
{"keys", (PyCFunction)dbm_keys, METH_NOARGS, dbm_keys__doc__}, {"keys", (PyCFunction)dbm_keys, METH_NOARGS, dbm_keys__doc__},
{"__contains__",(PyCFunction)dbm_contains,METH_VARARGS, dbm_contains__doc__},
{"firstkey", (PyCFunction)dbm_firstkey,METH_NOARGS, dbm_firstkey__doc__}, {"firstkey", (PyCFunction)dbm_firstkey,METH_NOARGS, dbm_firstkey__doc__},
{"nextkey", (PyCFunction)dbm_nextkey, METH_VARARGS, dbm_nextkey__doc__}, {"nextkey", (PyCFunction)dbm_nextkey, METH_VARARGS, dbm_nextkey__doc__},
{"reorganize",(PyCFunction)dbm_reorganize,METH_NOARGS, dbm_reorganize__doc__}, {"reorganize",(PyCFunction)dbm_reorganize,METH_NOARGS, dbm_reorganize__doc__},
@ -382,7 +401,7 @@ static PyTypeObject Dbmtype = {
0, /*tp_compare*/ 0, /*tp_compare*/
0, /*tp_repr*/ 0, /*tp_repr*/
0, /*tp_as_number*/ 0, /*tp_as_number*/
0, /*tp_as_sequence*/ &dbm_as_sequence, /*tp_as_sequence*/
&dbm_as_mapping, /*tp_as_mapping*/ &dbm_as_mapping, /*tp_as_mapping*/
0, /*tp_hash*/ 0, /*tp_hash*/
0, /*tp_call*/ 0, /*tp_call*/
@ -498,7 +517,8 @@ PyMODINIT_FUNC
initgdbm(void) { initgdbm(void) {
PyObject *m, *d, *s; PyObject *m, *d, *s;
Dbmtype.ob_type = &PyType_Type; if (PyType_Ready(&Dbmtype) < 0)
return;
m = Py_InitModule4("gdbm", dbmmodule_methods, m = Py_InitModule4("gdbm", dbmmodule_methods,
gdbmmodule__doc__, (PyObject *)NULL, gdbmmodule__doc__, (PyObject *)NULL,
PYTHON_API_VERSION); PYTHON_API_VERSION);