如何使用 pybind11 提供默认枚举值?

Question

我想让函数接受枚举默认参数。

但是我找不到在 PyBind11 枚举示例和文档中提供枚举默认值的正确方法，例如：

struct Pet
{
enum Kind
{
Dog = 0,
Cat
};
Pet(const std::string &name) : name(name)
{
}
void setName(const std::string &name_)
{
name = name_;
}
const std::string &getName() const
{
return name;
}
Kind test(Kind kind = Dog)
{
if(kind == Dog)
std::cout << "Dog" << std::endl;
if(kind == Cat)
std::cout << "Cat" << std::endl;
return kind;
}
std::string name;
};
PYBIND11_MODULE(pet,m)
{
py::class_<Pet> pet(m, "Pet");
pet.def(py::init<const std::string &>())
.def("setName", &Pet::setName)
.def("getName", &Pet::getName)
.def("test", &Pet::test, py::arg("kind") = Pet::Kind::Dog)
.def("__repr__", [](const Pet &a) { return "<example.Pet named '" + a.name + "'>"; }
);
py::enum_<Pet::Kind>(pet, "Kind")
.value("Dog", Pet::Kind::Dog)
.value("Cat", Pet::Kind::Cat)
.export_values();

但它不起作用：

py::arg("kind") = Pet::Kind::Dog 

当我在Python中运行时，我收到错误。

from pet import Pet as p
Traceback (most recent call last):
File "<stdin>", line 1, in <module>
ImportError: arg(): could not convert default argument into a Python object (type not registered yet?). Compile in debug mode for more information.

我收到错误，并且尝试使用字符串值(即"Dog"或 1(初始化它时。

Answer 1

这是一个简单的排序问题：在加载时，执行模块定义中的语句来创建 Python 类、函数等。因此，首先定义Kind(即在 Python 端(，以允许解释器在稍后定义默认值时在设置默认值期间找到它test。即，使用以下顺序：

PYBIND11_MODULE(pet,m)
{
py::class_<Pet> pet(m, "Pet");
py::enum_<Pet::Kind>(pet, "Kind")
.value("Dog", Pet::Kind::Dog)
.value("Cat", Pet::Kind::Cat)
.export_values();
pet.def(py::init<const std::string &>())
.def("setName", &Pet::setName)
.def("getName", &Pet::getName)
.def("test", &Pet::test, py::arg("kind") = Pet::Kind::Dog)
.def("__repr__", [](const Pet &a) { return "<example.Pet named '" + a.name + "'>"; }
);
}