python: Raise IndexError when adding the same network twice

[people/ms/libloc.git] / src / python / writer.c

diff --git a/src/python/writer.c b/src/python/writer.c
index e0c37b6b88c6f286d88dfdf26fa963fa6161b05f..a15f77e8cffd8e9cb2ea92070bd329eb298a227f 100644 (file)
--- a/src/python/writer.c
+++ b/src/python/writer.c
@@ -128,8 +128,15 @@ static PyObject* Writer_add_network(WriterObject* self, PyObject* args) {
        // Create network object
        int r = loc_writer_add_network(self->writer, &network, string);
        if (r) {
-               if (r == -EINVAL)
-                       PyErr_SetString(PyExc_ValueError, "Invalid network");
+               switch (r) {
+                       case -EINVAL:
+                               PyErr_SetString(PyExc_ValueError, "Invalid network");
+                               break;
+
+                       case -EBUSY:
+                               PyErr_SetString(PyExc_IndexError, "A network already exists here");
+                               break;
+               }
 
                return NULL;
        }