Add helper to convert python file objects to rpmio FD_t type

1 files changed, 22 insertions, 0 deletions

diff --git a/python/rpmfd-py.c b/python/rpmfd-py.c
new file mode 100644
index 000000000..2deb06793
--- /dev/null
+++ b/python/rpmfd-py.c
@@ -0,0 +1,22 @@
+
+#include "rpmsystem-py.h"
+#include "rpmfd-py.h"
+
+FD_t rpmFdFromPyObject(PyObject *obj)
+{
+    FD_t fd = NULL;
+
+    if (PyInt_Check(obj)) {
+	fd = fdDup(PyInt_AsLong(obj));
+    } else if (PyFile_Check(obj)) {
+	FILE *fp = PyFile_AsFile(obj);
+	fd = fdDup(fileno(fp));
+    } else {
+	PyErr_SetString(PyExc_TypeError, "integer or file object expected");
+	return NULL;
+    }
+    if (fd == NULL || Ferror(fd)) {
+	PyErr_SetFromErrno(PyExc_IOError);
+    }
+    return fd;
+}