/*

** Copyright (C) 2020 SoftBank Robotics Europe

** See COPYING for the license

*/

#include <qipython/common.hpp>

#include <qipython/pymodule.hpp>

#include <qipython/pyobject.hpp>

#include <qipython/pyguard.hpp>

#include <qi/anyobject.hpp>

#include <qi/anymodule.hpp>

#include <pybind11/pybind11.h>

namespace py = pybind11;

namespace qi

{

namespace py

{

namespace

{

::py::object call(::py::object obj, ::py::str name,

::py::args args, ::py::kwargs kwargs)

{

GILAcquire lock;

return obj.attr(name)(*args, **kwargs);

}

::py::object getModule(const std::string& name)

{

const auto mod = import(name);

const auto pyMod = toPyObject(mod);

const ::py::cpp_function callFn(&call, ::py::is_method(pyMod.get_type()),

::py::arg("name"));

const auto types = ::py::module::import("types");

::py::setattr(pyMod, "createObject", types.attr("MethodType")(callFn, pyMod));

return pyMod;

}

::py::list listModules()

{

const auto modules = invokeGuarded<GILRelease>(&qi::listModules);

return castToPyObject(AnyReference::from(modules));

}

} // namespace

void exportObjectFactory(::py::module& m)

{

using namespace ::py;

GILAcquire lock;

m.def("module", &getModule,

doc(":returns: an object that represents the requested module.\n"));

m.def("listModules", &listModules);

}

} // namespace py

} // namespace qi