diff --git a/dfhack/python/pydfhack.cpp b/dfhack/python/pydfhack.cpp index a1ee27387..feb2da2cc 100644 --- a/dfhack/python/pydfhack.cpp +++ b/dfhack/python/pydfhack.cpp @@ -26,6 +26,8 @@ distribution. #include "DF_Imports.cpp" #include "DF_MemInfo.cpp" #include "DF_Material.cpp" +#include "DF_CreatureType.cpp" +#include "DF_CreatureManager.cpp" #include "DF_API.cpp" #ifndef PyMODINIT_FUNC @@ -72,17 +74,27 @@ PyMODINIT_FUNC initpydfhack(void) if(PyType_Ready(&DF_Material_type) < 0) return; + if(PyType_Ready(&DF_Creature_Base_type) < 0) + return; + + if(PyType_Ready(&DF_CreatureManager_type) < 0) + return; + module = Py_InitModule3("pydfhack", module_methods, "pydfhack extension module"); Py_INCREF(&DF_API_type); Py_INCREF(&DF_MemInfo_type); Py_INCREF(&DF_Position_type); Py_INCREF(&DF_Material_type); + Py_INCREF(&DF_Creature_Base_type); + Py_INCREF(&DF_CreatureManager_type); PyModule_AddObject(module, "API", (PyObject*)&DF_API_type); PyModule_AddObject(module, "MemInfo", (PyObject*)&DF_MemInfo_type); PyModule_AddObject(module, "Position", (PyObject*)&DF_Position_type); PyModule_AddObject(module, "Materials", (PyObject*)&DF_Material_type); + PyModule_AddObject(module, "Creature_Base", (PyObject*)&DF_Position_type); + PyModule_AddObject(module, "CreatureManager", (PyObject*)&DF_Material_type); DoImports(); } \ No newline at end of file