Permit setting and getting ts keyring from python

diff --git a/python/rpmts-py.c b/python/rpmts-py.c
index 0012660..c1c9779 100644 (file)
--- a/python/rpmts-py.c
+++ b/python/rpmts-py.c
@@ -9,6 +9,7 @@
 #include "header-py.h"
 #include "rpmds-py.h"  /* XXX for rpmdsNew */
 #include "rpmfd-py.h"
+#include "rpmkeyring-py.h"
 #include "rpmfi-py.h"  /* XXX for rpmfiNew */
 #include "rpmmi-py.h"
 #include "rpmps-py.h"
@@ -420,6 +421,29 @@ rpmts_PgpImportPubkey(rpmtsObject * s, PyObject * args, PyObject * kwds)
     return Py_BuildValue("i", rc);
 }
 
+static PyObject *rpmts_setKeyring(rpmtsObject *s, PyObject *arg)
+{
+    rpmKeyring keyring = NULL;
+    if (!PyArg_Parse(arg, "O&", rpmKeyringFromPyObject, &keyring))
+       return NULL;
+
+    return PyBool_FromLong(rpmtsSetKeyring(s->ts, keyring) == 0);
+}
+
+static PyObject *rpmts_getKeyring(rpmtsObject *s, PyObject *args, PyObject *kwds)
+{
+    rpmKeyring keyring = NULL;
+    int autoload = 1;
+    char * kwlist[] = { "autoload", NULL };
+
+    if (!PyArg_ParseTupleAndKeywords(args, kwds, "|i:getKeyring",
+                                    kwlist, &autoload))
+       return NULL;
+
+    keyring = rpmtsGetKeyring(s->ts, autoload);
+    return rpmKeyring_Wrap(&rpmKeyring_Type, keyring);
+}
+
 static void *
 rpmtsCallback(const void * hd, const rpmCallbackType what,
                         const rpm_loff_t amount, const rpm_loff_t total,
@@ -632,6 +656,10 @@ static struct PyMethodDef rpmts_methods[] = {
        NULL },
  {"pgpImportPubkey",   (PyCFunction) rpmts_PgpImportPubkey,    METH_VARARGS|METH_KEYWORDS,
        NULL },
+ {"getKeyring",        (PyCFunction) rpmts_getKeyring, METH_VARARGS|METH_KEYWORDS, 
+       NULL },
+ {"setKeyring",        (PyCFunction) rpmts_setKeyring, METH_O, 
+       NULL },
  {"dbMatch",   (PyCFunction) rpmts_Match,      METH_VARARGS|METH_KEYWORDS,
 "ts.dbMatch([TagN, [key, [len]]]) -> mi\n\
 - Create a match iterator for the default transaction rpmdb.\n" },