mirror of
https://github.com/pybind/pybind11.git
synced 2026-05-13 09:46:10 +00:00
added exec functions
This commit is contained in:
committed by
Wenzel Jakob
parent
c2ee3f52b5
commit
c6ad2c4993
156
include/pybind11/exec.h
Normal file
156
include/pybind11/exec.h
Normal file
@@ -0,0 +1,156 @@
|
||||
/*
|
||||
pybind11/exec.h: Functions to execute python from C++. Based on code from boost.python.
|
||||
|
||||
Copyright (c) 2005 Stefan Seefeld
|
||||
|
||||
This code is based on the boost.python implementation, so a different license applies to this file.
|
||||
|
||||
Boost Software License - Version 1.0 - August 17th, 2003
|
||||
|
||||
Permission is hereby granted, free of charge, to any person or organization
|
||||
obtaining a copy of the software and accompanying documentation covered by
|
||||
this license (the "Software") to use, reproduce, display, distribute,
|
||||
execute, and transmit the Software, and to prepare derivative works of the
|
||||
Software, and to permit third-parties to whom the Software is furnished to
|
||||
do so, all subject to the following:
|
||||
|
||||
The copyright notices in the Software and this entire statement, including
|
||||
the above license grant, this restriction and the following disclaimer,
|
||||
must be included in all copies of the Software, in whole or in part, and
|
||||
all derivative works of the Software, unless such copies or derivative
|
||||
works are solely in the form of machine-executable object code generated by
|
||||
a source language processor.
|
||||
|
||||
THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
||||
IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
||||
FITNESS FOR A PARTICULAR PURPOSE, TITLE AND NON-INFRINGEMENT. IN NO EVENT
|
||||
SHALL THE COPYRIGHT HOLDERS OR ANYONE DISTRIBUTING THE SOFTWARE BE LIABLE
|
||||
FOR ANY DAMAGES OR OTHER LIABILITY, WHETHER IN CONTRACT, TORT OR OTHERWISE,
|
||||
ARISING FROM, OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER
|
||||
DEALINGS IN THE SOFTWARE.
|
||||
|
||||
*/
|
||||
|
||||
#pragma once
|
||||
|
||||
#include "pytypes.h"
|
||||
|
||||
NAMESPACE_BEGIN(pybind11)
|
||||
|
||||
|
||||
|
||||
inline object eval (const std::string& st, object global = object(), object local = object()) {
|
||||
if (!global) {
|
||||
if (PyObject *g = PyEval_GetGlobals())
|
||||
global = object(g, true);
|
||||
else
|
||||
global = dict();
|
||||
}
|
||||
if (!local)
|
||||
local = global;
|
||||
|
||||
PyObject *res = PyRun_String(st.c_str() , Py_eval_input, global.ptr(), local.ptr());
|
||||
|
||||
if (res == nullptr)
|
||||
throw error_already_set();
|
||||
|
||||
return {res, false};
|
||||
}
|
||||
|
||||
inline object exec (const std::string& st, object global = object(), object local = object()) {
|
||||
if (!global) {
|
||||
if (PyObject *g = PyEval_GetGlobals())
|
||||
global = object(g, true);
|
||||
else
|
||||
global = dict();
|
||||
}
|
||||
if (!local)
|
||||
local = global;
|
||||
|
||||
PyObject *res = PyRun_String(st.c_str() , Py_file_input, global.ptr(), local.ptr());
|
||||
|
||||
if (res == nullptr)
|
||||
throw error_already_set();
|
||||
|
||||
return {res, false};
|
||||
}
|
||||
|
||||
inline object exec_statement (const std::string& st, object global = object(), object local = object()) {
|
||||
if (!global) {
|
||||
if (PyObject *g = PyEval_GetGlobals())
|
||||
global = object(g, true);
|
||||
else
|
||||
global = dict();
|
||||
}
|
||||
if (!local)
|
||||
local = global;
|
||||
|
||||
PyObject *res = PyRun_String(st.c_str() , Py_single_input, global.ptr(), local.ptr());
|
||||
if (res == nullptr)
|
||||
throw error_already_set();
|
||||
|
||||
return {res, false};
|
||||
}
|
||||
|
||||
inline object exec_file(const std::string& filename, object global = object(), object local = object()) {
|
||||
// Set suitable default values for global and local dicts.
|
||||
if (!global) {
|
||||
if (PyObject *g = PyEval_GetGlobals())
|
||||
global = object(g, true);
|
||||
else
|
||||
global = dict();
|
||||
}
|
||||
if (!local) local = global;
|
||||
|
||||
std::string f = filename; //need to copy for the signature of PyFile_FromString
|
||||
|
||||
|
||||
// Let python open the file to avoid potential binary incompatibilities.
|
||||
#if PY_VERSION_HEX >= 0x03040000
|
||||
const static int close_it = 1;
|
||||
FILE *fs = _Py_fopen(f.c_str(), "r");
|
||||
#elif PY_VERSION_HEX >= 0x03000000
|
||||
const static int close_it = 1;
|
||||
PyObject *fo = Py_BuildValue("s", f.c_str());
|
||||
FILE *fs = _Py_fopen(fo, "r");
|
||||
Py_DECREF(fo);
|
||||
#else
|
||||
const static int close_it = 0;
|
||||
PyObject *pyfile = PyFile_FromString(&f.front(), const_cast<char*>("r"));
|
||||
if (!pyfile)
|
||||
throw std::invalid_argument(std::string(f) + " : no such file");
|
||||
object file(pyfile, false);
|
||||
FILE *fs = PyFile_AsFile(file.ptr());
|
||||
#endif
|
||||
if (fs == nullptr)
|
||||
throw std::invalid_argument(std::string(f) + " : could not be opened");
|
||||
|
||||
PyObject* res = PyRun_FileEx(fs,
|
||||
f.c_str(),
|
||||
Py_file_input,
|
||||
global.ptr(), local.ptr(),
|
||||
close_it);
|
||||
|
||||
if (res == nullptr)
|
||||
throw error_already_set();
|
||||
|
||||
return {res, false};
|
||||
|
||||
}
|
||||
|
||||
inline object exec (str string, object global = object(), object local = object()) {
|
||||
return exec(static_cast<std::string>(string), global, local);
|
||||
}
|
||||
|
||||
inline object eval (str string, object global = object(), object local = object()) {
|
||||
return eval(static_cast<std::string>(string), global, local);
|
||||
}
|
||||
|
||||
inline object exec_file(str filename, object global = object(), object local = object()) {
|
||||
return exec_file(static_cast<std::string>(filename), global, local);
|
||||
}
|
||||
inline object exec_statement (str string, object global = object(), object local = object()) {
|
||||
return exec_statement(static_cast<std::string>(string), global, local);
|
||||
}
|
||||
|
||||
NAMESPACE_END(pybind11)
|
||||
Reference in New Issue
Block a user