PyGetString.h

Go to the documentation of this file.

// This file's extension implies that it's C, but it's really -*- C++ -*-.
/*
 * Copyright (C) 2002-2023 CERN for the benefit of the ATLAS collaboration.
 */
#ifndef ROOTUTILS_PYGETSTRING_H
#define ROOTUTILS_PYGETSTRING_H
 
#ifdef _POSIX_C_SOURCE
# undef _POSIX_C_SOURCE
#endif
#ifdef _XOPEN_SOURCE
# undef _XOPEN_SOURCE
#endif
#include "Python.h"
#include <string>
#include <utility>
 
 
namespace RootUtils {
 
 
inline
std::pair<std::string, bool> PyGetString (PyObject* s)
{
  const char* cstr = PyUnicode_AsUTF8AndSize (s, nullptr);
  if (!cstr && PyBytes_Check (s)) {
    PyErr_Clear();
    cstr = PyBytes_AsString (s);
  }
  if (cstr) {
    return std::make_pair (std::string (cstr), true);
  }
  return std::make_pair (std::string(), false);
}
 
 
} // namespace RootUtils
 
 
#endif // not ROOTUTILS_PYGETSTRING_H