1 files changed, 40 insertions, 0 deletions
diff --git a/fdisk.h b/fdisk.h
new file mode 100644
index 0000000..768a832
--- /dev/null
+++ b/fdisk.h
@@ -0,0 +1,40 @@
+#ifndef UTIL_LINUX_PYLIBFDISK_H
+#define UTIL_LINUX_PYLIBFDISK_H
+
+#define PY_SSIZE_T_CLEAN
+#include <Python.h>
+#include <structmember.h>
+
+#include <libfdisk/libfdisk.h>
+
+#define CONSTRUCT_ERR	"Error during object construction"
+#define ARG_ERR		"Invalid number or type of arguments"
+
+typedef struct {
+	PyObject_HEAD
+	struct fdisk_context		*cxt;
+	struct fdisk_label		*lb;
+	struct fdisk_table		*tb;
+} ContextObject;
+
+typedef struct {
+	PyObject_HEAD
+	struct fdisk_label		*lb;
+} LabelObject;
+
+typedef struct {
+	PyObject_HEAD
+	struct fdisk_partition		*pa;
+} PartitionObject;
+
+extern PyTypeObject ContextType;
+extern PyTypeObject PartitionType;
+
+extern void Context_AddModuleObject(PyObject *mod);
+extern void Label_AddModuleObject(PyObject *mod);
+
+extern PyObject *PyObjectResultStr(const char *s);
+extern PyObject *PyObjectResultLabel(struct fdisk_label *lb);
+extern PyObject *PyObjectResultPartition(struct fdisk_partition *pa);
+
+#endif