fdisk: add set_PyErr_from_rc

python-libfdisk raises Python exceptions when the libfdisk reports an
error when executing some function. libfdisk returns negative
errno values when reporting some error.

Adds utility function to set PyErr string based on the strerror of
a given errno code. Useful when raising Python exceptions.

1 files changed, 19 insertions, 0 deletions

diff --git a/fdisk.c b/fdisk.c
index d7d3fe4..d78e6e2 100644
--- a/fdisk.c
+++ b/fdisk.c
@@ -14,6 +14,25 @@
 
 #include "fdisk.h"
 
+void *set_PyErr_from_rc(int e)
+{
+	switch (e) {
+	case ENOMEM:
+		PyErr_SetString(PyExc_MemoryError, strerror(e));
+		break;
+	case EINVAL:
+		PyErr_SetString(PyExc_TypeError, strerror(e));
+		break;
+	case ENOSYS:
+		PyErr_SetString(PyExc_NotImplementedError, strerror(e));
+		break;
+	default:
+		PyErr_SetString(PyExc_Exception, strerror(e));
+	}
+
+	return NULL;
+}
+
 PyObject *PyObjectResultStr(const char *s)
 {
 	PyObject *result;