Web: support to query values via TCP/IP communication

FreeCAD · Sep 25, 2021 · 5387e12 · 5387e12 · dcapeletti · Oct 3, 2021
1 parent bd46e97
commit 5387e12
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 1 deletion.
diff --git a/src/Mod/Web/App/Server.cpp b/src/Mod/Web/App/Server.cpp
@@ -112,6 +112,13 @@ const QByteArray& ServerEvent::request() const
 AppServer::AppServer(QObject* parent)
   : QTcpServer(parent)
 {
+    PyObject* mod = PyImport_ImportModule("__main__");
+    if (mod) {
+        module = mod;
+    }
+    else {
+        throw Py::RuntimeError("Cannot load __main__ module");
+    }
 }
 
 void AppServer::incomingConnection(qintptr socket)
@@ -147,12 +154,35 @@ void AppServer::customEvent(QEvent* e)
     QByteArray msg = ev->request();
     QTcpSocket* socket = ev->socket();
 
-    std::string str = runPython(msg);
+    std::string str;
+    if (msg.startsWith("GET ")) {
+        msg = QByteArray("GET = ") + msg.mid(4);
+        str = runPython(msg);
+        if (str == "None") {
+            str = getRequest(str);
+        }
+    }
+    else {
+        str = runPython(msg);
+    }
 
     socket->write(str.c_str());
     socket->close();
 }
 
+std::string AppServer::getRequest(const std::string& str) const
+{
+    try {
+        Base::PyGILStateLocker lock;
+        Py::Object attr = module.getAttr(std::string("GET"));
+        return attr.as_string();
+    }
+    catch (Py::Exception &e) {
+        e.clear();
+        return str;
+    }
+}
+
 std::string AppServer::runPython(const QByteArray& msg)
 {
     std::string str;

diff --git a/src/Mod/Web/App/Server.h b/src/Mod/Web/App/Server.h
@@ -89,12 +89,14 @@ class AppServer : public QTcpServer
 
 protected:
     void customEvent(QEvent* e);
+    std::string getRequest(const std::string&) const;
 
 private Q_SLOTS:
     void readClient();
     void discardClient();
 
 private:
+    Py::Object module;
 };
 
 }