diff --git a/dfhack/python/pydfhack.cpp b/dfhack/python/pydfhack.cpp index 90bc7a59b..b23a93b45 100644 --- a/dfhack/python/pydfhack.cpp +++ b/dfhack/python/pydfhack.cpp @@ -30,6 +30,7 @@ distribution. #include "DF_CreatureManager.cpp" #include "DF_Translate.cpp" #include "DF_Vegetation.cpp" +#include "DF_Buildings.cpp" #include "DF_API.cpp" #ifndef PyMODINIT_FUNC @@ -88,6 +89,9 @@ PyMODINIT_FUNC initpydfhack(void) if(PyType_Ready(&DF_Vegetation_type) < 0) return; + if(PyType_Ready(&DF_Building_type) < 0) + return; + module = Py_InitModule3("pydfhack", module_methods, "pydfhack extension module"); Py_INCREF(&DF_API_type); @@ -107,6 +111,7 @@ PyMODINIT_FUNC initpydfhack(void) PyModule_AddObject(module, "CreatureManager", (PyObject*)&DF_CreatureManager_type); PyModule_AddObject(module, "Translate", (PyObject*)&DF_Translate_type); PyModule_AddObject(module, "Vegetation", (PyObject*)&DF_Vegetation_type); + PyModule_AddObject(module, "Building", (PyObject*)&DF_Building_type); //DoImports(); } \ No newline at end of file