folkslinux
/
mamba
mirror of https://github.com/mamba-org/mamba.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

add stubs with pybind11-stubgen (#1983) 

* add stubs with pybind11-stubgen
* move stubs to libmambapy/__init__.pyi
This commit is contained in:
Daniel Holth 2022-10-13 08:08:31 -04:00 committed by GitHub
parent e3ab841b2c
commit f055ff777c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 1271 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1259
libmambapy/libmambapy/__init__.pyi Normal file
View File

File diff suppressed because it is too large Load Diff

18
libmambapy/src/main.cpp
View File

@ -53,6 +53,14 @@ PYBIND11_MODULE(bindings, m)
{ {
using namespace mamba; using namespace mamba;
// declare earlier to avoid C++ types in docstrings
auto pyChannel = py::class_<Channel, std::unique_ptr<Channel, py::nodelete>>(m, "Channel");
auto pyPackageInfo = py::class_<PackageInfo>(m, "PackageInfo");
auto pyPrefixData = py::class_<PrefixData>(m, "PrefixData");
auto pySolver = py::class_<MSolver>(m, "Solver");
// only used in a return type; does it belong in the module?
auto pyRootRole = py::class_<validate::RootRole>(m, "RootRole");
py::class_<fs::u8path>(m, "Path") py::class_<fs::u8path>(m, "Path")
.def(py::init<std::string>()) .def(py::init<std::string>())
.def("__str__", [](fs::u8path& self) -> std::string { return self.string(); }) .def("__str__", [](fs::u8path& self) -> std::string { return self.string(); })
@ -143,8 +151,7 @@ PYBIND11_MODULE(bindings, m)
.def("find_python_version", &MTransaction::find_python_version) .def("find_python_version", &MTransaction::find_python_version)
.def("execute", &MTransaction::execute); .def("execute", &MTransaction::execute);
py::class_<MSolver>(m, "Solver") pySolver.def(py::init<MPool&, std::vector<std::pair<int, int>>>(), py::keep_alive<1, 2>())
.def(py::init<MPool&, std::vector<std::pair<int, int>>>(), py::keep_alive<1, 2>())
.def("add_jobs", &MSolver::add_jobs) .def("add_jobs", &MSolver::add_jobs)
.def("add_global_job", &MSolver::add_global_job) .def("add_global_job", &MSolver::add_global_job)
.def("add_constraint", &MSolver::add_constraint) .def("add_constraint", &MSolver::add_constraint)
@ -343,7 +350,7 @@ PYBIND11_MODULE(bindings, m)
.def("set_verbosity", &Context::set_verbosity) .def("set_verbosity", &Context::set_verbosity)
.def("set_log_level", &Context::set_log_level); .def("set_log_level", &Context::set_log_level);
py::class_<PrefixData>(m, "PrefixData") pyPrefixData
.def(py::init( .def(py::init(
[](const fs::u8path& prefix_path) -> PrefixData [](const fs::u8path& prefix_path) -> PrefixData
{ {
@ -360,8 +367,7 @@ PYBIND11_MODULE(bindings, m)
.def_property_readonly("package_records", &PrefixData::records) .def_property_readonly("package_records", &PrefixData::records)
.def("add_packages", &PrefixData::add_packages); .def("add_packages", &PrefixData::add_packages);
py::class_<PackageInfo>(m, "PackageInfo") pyPackageInfo.def(py::init<Solvable*>())
.def(py::init<Solvable*>())
.def(py::init<const std::string&>(), py::arg("name")) .def(py::init<const std::string&>(), py::arg("name"))
.def(py::init<const std::string&, const std::string&, const std::string&, std::size_t>(), .def(py::init<const std::string&, const std::string&, const std::string&, std::size_t>(),
py::arg("name"), py::arg("name"),
@ -475,7 +481,7 @@ PYBIND11_MODULE(bindings, m)
{ return role.create_key_mgr(nlohmann::json::parse(json_str)); }, { return role.create_key_mgr(nlohmann::json::parse(json_str)); },
py::arg("json_str")); py::arg("json_str"));
py::class_<Channel, std::unique_ptr<Channel, py::nodelete>>(m, "Channel") pyChannel
.def(py::init([](const std::string& value) .def(py::init([](const std::string& value)
{ return const_cast<Channel*>(&make_channel(value)); })) { return const_cast<Channel*>(&make_channel(value)); }))
.def_property_readonly("scheme", &Channel::scheme) .def_property_readonly("scheme", &Channel::scheme)