Initial commit

Add sources, setup.py and .gitignore

Build/Install:

	python setup.py build
	python setup.py install

1 files changed, 237 insertions, 0 deletions

diff --git a/context.c b/context.c
new file mode 100644
index 0000000..8c0d3fa
--- /dev/null
+++ b/context.c
@@ -0,0 +1,237 @@
+#include "fdisk.h"
+
+static PyMemberDef Context_members[] = {
+	{ NULL }
+};
+
+
+static void Context_dealloc(ContextObject *self)
+{
+	if (!self->cxt) /* if init fails */
+		return;
+
+	fdisk_unref_context(self->cxt);
+	Py_TYPE(self)->tp_free((PyObject *) self);
+}
+
+static PyObject *Context_new(PyTypeObject *type,
+			 PyObject *args __attribute__((unused)),
+			 PyObject *kwds __attribute__((unused)))
+{
+	ContextObject *self = (ContextObject*) type->tp_alloc(type, 0);
+
+	if (self) {
+		self->cxt = NULL;
+		self->lb = NULL;
+		self->tb = NULL;
+	}
+
+	return (PyObject *)self;
+}
+
+#define Context_HELP "Context(device=None, details=None)"
+static int Context_init(ContextObject *self, PyObject *args, PyObject *kwds)
+{
+	char *device = NULL;
+	int details, rc = 0;
+	char *kwlist[] = {
+		"device", "details",
+		NULL
+	};
+
+	if (!PyArg_ParseTupleAndKeywords(args,
+					kwds, "|sp", kwlist,
+					&device, &details)) {
+		PyErr_SetString(PyExc_TypeError, ARG_ERR);
+		return -1;
+	}
+
+	if (self->cxt)
+		fdisk_unref_context(self->cxt);
+
+	self->cxt = fdisk_new_context();
+	if (!self->cxt) {
+		PyErr_SetString(PyExc_MemoryError, "Couldn't allocate context");
+		return -1;
+	}
+
+	/* XXX: always readonly */
+	if (device && (rc = fdisk_assign_device(self->cxt, device, 1)))
+		return -1;
+	if (details && (rc = fdisk_enable_details(self->cxt, details)))
+		return -1;
+
+	fdisk_get_partitions(self->cxt, &self->tb);
+
+	return 0;
+}
+
+
+#define Context_assign_device_HELP "assign_device(device)\n\n" \
+	"Open the device, discovery topology, geometry, detect disklabel " \
+	"and switch the current label driver to reflect the probing result. "
+static PyObject *Context_assign_device(ContextObject *self, PyObject *args, PyObject *kwds)
+{
+	char *fname;
+
+	if (!PyArg_ParseTuple(args, "s", &fname)) {
+		PyErr_SetString(PyExc_TypeError, ARG_ERR);
+		return NULL;
+	}
+
+	/* Assert self->cxt */
+
+	/* XXX: readonly */
+	fdisk_assign_device(self->cxt, fname, 1);
+
+	self->lb = fdisk_get_label(self->cxt, NULL);
+	fdisk_get_partitions(self->cxt, &self->tb);
+
+	/* XXX: check rc*/
+	Py_INCREF(Py_None);
+	return Py_None;
+}
+#define Context_partition_to_string_HELP "partition_to_string(pa, field)\n\n" \
+	"Retrieve partition field using fdisk_partition_to_string." \
+	"Field constants are available as FDISK_LABEL_*"
+static PyObject *Context_partition_to_string(ContextObject *self, PyObject *args, PyObject *kwds)
+{
+	struct fdisk_partition *pa;
+	enum fdisk_fieldtype field;
+	PartitionObject *part;
+	PyObject *ret;
+	char *data;
+
+	if (!PyArg_ParseTuple(args, "O!i", &PartitionType, &part, &field)) {
+		PyErr_SetString(PyExc_TypeError, ARG_ERR);
+		return NULL;
+	}
+
+	pa = part->pa;
+
+	fdisk_partition_to_string(pa, self->cxt, field, &data);
+	ret = Py_BuildValue("s", data);
+	free(data);
+
+	return ret;
+}
+static PyMethodDef Context_methods[] = {
+	{"assign_device",	(PyCFunction)Context_assign_device, METH_VARARGS, Context_assign_device_HELP},
+	{"partition_to_string",	(PyCFunction)Context_partition_to_string, METH_VARARGS, Context_partition_to_string_HELP},
+	{NULL}
+};
+
+
+static PyObject *Context_get_nsectors(ContextObject *self)
+{
+	return PyLong_FromUnsignedLong(fdisk_get_nsectors(self->cxt));
+}
+static PyObject *Context_get_sector_size(ContextObject *self)
+{
+	return PyLong_FromUnsignedLong(fdisk_get_sector_size(self->cxt));
+}
+static PyObject *Context_get_devname(ContextObject *self)
+{
+	return PyObjectResultStr(fdisk_get_devname(self->cxt));
+}
+static PyObject *Context_get_label(ContextObject *self)
+{
+	struct fdisk_context *cxt = self->cxt;
+
+	if (fdisk_has_label(cxt)) {
+		return PyObjectResultLabel(fdisk_get_label(cxt, NULL));
+	} else {
+		Py_RETURN_NONE;
+	}
+}
+static PyObject *Context_get_nparts(ContextObject *self)
+{
+	return PyLong_FromLong(fdisk_table_get_nents(self->tb));
+}
+static PyObject *Context_get_parts(ContextObject *self)
+{
+	PyObject *p, *list = PyList_New(0); /* XXX: null if failed*/
+	struct fdisk_partition *pa;
+	struct fdisk_iter *itr;
+	struct fdisk_table *tb;
+	/* char *data; */
+	
+	tb = self->tb;
+	itr = fdisk_new_iter(FDISK_ITER_FORWARD);
+
+	while(fdisk_table_next_partition(tb, itr, &pa) == 0) {
+		/* const char *name = fdisk_partition_get_name(pa);*/
+		p = PyObjectResultPartition(pa);
+		PyList_Append(list, p);
+		/*free(data);*/
+	}	
+
+	fdisk_free_iter(itr);
+
+	return list;
+}
+static PyGetSetDef Context_getseters[] = {
+	{"nsectors",	(getter)Context_get_nsectors, NULL, "context number of sectors", NULL},
+	{"sector_size",	(getter)Context_get_sector_size, NULL, "context sector size", NULL},
+	{"devname",	(getter)Context_get_devname, NULL, "context devname", NULL},
+	{"label",	(getter)Context_get_label, NULL, "context label type", NULL},
+	{"nparts",	(getter)Context_get_nparts, NULL, "context label number of existing partitions", NULL},
+	{"parts",	(getter)Context_get_parts, NULL, "context partitions", NULL},
+	{NULL}
+};
+
+static PyObject *Context_repr(ContextObject *self)
+{
+	return PyUnicode_FromFormat("<libfdisk.Context object at %p, details=%s>",
+			self, fdisk_is_details(self->cxt) ? "True" : "False");
+}
+
+PyTypeObject ContextType = {
+	PyVarObject_HEAD_INIT(NULL, 0)
+	"libfdisk.Context", /*tp_name*/
+	sizeof(ContextObject), /*tp_basicsize*/
+	0, /*tp_itemsize*/
+	(destructor)Context_dealloc, /*tp_dealloc*/
+	0, /*tp_print*/
+	NULL, /*tp_getattr*/
+	NULL, /*tp_setattr*/
+	NULL, /*tp_compare*/
+	(reprfunc) Context_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*/
+	Context_HELP, /* tp_doc */
+	NULL, /* tp_traverse */
+	NULL, /* tp_clear */
+	NULL, /* tp_richcompare */
+	0, /* tp_weaklistoffset */
+	NULL, /* tp_iter */
+	NULL, /* tp_iternext */
+	Context_methods, /* tp_methods */
+	Context_members, /* tp_members */
+	Context_getseters, /* tp_getset */
+	NULL, /* tp_base */
+	NULL, /* tp_dict */
+	NULL, /* tp_descr_get */
+	NULL, /* tp_descr_set */
+	0, /* tp_dictoffset */
+	(initproc)Context_init, /* tp_init */
+	NULL, /* tp_alloc */
+	Context_new, /* tp_new */
+};
+
+void Context_AddModuleObject(PyObject *mod)
+{
+	if (PyType_Ready(&ContextType) < 0)
+		return;
+
+	Py_INCREF(&ContextType);
+	PyModule_AddObject(mod, "Context", (PyObject *)&ContextType);
+}