Changeset a93896 for src/Python

Timestamp:

Aug 30, 2025, 2:41:40 PM (8 weeks ago)

Author:

Frederik Heber <frederik.heber@…>

Branches:

Candidate_v1.7.0, stable

Children:

454d7a

Parents:

9928ece

git-author:

Frederik Heber <frederik.heber@…> (08/14/25 14:36:30)

git-committer:

Frederik Heber <frederik.heber@…> (08/30/25 14:41:40)

Message:

Ignores pyMoleCuilder import in python script.

• TEST: LoadSession now uses an AS alias for the import.
• TESTFIX: load session with complex waits does not look for the wait texts but rather for output from the actions in the script.

Location:

src/Python

Files:

• PythonScripting.cpp (modified) (4 diffs)
• PythonScripting.hpp (modified) (1 diff)

src/Python/PythonScripting.cpp

@@ -37 +37 @@
 #include "PythonScripting.hpp"
 
+#include <string>
+
 //#include "CodePatterns/MemDebug.hpp"
 
@@ -58 +60 @@
   // move all lines with "import" to the beginning
   std::string current_line;
+  std::string moduleName;
   while(std::getline(input,current_line,'\n')){
-    if (current_line.find("import ") != std::string::npos)
-      buffer_import << current_line << std::endl;
-    else
+    if (current_line.find("import ") != std::string::npos) {
+      if (current_line.find(PYMOLECUILDER_MODULE_NAME) != std::string::npos) {
+          // ignore a PyMoleCuilder import, use the AS however.
+          const size_t lowerCaseAsPos = current_line.find("as");
+          const size_t upperCaseAsPos = current_line.find("AS");
+          size_t moduleNameStart = std::string::npos;
+                                if (lowerCaseAsPos != std::string::npos)
+                                        moduleNameStart = lowerCaseAsPos+2+1;
+                                if (upperCaseAsPos != std::string::npos)
+                                        moduleNameStart = upperCaseAsPos+2+1;
+                                if (moduleNameStart != std::string::npos) {
+                                        const size_t moduleNameEnd = current_line.find(" ", moduleNameStart);
+                                        moduleName = current_line.substr(
+                  moduleNameStart,
+                                                moduleNameEnd == std::string::npos ? moduleNameEnd : moduleNameEnd  - moduleNameStart
+                                        );
+                ELOG(2, "Ignoring found " << PYMOLECUILDER_MODULE_NAME << " import, but keeping AS name " << moduleName << ".");
+        } else {
+                ELOG(2, "Ignoring found " << PYMOLECUILDER_MODULE_NAME << " import.");
+        }
+      } else {
+        buffer_import << current_line << std::endl;
+      }
+    } else {
       buffer_else << current_line << std::endl;
+    }
   }
   std::string pythonscript_import( std::istreambuf_iterator<char>(buffer_import),
@@ -71 +96 @@
   try {
 #ifdef HAVE_PYTHON2
-    PyImport_AppendInittab( "pyMoleCuilder", &initpyMoleCuilder );
+    PyImport_AppendInittab( PYMOLECUILDER_MODULE_NAME.c_str(), &initpyMoleCuilder );
 #else
-    PyImport_AppendInittab( "pyMoleCuilder", &PyInit_pyMoleCuilder );
+    PyImport_AppendInittab( PYMOLECUILDER_MODULE_NAME.c_str(), &PyInit_pyMoleCuilder );
 #endif
 
@@ -89 +114 @@
 
     boost::python::object molecuilder_module( (boost::python::handle<>(PyImport_ImportModule("pyMoleCuilder"))) );
-    main_namespace["pyMoleCuilder"] = molecuilder_module;
+    main_namespace[PYMOLECUILDER_MODULE_NAME] = molecuilder_module;
+    if (moduleName.size() > 0)
+        main_namespace[moduleName] = molecuilder_module;
 
     std::string enveloped_script("print(\"BEGIN of "+python_scripttitle+":\")\n");

src/Python/PythonScripting.hpp

@@ -20 +20 @@
 bool executePythonScript(const std::string &python_string, const std::string &python_scripttitle);
 
+const std::string PYMOLECUILDER_MODULE_NAME = "pyMoleCuilder";
+
 #endif /* PYTHONSCRIPTING_HPP_ */