And the gdbm module's test now passes again.
This commit is contained in:
parent
ef67111634
commit
0da7e03e12
|
@ -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);
|
||||||
|
|
Loading…
Reference in New Issue