mirror of
https://github.com/samba-team/samba.git
synced 2025-01-10 01:18:15 +03:00
pytdb: Add support for tdb_add_flags() & tdb_remove_flags()
Note, unlike tdb_open where flags is `int', tdb_{add,remove}_flags want flags as `unsigned', so instead of "i" I used "I" in PyArg_ParseTuple. Cc: 597386@bugs.debian.org Signed-off-by: Kirill Smelkov <kirr@mns.spb.ru> Signed-off-by: Jelmer Vernooij <jelmer@samba.org>
This commit is contained in:
parent
92eccf89a1
commit
c4b1971259
@ -259,6 +259,27 @@ static PyObject *obj_store(PyTdbObject *self, PyObject *args)
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
static PyObject *obj_add_flags(PyTdbObject *self, PyObject *args)
|
||||
{
|
||||
unsigned flags;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "I", &flags))
|
||||
return NULL;
|
||||
|
||||
tdb_add_flags(self->ctx, flags);
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
static PyObject *obj_remove_flags(PyTdbObject *self, PyObject *args)
|
||||
{
|
||||
unsigned flags;
|
||||
|
||||
if (!PyArg_ParseTuple(args, "I", &flags))
|
||||
return NULL;
|
||||
|
||||
tdb_remove_flags(self->ctx, flags);
|
||||
Py_RETURN_NONE;
|
||||
}
|
||||
|
||||
typedef struct {
|
||||
PyObject_HEAD
|
||||
@ -341,6 +362,8 @@ static PyMethodDef tdb_object_methods[] = {
|
||||
"Check whether key exists in this database." },
|
||||
{ "store", (PyCFunction)obj_store, METH_VARARGS, "S.store(key, data, flag=REPLACE) -> None"
|
||||
"Store data." },
|
||||
{ "add_flags", (PyCFunction)obj_add_flags, METH_VARARGS, "S.add_flags(flags) -> None" },
|
||||
{ "remove_flags", (PyCFunction)obj_remove_flags, METH_VARARGS, "S.remove_flags(flags) -> None" },
|
||||
{ "iterkeys", (PyCFunction)tdb_object_iter, METH_NOARGS, "S.iterkeys() -> iterator" },
|
||||
{ "clear", (PyCFunction)obj_clear, METH_NOARGS, "S.clear() -> None\n"
|
||||
"Wipe the entire database." },
|
||||
|
@ -132,6 +132,10 @@ class SimpleTdbTests(TestCase):
|
||||
self.tdb["entry"] = "value"
|
||||
self.assertEquals(1, len(list(self.tdb)))
|
||||
|
||||
def test_add_flags(self):
|
||||
self.tdb.add_flags(tdb.NOMMAP)
|
||||
self.tdb.remove_flags(tdb.NOMMAP)
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
import unittest
|
||||
|
Loading…
Reference in New Issue
Block a user