python/rpmdb-py.c

Go to the documentation of this file.

#include "system.h"

#include <rpmlib.h>

#include "rpmdb-py.h"
#include "rpmmi-py.h"
#include "header-py.h"

#include "debug.h"

/*@access Header @*/


/*@null@*/
static rpmmiObject *
rpmdb_Match (rpmdbObject * s, PyObject * args, PyObject * kwds)
        /*@globals rpmGlobalMacroContext @*/
        /*@modifies s, rpmGlobalMacroContext @*/
{
    PyObject *TagN = NULL;
    char *key = NULL;
    int len = 0;
    int tag = RPMDBI_PACKAGES;
    char * kwlist[] = {"tagNumber", "key", "len", NULL};

    if (!PyArg_ParseTupleAndKeywords(args, kwds, "|Ozi", kwlist,
            &TagN, &key, &len))
        return NULL;

    if (TagN && (tag = tagNumFromPyObject (TagN)) == -1) {
        PyErr_SetString(PyExc_TypeError, "unknown tag type");
        return NULL;
    }

    return rpmmi_Wrap( rpmdbInitIterator(s->db, tag, key, len), (PyObject *)s);
}

/*@-fullinitblock@*/
/*@unchecked@*/ /*@observer@*/
static struct PyMethodDef rpmdb_methods[] = {
    {"match",       (PyCFunction) rpmdb_Match,  METH_VARARGS|METH_KEYWORDS,
"db.match([TagN, [key, [len]]]) -> mi\n\
- Create an rpm db match iterator.\n" },
    {NULL,              NULL}           /* sentinel */
};
/*@=fullinitblock@*/

static int
rpmdb_length(rpmdbObject * s)
        /*@globals rpmGlobalMacroContext @*/
        /*@modifies s, rpmGlobalMacroContext @*/
{
    rpmdbMatchIterator mi;
    int count = 0;

    mi = rpmdbInitIterator(s->db, RPMDBI_PACKAGES, NULL, 0);
    while (rpmdbNextIterator(mi) != NULL)
        count++;
    mi = rpmdbFreeIterator(mi);

    return count;
}

/*@null@*/
static hdrObject *
rpmdb_subscript(rpmdbObject * s, PyObject * key)
        /*@globals rpmGlobalMacroContext @*/
        /*@modifies s, rpmGlobalMacroContext @*/
{
    int offset;
    hdrObject * ho;
    Header h;
    rpmdbMatchIterator mi;

    if (!PyInt_Check(key)) {
        PyErr_SetString(PyExc_TypeError, "integer expected");
        return NULL;
    }

    offset = (int) PyInt_AsLong(key);

    mi = rpmdbInitIterator(s->db, RPMDBI_PACKAGES, &offset, sizeof(offset));
    if (!(h = rpmdbNextIterator(mi))) {
        mi = rpmdbFreeIterator(mi);
        PyErr_SetString(pyrpmError, "cannot read rpmdb entry");
        return NULL;
    }

    ho = hdr_Wrap(h);
    h = headerFree(h);

    return ho;
}

/*@unchecked@*/ /*@observer@*/
static PyMappingMethods rpmdb_as_mapping = {
        (inquiry) rpmdb_length,         /* mp_length */
        (binaryfunc) rpmdb_subscript,   /* mp_subscript */
        (objobjargproc)0,               /* mp_ass_subscript */
};

static void rpmdb_dealloc(rpmdbObject * s)
        /*@modifies s @*/
{
    if (s->db != NULL)
        rpmdbClose(s->db);
    PyObject_Del(s);
}

static PyObject * rpmdb_getattro(PyObject * o, PyObject * n)
        /*@*/
{
    return PyObject_GenericGetAttr(o, n);
}

static int rpmdb_setattro(PyObject * o, PyObject * n, PyObject * v)
        /*@*/
{
    return PyObject_GenericSetAttr(o, n, v);
}

