allow nested flags from nested python dictionaries

This commit is contained in:
Christopher Lackner 2024-05-31 12:55:59 +02:00
parent 0e0ea2d5f8
commit 3029b5422a
2 changed files with 12 additions and 12 deletions

View File

@ -14,13 +14,11 @@ namespace ngcore
{ {
if (py::isinstance<py::dict>(value)) if (py::isinstance<py::dict>(value))
{ {
py::dict vdd(value); Flags nested_flags;
// call recursively to set dictionary for(auto item : value.cast<py::dict>())
for (auto item : vdd) { SetFlag(nested_flags, item.first.cast<string>(),
string name = item.first.cast<string>(); item.second.cast<py::object>());
py::object val = py::reinterpret_borrow<py::object>(item.second); flags.SetFlag(s, nested_flags);
SetFlag(flags, name, val);
}
return; return;
} }
@ -103,7 +101,9 @@ namespace ngcore
} }
} }
auto flags = py::cast<Flags>(flags_dict); Flags flags;
for(auto item : flags_dict)
SetFlag(flags, item.first.cast<string>(), item.second.cast<py::object>());
for (auto item : kwargs) for (auto item : kwargs)
{ {

View File

@ -149,12 +149,12 @@ PYBIND11_MODULE(pyngcore, m) // NOLINT
py::class_<Flags>(m, "Flags") py::class_<Flags>(m, "Flags")
.def(py::init<>()) .def(py::init<>())
.def("__str__", &ToString<Flags>) .def("__str__", &ToString<Flags>)
.def(py::init([](py::object & obj) { .def(py::init([](py::kwargs kwargs) {
Flags flags; Flags flags;
py::dict d(obj); for (auto d : kwargs)
SetFlag (flags, "", d); SetFlag(flags, d.first.cast<string>(), d.second.cast<py::object>());
return flags; return flags;
}), py::arg("obj"), "Create Flags by given object") }), "Create flags from kwargs")
.def(py::pickle([] (const Flags& self) .def(py::pickle([] (const Flags& self)
{ {
std::stringstream str; std::stringstream str;