如何使py :: objects在module.reload()之后保持活动状态

时间:2019-09-08 09:00:34

标签: c++ pybind11

在重新加载模块my_module之后,将删除其所有对象,但是我想从C ++中删除对象。如何防止Python删除对象?

#include <iostream>
#include <pybind11/embed.h>
class Obj{ ~Obj(){std::cout << "deleted" << std::endl;} };

PYBIND11_EMBEDDED_MODULE(module_obj, m) {
  py::class_<Obj>(m,"Obj");
}

/*
#my_module.py
import module_obj
o = module_obj.Obj()
*/

py::scoped_interpreter intpr;
auto my_module = py::module::import("my_module")
auto* o = my_module.attr("o").cast<Obj*>()
my_module.reload() // now o is deleted, how to prevent it?

1 个答案:

答案 0 :(得分:0)

您需要将o对象的引用计数保持在零以上。最简单的方法是在C ++端制作py::object(或等效地在python端创建一个对象)

#include <iostream>
#include <pybind11/embed.h>

namespace py=pybind11;

class Obj {
public:
  ~Obj() {
    std::cout << "deleted" << std::endl;
  }
};

PYBIND11_EMBEDDED_MODULE(my_module, m) {
  py::class_<Obj>(m, "Cat")
      .def(py::init());
}

int main() {
  py::scoped_interpreter intpr;
  std::cout << "Loading... " << std::endl;
  auto my_module = py::module::import("my_module");
  std::cout << "Load complete. " << std::endl;

  py::object o = my_module.attr("Cat")(); // create new cat, automatically increases/decreases ref count 
  std::cout << "Reloading... " << std::endl;
  my_module.reload(); // now o is deleted, how to prevent it?
  std::cout << "Reload complete. " << std::endl;
}

输出:

Loading... 
Load complete. 
Reloading... 
Reload complete. 
deleted