python: Handle error codes when searching for an AS

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

diff --git a/src/python/database.c b/src/python/database.c
index 931b54d4040e3650cec27b265b8ff61ac065d24e..7bac3bca0b8883af29804e68800149b1acb8b01d 100644 (file)
--- a/src/python/database.c
+++ b/src/python/database.c
@@ -86,19 +86,21 @@ static PyObject* Database_get_as(DatabaseObject* self, PyObject* args) {
 
        // Try to retrieve the AS
        int r = loc_database_get_as(self->db, &as, number);
-       if (r)
-               return NULL;
 
-       // Create an AS object
-       if (as) {
+       // We got an AS
+       if (r == 0) {
                PyObject* obj = new_as(&ASType, as);
                loc_as_unref(as);
 
                return obj;
-       }
 
        // Nothing found
-       Py_RETURN_NONE;
+       } else if (r == 1) {
+               Py_RETURN_NONE;
+       }
+
+       // Unexpected error
+       return NULL;
 }
 
 static PyObject* Database_lookup(DatabaseObject* self, PyObject* args) {