From: Panu Matilainen <pmatilai@redhat.com>
Date: Wed, 30 Sep 2009 11:51:33 +0000 (+0300)
Subject: Add iterator support to python header objects
X-Git-Tag: rpm-4.8.0-beta1~171
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=32b0bba8a2fb5e534bddc9c02ff170726457bdcc;p=platform%2Fupstream%2Frpm.git

Add iterator support to python header objects
---

diff --git a/python/header-py.c b/python/header-py.c
index 32ec7bd..a2c71a6 100644
--- a/python/header-py.c
+++ b/python/header-py.c
@@ -129,7 +129,8 @@
 struct hdrObject_s {
     PyObject_HEAD
     Header h;
-} ;
+    HeaderIterator hi;
+};
 
 static PyObject * hdrKeyList(hdrObject * s)
 {
@@ -405,6 +406,23 @@ static void hdr_dealloc(hdrObject * s)
     s->ob_type->tp_free((PyObject *)s);
 }
 
+static PyObject * hdr_iternext(hdrObject *s)
+{
+    PyObject *res = NULL;
+    rpmTag tag;
+
+    if (s->hi == NULL) s->hi = headerInitIterator(s->h);
+
+    if ((tag = headerNextTag(s->hi)) != RPMTAG_NOT_FOUND) {
+	int raw = 1;
+	res = PyObject_Call((PyObject *) &rpmtd_Type,
+			    Py_BuildValue("(Oii)", s, tag, raw), NULL);
+    } else {
+	s->hi = headerFreeIterator(s->hi);
+    }
+    return res;
+}
+
 int tagNumFromPyObject (PyObject *item, rpmTag *tagp)
 {
     rpmTag tag = RPMTAG_NOT_FOUND;
@@ -501,8 +519,8 @@ PyTypeObject hdr_Type = {
 	0,				/* tp_clear */
 	0,				/* tp_richcompare */
 	0,				/* tp_weaklistoffset */
-	0,				/* tp_iter */
-	0,				/* tp_iternext */
+	PyObject_SelfIter,		/* tp_iter */
+	hdr_iternext,			/* tp_iternext */
 	hdr_methods,			/* tp_methods */
 	0,				/* tp_members */
 	0,				/* tp_getset */