python: Add Database class

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

diff --git a/src/python/locationmodule.c b/src/python/locationmodule.c
index dfd951ddf25108148bda98bd2e391eec44c3318d..02aeb32fc5853ea020e4fbc1c0e11b6edf42d180 100644 (file)
--- a/src/python/locationmodule.c
+++ b/src/python/locationmodule.c
@@ -16,6 +16,8 @@
 
 #include <Python.h>
 
+#include "database.h"
+
 PyMODINIT_FUNC PyInit_location(void);
 
 static PyMethodDef location_module_methods[] = {
@@ -35,5 +37,12 @@ PyMODINIT_FUNC PyInit_location(void) {
        if (!m)
                return NULL;
 
+       // Database
+       if (PyType_Ready(&DatabaseType) < 0)
+               return NULL;
+
+       Py_INCREF(&DatabaseType);
+       PyModule_AddObject(m, "Database", (PyObject *)&DatabaseType);
+
        return m;
 }