/*@unchecked@*/ /*@observer@*/
static char rpmdb_doc[] =
"";

/*@-fullinitblock@*/
PyTypeObject rpmdb_Type = {
        PyObject_HEAD_INIT(&PyType_Type)
        0,                              /* ob_size */
        "rpm.db",                       /* tp_name */
        sizeof(rpmdbObject),            /* tp_size */
        0,                              /* tp_itemsize */
        (destructor) rpmdb_dealloc,     /* tp_dealloc */
        0,                              /* tp_print */
        (getattrfunc)0,                 /* tp_getattr */
        0,                              /* tp_setattr */
        0,                              /* tp_compare */
        0,                              /* tp_repr */
        0,                              /* tp_as_number */
        0,                              /* tp_as_sequence */
        &rpmdb_as_mapping,              /* tp_as_mapping */
        0,                              /* tp_hash */
        0,                              /* tp_call */
        0,                              /* tp_str */
        (getattrofunc) rpmdb_getattro,  /* tp_getattro */
        (setattrofunc) rpmdb_setattro,  /* tp_setattro */
        0,                              /* tp_as_buffer */
        Py_TPFLAGS_DEFAULT,             /* tp_flags */
        rpmdb_doc,                      /* tp_doc */
#if Py_TPFLAGS_HAVE_ITER
        0,                              /* tp_traverse */
        0,                              /* tp_clear */
        0,                              /* tp_richcompare */
        0,                              /* tp_weaklistoffset */
        0,                              /* tp_iter */
        0,                              /* tp_iternext */
        rpmdb_methods,                  /* tp_methods */
        0,                              /* tp_members */
        0,                              /* tp_getset */
        0,                              /* tp_base */
        0,                              /* tp_dict */
        0,                              /* tp_descr_get */
        0,                              /* tp_descr_set */
        0,                              /* tp_dictoffset */
        0,                              /* tp_init */
        0,                              /* tp_alloc */
        0,                              /* tp_new */
        0,                              /* tp_free */
        0,                              /* tp_is_gc */
#endif
};
/*@=fullinitblock@*/

#ifdef  _LEGACY_BINDINGS_TOO
rpmdb dbFromDb(rpmdbObject * db)
{
    return db->db;
}

rpmdbObject *
rpmOpenDB(/*@unused@*/ PyObject * self, PyObject * args, PyObject * kwds) {
    rpmdbObject * o;
    char * root = "";
    int forWrite = 0;
    char * kwlist[] = {"forWrite", "rootdir", NULL};

    if (!PyArg_ParseTupleAndKeywords(args, kwds, "|is", kwlist,
            &forWrite, &root))
        return NULL;

    o = PyObject_New(rpmdbObject, &rpmdb_Type);
    o->db = NULL;

    if (rpmdbOpen(root, &o->db, forWrite ? O_RDWR | O_CREAT: O_RDONLY, 0644)) {
        char * errmsg = "cannot open database in %s";
        char * errstr = NULL;
        int errsize;

        Py_DECREF(o);
        /* PyErr_SetString should take varargs... */
        errsize = strlen(errmsg) + *root == '\0' ? 15 /* "/var/lib/rpm" */ : strlen(root);
        errstr = alloca(errsize);
/*@-formatconst@*/
        snprintf(errstr, errsize, errmsg, *root == '\0' ? "/var/lib/rpm" : root);
/*@=formatconst@*/
        PyErr_SetString(pyrpmError, errstr);
        return NULL;
    }

    return o;
}

PyObject *
rebuildDB (/*@unused@*/ PyObject * self, PyObject * args, PyObject * kwds)
{
    char * rootDir = "";
    char * kwlist[] = {"rootdir", NULL};

    if (!PyArg_ParseTupleAndKeywords(args, kwds, "s", kwlist, &rootDir))
        return NULL;

    return Py_BuildValue("i", rpmdbRebuild(rootDir, NULL, NULL));
}
#endif

---