1 files changed, 147 insertions, 0 deletions
diff --git a/label.c b/label.c
new file mode 100644
index 0000000..c41f59b
--- /dev/null
+++ b/label.c
@@ -0,0 +1,147 @@
+#include "fdisk.h"
+
+
+static PyMemberDef Label_members[] = {
+	{ NULL }
+};
+
+
+static void Label_dealloc(LabelObject *self)
+{
+	Py_TYPE(self)->tp_free((PyObject *) self);
+}
+
+static PyObject *Label_new(PyTypeObject *type,
+			 PyObject *args __attribute__((unused)),
+			 PyObject *kwds __attribute__((unused)))
+{
+	LabelObject *self = (LabelObject*) type->tp_alloc(type, 0);
+
+	if (self) {
+		self->lb = NULL;
+	}
+
+	return (PyObject *)self;
+}
+
+#define Label_HELP "Label(context=None)"
+static int Label_init(LabelObject *self, PyObject *args, PyObject *kwds)
+{
+	ContextObject *cxt = NULL;
+	struct fdisk_label *lb;
+	char *kwlist[] = {
+		"context",
+		NULL
+	};
+	
+	if (!PyArg_ParseTupleAndKeywords(args,
+					kwds, "|O!", kwlist,
+					&ContextType, &cxt)) {
+		PyErr_SetString(PyExc_TypeError, "Error");
+		return -1;
+	}
+
+	if (cxt && (lb = fdisk_get_label(cxt->cxt, NULL))) {
+		self->lb = lb;
+	}
+
+	return 0;
+}
+
+
+static PyMethodDef Label_methods[] = {
+	{NULL}
+};
+
+
+static PyObject *Label_get_type(LabelObject *self)
+{
+	return PyLong_FromLong(fdisk_label_get_type(self->lb));
+}
+static PyObject *Label_get_name(LabelObject *self)
+{
+	return PyObjectResultStr(fdisk_label_get_name(self->lb));
+}
+static PyGetSetDef Label_getseters[] = {
+	{"type",	(getter)Label_get_type, NULL, "label type", NULL},
+	{"name",	(getter)Label_get_name, NULL, "label name", NULL},
+	{NULL}
+};
+
+static PyObject *Label_repr(LabelObject *self)
+{
+	return PyUnicode_FromFormat("<libfdisk.Label object at %p, name=%s>",
+			self, fdisk_label_get_name(self->lb));
+}
+
+PyTypeObject LabelType = {
+	PyVarObject_HEAD_INIT(NULL, 0)
+	"libfdisk.Label", /*tp_name*/
+	sizeof(LabelObject), /*tp_basicsize*/
+	0, /*tp_itemsize*/
+	(destructor)Label_dealloc, /*tp_dealloc*/
+	0, /*tp_print*/
+	NULL, /*tp_getattr*/
+	NULL, /*tp_setattr*/
+	NULL, /*tp_compare*/
+	(reprfunc) Label_repr,
+	NULL, /*tp_as_number*/
+	NULL, /*tp_as_sequence*/
+	NULL, /*tp_as_mapping*/
+	NULL, /*tp_hash */
+	NULL, /*tp_call*/
+	NULL, /*tp_str*/
+	NULL, /*tp_getattro*/
+	NULL, /*tp_setattro*/
+	NULL, /*tp_as_buffer*/
+	Py_TPFLAGS_DEFAULT | Py_TPFLAGS_BASETYPE, /*tp_flags*/
+	Label_HELP, /* tp_doc */
+	NULL, /* tp_traverse */
+	NULL, /* tp_clear */
+	NULL, /* tp_richcompare */
+	0, /* tp_weaklistoffset */
+	NULL, /* tp_iter */
+	NULL, /* tp_iternext */
+	Label_methods, /* tp_methods */
+	Label_members, /* tp_members */
+	Label_getseters, /* tp_getset */
+	NULL, /* tp_base */
+	NULL, /* tp_dict */
+	NULL, /* tp_descr_get */
+	NULL, /* tp_descr_set */
+	0, /* tp_dictoffset */
+	(initproc)Label_init, /* tp_init */
+	NULL, /* tp_alloc */
+	Label_new, /* tp_new */
+};
+
+PyObject *PyObjectResultLabel(struct fdisk_label *lb)
+{
+        LabelObject *result;
+
+        if (!lb) {
+                PyErr_SetString(PyExc_AssertionError, "lb assert failed");
+                return NULL;
+        }
+
+
+        result = PyObject_New(LabelObject, &LabelType);
+        if (!result) {
+                PyErr_SetString(PyExc_MemoryError, "Couldn't allocate Label object");
+                return NULL;
+        }
+
+        /* Py_INCREF(result); */
+
+        result->lb = lb;
+        return (PyObject *) result;
+}
+
+void Label_AddModuleObject(PyObject *mod)
+{
+	if (PyType_Ready(&LabelType) < 0)
+		return;
+
+	Py_INCREF(&LabelType);
+	PyModule_AddObject(mod, "Label", (PyObject *)&LabelType);
+}