diff options
| -rw-r--r-- | partition.c | 36 |
1 files changed, 27 insertions, 9 deletions
diff --git a/partition.c b/partition.c index 6057f5b..59c359b 100644 --- a/partition.c +++ b/partition.c @@ -37,25 +37,43 @@ static PyObject *Partition_new(PyTypeObject *type, return (PyObject *)self; } -#define Partition_HELP "Partition()" +#define Partition_HELP "Partition(partno_follow_default=False, " \ + "start_follow_default=False, " \ + "end_follow_default=False)" static int Partition_init(PartitionObject *self, PyObject *args, PyObject *kwds) { - /* char *kwlist[] = { - "context", + "partno_follow_default", + "start_follow_default", + "end_follow_default", NULL }; - + int partno_follow_default = 0, + start_follow_default = 0, + end_follow_default = 0; + if (!PyArg_ParseTupleAndKeywords(args, - kwds, "|O!", kwlist, - &ContextType, &cxt)) { - PyErr_SetString(PyExc_TypeError, "Error"); + kwds, "|ppp", kwlist, + &partno_follow_default, + &start_follow_default, + &end_follow_default)) { + PyErr_SetString(PyExc_TypeError, "Invalid arguments for new partition"); return -1; } - */ self->pa = fdisk_new_partition(); - fdisk_partition_start_follow_default(self->pa, 1); + if (fdisk_partition_partno_follow_default(self->pa, partno_follow_default) < 0) { + PyErr_SetString(PyExc_RuntimeError, "Error setting partno_follow_default"); + return -1; + } + if (fdisk_partition_start_follow_default(self->pa, start_follow_default) < 0) { + PyErr_SetString(PyExc_RuntimeError, "Error setting start_follow_default"); + return -1; + } + if (fdisk_partition_end_follow_default(self->pa, end_follow_default) < 0) { + PyErr_SetString(PyExc_RuntimeError, "Error setting end_follow_default"); + return -1; + } return 0; } |