PyOutputWrapper.cpp

Go to the documentation of this file.

/*
 * @file PyOutputWrapper.cpp
 * @author Alejandro Alvarez Ayllon
 */
 
#include "SEImplementation/PythonConfig/PyOutputWrapper.h"
#include <boost/locale/encoding_utf.hpp>
#include <boost/python/extract.hpp>
 
namespace SourceXtractor {
 
namespace bp = boost::python;
using boost::locale::conv::utf_to_utf;
 
PyOutputWrapper::PyOutputWrapper(Elements::Logging& logger) : closed(false), m_logger(logger) {}
 
void PyOutputWrapper::close() {
  PyErr_Format(PyExc_OSError, "Method %s not supported on wrapper file objects", __func__);
  bp::throw_error_already_set();
}
 
int PyOutputWrapper::fileno() const {
  PyErr_Format(PyExc_OSError, "Method %s not supported on wrapper file objects", __func__);
  bp::throw_error_already_set();
  return -1;
}
 
void PyOutputWrapper::flush() {
}
 
bool PyOutputWrapper::isatty() const {
  return false;
}
 
bool PyOutputWrapper::readable() const {
  return false;
}
 
std::string PyOutputWrapper::read(int) {
  PyErr_Format(PyExc_IOError, "Object is write only");
  bp::throw_error_already_set();
  return {};
}
 
std::string PyOutputWrapper::readline(int) {
  PyErr_Format(PyExc_IOError, "Object is write only");
  bp::throw_error_already_set();
  return {};
}
 
bp::list PyOutputWrapper::readlines(int) {
  PyErr_Format(PyExc_IOError, "Object is write only");
  bp::throw_error_already_set();
  return {};
}
 
int PyOutputWrapper::seek(int, int) {
  PyErr_Format(PyExc_OSError, "Method %s not supported on wrapper file objects", __func__);
  bp::throw_error_already_set();
  return 0;
}
 
bool PyOutputWrapper::seekable() const {
  return false;
}
 
int PyOutputWrapper::tell() const {
  PyErr_Format(PyExc_OSError, "Method %s not supported on wrapper file objects", __func__);
  bp::throw_error_already_set();
  return 0;
}
 
void PyOutputWrapper::truncate(int) {
  PyErr_Format(PyExc_OSError, "Method %s not supported on wrapper file objects", __func__);
  bp::throw_error_already_set();
}
 
bool PyOutputWrapper::writable() const {
  return true;
}
 
int PyOutputWrapper::write(const bp::object& obj) {
  bp::extract<std::string> string_extractor{obj};
  bp::extract<std::wstring> unicode_extractor{obj};
  std::string str;
 
  if (string_extractor.check()) {
    str = string_extractor;
  }
  else if (unicode_extractor.check()) {
    std::wstring unicode = unicode_extractor;
    str = utf_to_utf<char>(unicode.c_str(), unicode.c_str() + unicode.size());
  }
  else {
    std::string obj_type_name = bp::extract<std::string>(obj.attr("__class__").attr("__name__"));
    PyErr_Format(PyExc_TypeError,
                 "SourceXtractor output wrapper only accepts classic strings or unicode strings, got %s",
                 obj_type_name.c_str()
    );
    bp::throw_error_already_set();
  }
 
  for (auto c : str) {
    if (c == '\n') {
      m_logger.info() << m_buffer.str();
      m_buffer.str(std::string());
    } else {
      m_buffer << c;
    }
  }
  return str.size();
}
 
void PyOutputWrapper::writelines(const bp::list& lines) {
  for (int i = 0; i < bp::len(lines); ++i) {
    write(lines[i]);
  }
}
 
} // end namespace SourceXtractor