cutter/src/common/PythonManager.cpp

169 lines
4.2 KiB
C++
Raw Normal View History

2018-06-23 16:59:23 +00:00
#include "PythonAPI.h"
#include "PythonManager.h"
#include <marshal.h>
2018-06-24 09:51:16 +00:00
#include <QDebug>
2018-06-23 16:59:23 +00:00
#include <QFile>
#include <QDebug>
#include "QtResImporter.h"
#include "plugins/CutterPythonPlugin.h"
Q_GLOBAL_STATIC(PythonManager, uniqueInstance)
PythonManager *PythonManager::getInstance()
{
return uniqueInstance;
}
PythonManager::PythonManager()
{
}
PythonManager::~PythonManager()
{
if (pyThreadState) {
PyEval_RestoreThread(pyThreadState);
if (cutterNotebookAppInstance) {
auto stopFunc = PyObject_GetAttrString(cutterNotebookAppInstance, "stop");
PyObject_CallObject(stopFunc, nullptr);
Py_DECREF(cutterNotebookAppInstance);
}
2018-06-24 09:51:16 +00:00
// TODO PyDECREF plugins
2018-06-23 16:59:23 +00:00
Py_Finalize();
}
if (pythonHome) {
PyMem_RawFree(pythonHome);
}
}
void PythonManager::initPythonHome()
{
#if defined(APPIMAGE) || defined(MACOS_PYTHON_FRAMEWORK_BUNDLED)
if (customPythonHome.isNull()) {
auto pythonHomeDir = QDir(QCoreApplication::applicationDirPath());
# ifdef APPIMAGE
// Executable is in appdir/bin
pythonHomeDir.cdUp();
qInfo() << "Setting PYTHONHOME =" << pythonHomeDir.absolutePath() << " for AppImage.";
# else // MACOS_PYTHON_FRAMEWORK_BUNDLED
// @executable_path/../Frameworks/Python.framework/Versions/Current
pythonHomeDir.cd("../Frameworks/Python.framework/Versions/Current");
qInfo() << "Setting PYTHONHOME =" << pythonHomeDir.absolutePath() <<
" for macOS Application Bundle.";
# endif
customPythonHome = pythonHomeDir.absolutePath();
}
#endif
if (!customPythonHome.isNull()) {
qInfo() << "PYTHONHOME =" << customPythonHome;
pythonHome = Py_DecodeLocale(customPythonHome.toLocal8Bit().constData(), nullptr);
Py_SetPythonHome(pythonHome);
}
}
void PythonManager::initialize()
{
initPythonHome();
PyImport_AppendInittab("_cutter", &PyInit_api);
PyImport_AppendInittab("cutter_internal", &PyInit_api_internal);
PyImport_AppendInittab("_qtres", &PyInit_qtres);
Py_Initialize();
PyEval_InitThreads();
// Import other modules
cutterJupyterModule = QtResImport("cutter_jupyter");
cutterPluginModule = QtResImport("cutter_plugin");
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
}
void PythonManager::addPythonPath(char *path) {
2018-06-24 09:51:16 +00:00
restoreThread();
2018-06-23 16:59:23 +00:00
PyObject *sysModule = PyImport_ImportModule("sys");
if (!sysModule) {
return;
}
PyObject *pythonPath = PyObject_GetAttrString(sysModule, "path");
if (!pythonPath) {
return;
}
PyObject *append = PyObject_GetAttrString(pythonPath, "append");
if (!append) {
return;
}
PyEval_CallFunction(append, "(s)", path);
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
}
bool PythonManager::startJupyterNotebook()
{
2018-06-24 09:51:16 +00:00
restoreThread();
2018-06-23 16:59:23 +00:00
PyObject* startFunc = PyObject_GetAttrString(cutterJupyterModule, "start_jupyter");
if (!startFunc) {
qWarning() << "Couldn't get attribute start_jupyter.";
return false;
}
cutterNotebookAppInstance = PyObject_CallObject(startFunc, nullptr);
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
return cutterNotebookAppInstance != nullptr;
}
QString PythonManager::getJupyterUrl()
{
2018-06-24 09:51:16 +00:00
restoreThread();
2018-06-23 16:59:23 +00:00
auto urlWithToken = PyObject_GetAttrString(cutterNotebookAppInstance, "url_with_token");
auto asciiBytes = PyUnicode_AsASCIIString(urlWithToken);
auto urlWithTokenString = QString::fromUtf8(PyBytes_AsString(asciiBytes));
Py_DECREF(asciiBytes);
Py_DECREF(urlWithToken);
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
return urlWithTokenString;
}
2018-06-24 09:51:16 +00:00
CutterPythonPlugin* PythonManager::loadPlugin(const char *pluginName) {
2018-06-23 16:59:23 +00:00
CutterPythonPlugin *plugin = nullptr;
if (!cutterPluginModule) {
return plugin;
}
2018-06-24 09:51:16 +00:00
restoreThread();
2018-06-23 16:59:23 +00:00
PyObject *pluginModule = PyImport_ImportModule(pluginName);
if (!pluginModule) {
qWarning() << "Couldn't import the plugin" << QString(pluginName);
2018-06-24 09:51:16 +00:00
PyErr_PrintEx(10);
} else {
plugin = new CutterPythonPlugin(pluginModule);
//Py_DECREF(pluginModule);
2018-06-23 16:59:23 +00:00
}
2018-06-24 09:51:16 +00:00
saveThread();
2018-06-23 16:59:23 +00:00
return plugin;
}
2018-06-24 09:51:16 +00:00
void PythonManager::restoreThread()
2018-06-23 16:59:23 +00:00
{
if (pyThreadState) {
PyEval_RestoreThread(pyThreadState);
}
2018-06-24 09:51:16 +00:00
}
2018-06-23 16:59:23 +00:00
2018-06-24 09:51:16 +00:00
void PythonManager::saveThread()
{
2018-06-23 16:59:23 +00:00
pyThreadState = PyEval_SaveThread();
}