Method from_python

api/python/src/pyIOStream.cpp:28–68

PyIOStream& PyIOStream::operator=(PyIOStream&& other) = default;

result<PyIOStream> PyIOStream::from_python(nb::object object) {
  const nb::module_ mod_io = nb::module_::import_("io");
  const nb::object IOBase = mod_io.attr("IOBase");

  if (!nb::isinstance(object, IOBase)) {
    logging::log(logging::LEVEL::ERR,
        "The provided io object does not sub-class io.IOBase");
    return make_error_code(lief_errors::read_error);
  }

  if (!nb::hasattr(object, "read") && !nb::hasattr(object, "readinto")) {
    logging::log(logging::LEVEL::ERR,
        "The provided io object does not implement read() or readinto()");
    return make_error_code(lief_errors::read_error);
  }

  auto seek = object.attr("seek");
  seek(0, PY_SEEK_SET);
  seek(0, PY_SEEK_END);
  const auto size = nb::cast<size_t>(object.attr("tell")());

  if (size == 0) {
    return PyIOStream(std::move(object), {});
  }

  std::vector<uint8_t> data;
  data.resize(size);

  seek(0, PY_SEEK_SET);
  if (nb::hasattr(object, "readinto")) {
    auto view = nb::memoryview::from_memory(data.data(), size);
    object.attr("readinto")(view);
  }
  else if (nb::hasattr(object, "read")) {
    auto content = nb::cast<nb::bytes>(object.attr("read")(size));
    std::string tmp(content.c_str(), content.size());
    std::move(std::begin(tmp), std::end(tmp), data.data());
  }

  return PyIOStream(std::move(object), std::move(data));
}

PyIOStream::PyIOStream(nb::object io, std::vector<uint8_t> data) :
  VectorStream(std::move(data)),