Prevent ref count error

aerospike · Sep 24, 2024 · 8e512bc · 8e512bc
1 parent 9469959
commit 8e512bc
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/main/aerospike.c b/src/main/aerospike.c
@@ -558,6 +558,11 @@ PyMODINIT_FUNC PyInit_aerospike(void)
             goto GLOBAL_HOSTS_CLEANUP_ON_ERROR;
         }
 
+        if (PyType_Check(py_member)) {
+            // Documentation example shows that we need to increment reference count for static types before adding to module
+            // It doesn't make clear why though
+            Py_INCREF(py_member);
+        }
         retval = PyModule_AddObject(py_aerospike_module,
                                     module_pyobjects[i].obj_name, py_member);
         if (retval == -1) {