Function register_lgraph_plugin

src/python/python_api.cpp:1893–1937

}  // NOLINT

void register_lgraph_plugin(pybind11::module& m) {
    //======================================
    // Register Python task runner
    //======================================
    pybind11::class_<lgraph::python_plugin::TaskInput>(m, "TaskInput")
        .def_static(
            "ReadTaskInput",
            [](const std::string& pipename) {
                boost::interprocess::message_queue mq(boost::interprocess::open_only,
                                                      pipename.c_str());
                lgraph::python_plugin::TaskInput input;
                while (!input.ReadFromMessageQueue(mq, 100)) {
                    SignalsGuard signalsGuard;
                }
                return input;
            },
            "Read TaskInput from message queue",
            pybind11::call_guard<SignalsGuard>())
        .def_readonly("user", &lgraph::python_plugin::TaskInput::user, "user to be used")
        .def_readonly("graph", &lgraph::python_plugin::TaskInput::graph, "Graph to be used")
        .def_readonly("plugin_dir", &lgraph::python_plugin::TaskInput::plugin_dir,
                      "Directory where .py files are stored")
        .def_readonly("function", &lgraph::python_plugin::TaskInput::function, "The function name")
        .def(
            "get_input",
            [](const lgraph::python_plugin::TaskInput& in) { return pybind11::bytes(in.input); },
            "The input byte array",
            pybind11::call_guard<SignalsGuard>())
        .def_readonly("read_only", &lgraph::python_plugin::TaskInput::read_only);

    pybind11::class_<lgraph::python_plugin::TaskOutput>(m, "TaskOutput")
        .def_static(
            "WriteTaskOutput",
            [](const std::string& pipename, lgraph::python_plugin::TaskOutput::ErrorCode error_code,
               const std::string& output) {
                boost::interprocess::message_queue mq(boost::interprocess::open_only,
                                                      pipename.c_str());
                lgraph::python_plugin::TaskOutput out;
                out.error_code = error_code;
                out.output = output;
                out.WriteToMessageQueue(mq);
            },
            "Write TaskOutput to stdout",
            pybind11::call_guard<SignalsGuard>());
}

class EdgeListWriter {
    fma_common::OutputFmaStream ofs_;