1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
|
#ifndef SCRIPTINGPYTHON_H
#define SCRIPTINGPYTHON_H
#include <Python.h>
#include "scriptingpython_global.h"
#include "plugins/genericplugin.h"
#include "plugins/scriptingplugin.h"
#include "db/sqlquery.h"
#include <QCache>
class QMutex;
class SCRIPTINGPYTHONSHARED_EXPORT ScriptingPython : public GenericPlugin, public DbAwareScriptingPlugin
{
Q_OBJECT
SQLITESTUDIO_PLUGIN("scriptingpython.json")
public:
static PyObject* dbEval(PyObject *self, PyObject* const* args, Py_ssize_t nargs);
ScriptingPython();
~ScriptingPython();
bool init();
void deinit();
QString getLanguage() const;
Context* createContext();
void releaseContext(Context* context);
void resetContext(Context* context);
void setVariable(Context* context, const QString& name, const QVariant& value);
QVariant getVariable(Context* context, const QString& name);
bool hasError(Context* context) const;
QString getErrorMessage(Context* context) const;
QString getIconPath() const;
QVariant evaluate(Context* context, const QString& code, const FunctionInfo& funcInfo, const QList<QVariant>& args, Db* db, bool locking = false);
QVariant evaluate(const QString& code, const FunctionInfo& funcInfo, const QList<QVariant>& args, Db* db, bool locking = false, QString* errorMessage = nullptr);
private:
class ContextPython;
class ScriptObject
{
public:
ScriptObject(const QString& code, const FunctionInfo& funcInfo, ContextPython* context);
~ScriptObject();
PyObject* getCompiled() const;
private:
PyObject* compiled = nullptr;
};
class ContextPython : public ScriptingPlugin::Context
{
public:
ContextPython();
~ContextPython();
void reset();
PyThreadState* interp = nullptr;
PyObject* mainModule = nullptr;
PyObject* envDict = nullptr;
QCache<QString, ScriptObject> scriptCache;
QString error;
Db* db = nullptr;
bool useDbLocking = false;
private:
void init();
void clear();
};
ContextPython* getContext(ScriptingPlugin::Context* context) const;
QVariant compileAndEval(ContextPython* ctx, const QString& code, const FunctionInfo& funcInfo,
const QList<QVariant>& args, Db* db, bool locking);
void clearError(ContextPython* ctx);
ScriptObject* getScriptObject(const QString code, const ScriptingPlugin::FunctionInfo& funcInfo, ContextPython* ctx);
static QString extractError();
static PyObject* argsToPyArgs(const QVariantList& args, const QStringList& namedParameters);
static QVariant pythonObjToVariant(PyObject* obj);
static QString pythonObjToString(PyObject* obj);
static PyObject* variantToPythonObj(const QVariant& value);
static PyObject* stringToPythonObj(const QString& value);
static SqlQueryPtr dbCommonEval(PyObject* sqlArg, const char* fnName);
static QVariant getVariable(const QString& name);
static const constexpr int cacheSize = 5;
static QHash<PyThreadState*, ContextPython*> contexts;
ContextPython* mainContext = nullptr;
QMutex* mainInterpMutex = nullptr;
};
#endif // SCRIPTINGPYTHON_H
|
