PythonStdErrOutRedirect.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <pybind11/pybind11.h>
 
#include <utility>
 
namespace py = pybind11;
 
// largely taken from https://github.com/pybind/pybind11/issues/1622
class PyStdErrOutStreamRedirect
{
    py::object m_stdout;
    py::object m_stderr;
    py::object m_stdout_buffer;
    py::object m_stderr_buffer;
 
  public:
    PyStdErrOutStreamRedirect()
    {
        auto sysm = py::module::import("sys");
        m_stdout = sysm.attr("stdout");
        m_stderr = sysm.attr("stderr");
        auto ccConsoleOutput = py::module::import("ccinternals").attr("ccConsoleOutput");
        m_stdout_buffer = ccConsoleOutput();
        m_stderr_buffer = ccConsoleOutput();
        sysm.attr("stdout") = m_stdout_buffer;
        sysm.attr("stderr") = m_stderr_buffer;
    }
 
    PyStdErrOutStreamRedirect(py::object stdout_obj, py::object stderr_obj)
    {
        auto sysm = py::module::import("sys");
        m_stdout = sysm.attr("stdout");
        m_stderr = sysm.attr("stderr");
        m_stdout_buffer = std::move(stdout_obj);
        m_stderr_buffer = std::move(stderr_obj);
        sysm.attr("stdout") = m_stdout_buffer;
        sysm.attr("stderr") = m_stderr_buffer;
    }
 
    ~PyStdErrOutStreamRedirect() noexcept
    {
        try
        {
            const auto sysm = py::module::import("sys");
            sysm.attr("stdout") = m_stdout;
            sysm.attr("stderr") = m_stderr;
        }
        catch (const std::exception &)
        {
        }
    }
};