00001
00005 #include "system.h"
00006
00007 #include "Python.h"
00008 #ifdef __LCLINT__
00009 #undef PyObject_HEAD
00010 #define PyObject_HEAD int _PyObjectHead;
00011 #endif
00012
00013 #include <rpmlib.h>
00014 #include <rpmdb.h>
00015
00016 #include "rpmmi-py.h"
00017 #include "header-py.h"
00018
00019 #include "debug.h"
00020
00074
00077 static PyObject *
00078 rpmmi_iter(rpmmiObject * s)
00079
00080 {
00081 Py_INCREF(s);
00082 return (PyObject *)s;
00083 }
00084
00087 static PyObject *
00088 rpmmi_iternext(rpmmiObject * s)
00089
00090
00091 {
00092 Header h;
00093
00094 if (s->mi == NULL || (h = rpmdbNextIterator(s->mi)) == NULL) {
00095 s->mi = rpmdbFreeIterator(s->mi);
00096 return NULL;
00097 }
00098 return (PyObject *) hdr_Wrap(h);
00099 }
00100
00103 static PyObject *
00104 rpmmi_Next(rpmmiObject * s, PyObject *args)
00105
00106
00107 {
00108 PyObject * result;
00109
00110 if (!PyArg_ParseTuple(args, ":Next"))
00111 return NULL;
00112
00113 result = rpmmi_iternext(s);
00114
00115 if (result == NULL) {
00116 Py_INCREF(Py_None);
00117 return Py_None;
00118 }
00119 return result;
00120 }
00121
00124 static PyObject *
00125 rpmmi_Instance(rpmmiObject * s, PyObject * args)
00126
00127 {
00128 int rc = 0;
00129
00130 if (!PyArg_ParseTuple(args, ":Instance"))
00131 return NULL;
00132
00133 if (s->mi)
00134 rc = rpmdbGetIteratorOffset(s->mi);
00135
00136 return Py_BuildValue("i", rc);
00137 }
00138
00141 static PyObject *
00142 rpmmi_Count(rpmmiObject * s, PyObject * args)
00143
00144 {
00145 int rc = 0;
00146
00147 if (!PyArg_ParseTuple(args, ":Instance"))
00148 return NULL;
00149
00150 if (s->mi)
00151 rc = rpmdbGetIteratorCount(s->mi);
00152
00153 return Py_BuildValue("i", rc);
00154 }
00155
00158 static PyObject *
00159 rpmmi_Pattern(rpmmiObject * s, PyObject * args)
00160
00161
00162 {
00163 PyObject *TagN = NULL;
00164 int type;
00165 char * pattern;
00166 rpmTag tag;
00167
00168 if (!PyArg_ParseTuple(args, "Ois:Pattern", &TagN, &type, &pattern))
00169 return NULL;
00170
00171 if ((tag = tagNumFromPyObject (TagN)) == -1) {
00172 PyErr_SetString(PyExc_TypeError, "unknown tag type");
00173 return NULL;
00174 }
00175
00176 rpmdbSetIteratorRE(s->mi, tag, type, pattern);
00177
00178 Py_INCREF (Py_None);
00179 return Py_None;
00180
00181 }
00182
00185
00186
00187 static struct PyMethodDef rpmmi_methods[] = {
00188 {"next", (PyCFunction) rpmmi_Next, METH_VARARGS,
00189 "mi.next() -> hdr\n\
00190 - Retrieve next header that matches. Iterate directly in python if possible.\n" },
00191 {"instance", (PyCFunction) rpmmi_Instance, METH_VARARGS,
00192 NULL },
00193 {"count", (PyCFunction) rpmmi_Count, METH_VARARGS,
00194 NULL },
00195 {"pattern", (PyCFunction) rpmmi_Pattern, METH_VARARGS,
00196 "mi.pattern(TagN, mire_type, pattern)\n\
00197 - Set a secondary match pattern on tags from retrieved header.\n" },
00198 {NULL, NULL}
00199 };
00200
00201
00204 static void rpmmi_dealloc( rpmmiObject * s)
00205
00206
00207 {
00208 if (s) {
00209 s->mi = rpmdbFreeIterator(s->mi);
00210 PyObject_Del(s);
00211 }
00212 }
00213
00216 static PyObject * rpmmi_getattr (rpmmiObject *s, char *name)
00217
00218 {
00219 return Py_FindMethod (rpmmi_methods, (PyObject *) s, name);
00220 }
00221
00224
00225 static char rpmmi_doc[] =
00226 "";
00227
00230
00231 PyTypeObject rpmmi_Type = {
00232 PyObject_HEAD_INIT(&PyType_Type)
00233 0,
00234 "rpm.mi",
00235 sizeof(rpmmiObject),
00236 0,
00237 (destructor) rpmmi_dealloc,
00238 0,
00239 (getattrfunc) rpmmi_getattr,
00240 0,
00241 0,
00242 0,
00243 0,
00244 0,
00245 0,
00246 0,
00247 0,
00248 0,
00249 0,
00250 0,
00251 0,
00252 Py_TPFLAGS_DEFAULT,
00253 rpmmi_doc,
00254 #if Py_TPFLAGS_HAVE_ITER
00255 0,
00256 0,
00257 0,
00258 0,
00259 (getiterfunc) rpmmi_iter,
00260 (iternextfunc) rpmmi_iternext,
00261 rpmmi_methods,
00262 0,
00263 0,
00264 0,
00265 0,
00266 0,
00267 0,
00268 0,
00269 0,
00270 0,
00271 0,
00272 0,
00273 0,
00274 #endif
00275 };
00276
00277
00278 rpmmiObject * rpmmi_Wrap(rpmdbMatchIterator mi)
00279 {
00280 rpmmiObject * mio = (rpmmiObject *) PyObject_New(rpmmiObject, &rpmmi_Type);
00281
00282 if (mio == NULL) {
00283 PyErr_SetString(pyrpmError, "out of memory creating rpmmiObject");
00284 return NULL;
00285 }
00286 mio->mi = mi;
00287 return mio;
00288 }
00289