vtroshchinskiy
/
python-libfdisk
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

context: add_partition returns partno 

libfdisk context add_partition function initializes a passed argument
with the new partition's partno.

If add_partition is successful, return its partno.
master
Jose M. Guisado 2022-12-14 17:44:11 +01:00
parent bc6f5cbdff
commit eaf98290cf
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
context.c
View File

@ -161,10 +161,11 @@ static PyObject *Context_write_disklabel(ContextObject *self, PyObject *args, Py
Py_RETURN_NONE; Py_RETURN_NONE;
} }
#define Context_add_partition_HELP "add_partition(fdisk.Partition)\n\n" \ #define Context_add_partition_HELP "add_partition(fdisk.Partition)\n\n" \
"Adds partition to context" "Adds partition to context. Returns partno of the new partition."
static PyObject *Context_add_partition(ContextObject *self, PyObject *args, PyObject *kwds) static PyObject *Context_add_partition(ContextObject *self, PyObject *args, PyObject *kwds)
{ {
PartitionObject *partobj; PartitionObject *partobj;
size_t partno;
int rc; int rc;
if (!PyArg_ParseTuple(args, "O!", &PartitionType, &partobj)) { if (!PyArg_ParseTuple(args, "O!", &PartitionType, &partobj)) {
@ -177,13 +178,13 @@ static PyObject *Context_add_partition(ContextObject *self, PyObject *args, PyOb
return NULL; return NULL;
} }
rc = fdisk_add_partition(self->cxt, partobj->pa, NULL); rc = fdisk_add_partition(self->cxt, partobj->pa, &partno);
if (rc < 0) { if (rc < 0) {
PyErr_Format(PyExc_RuntimeError, "Error adding partition to context: %s", strerror(-rc)); PyErr_Format(PyExc_RuntimeError, "Error adding partition to context: %s", strerror(-rc));
return NULL; return NULL;
} }
Py_RETURN_NONE; return Py_BuildValue("n", partno);
} }
static PyMethodDef Context_methods[] = { static PyMethodDef Context_methods[] = {
{"assign_device", (PyCFunction)Context_assign_device, METH_VARARGS, Context_assign_device_HELP}, {"assign_device", (PyCFunction)Context_assign_device, METH_VARARGS, Context_assign_device_HELP},