python/arfile.cc: Make ArArchive iterable.

1 files changed, 9 insertions, 1 deletions

diff --git a/python/arfile.cc b/python/arfile.cc
index 1185c16a..2eb6f1cd 100644
--- a/python/arfile.cc
+++ b/python/arfile.cc
@@ -289,6 +289,14 @@ static PyObject *ararchive_getnames(PyArArchiveObject *self)
     return list;
 }
 
+// Just run getmembers() and return an iterator over the list.
+static PyObject *ararchive_iter(PyArArchiveObject *self) {
+    PyObject *members = ararchive_getmembers(self);
+    PyObject *iter = PyObject_GetIter(members);
+    Py_DECREF(members);
+    return iter;
+}
+
 PyMethodDef ararchive_methods[] = {
     {"getmember",(PyCFunction)ararchive_getmember,METH_O,
      ararchive_getmember_doc},
@@ -395,7 +403,7 @@ PyTypeObject PyArArchive_Type = {
     CppOwnedClear<ARArchive*>,           // tp_clear
     0,                                   // tp_richcompare
     0,                                   // tp_weaklistoffset
-    0,                                   // tp_iter
+    (getiterfunc)ararchive_iter,         // tp_iter
     0,                                   // tp_iternext
     ararchive_methods,                   // tp_methods
     0,                                   // tp_members