Python venv support added

Found by: simple Patch by: thommey
michaelortmann · Jul 12, 2024 · a49c225 · a49c225
1 parent d2f93d5
commit a49c225
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/mod/python.mod/python.c b/src/mod/python.mod/python.c
@@ -64,13 +64,23 @@ static int python_gil_lock() {
 }
 
 static char *init_python() {
+  const char *venv;
+  char venvpython[PATH_MAX];
   PyObject *pmodule;
   PyStatus status;
   PyConfig config;
 
   PyConfig_InitPythonConfig(&config);
   config.install_signal_handlers = 0;
   config.parse_argv = 0;
+  if ((venv = getenv("VIRTUAL_ENV"))) {
+    snprintf(venvpython, sizeof venvpython, "%s/bin/python3", venv);
+    status = PyConfig_SetBytesString(&config, &config.executable, venvpython);
+    if (PyStatus_Exception(status)) {
+      PyConfig_Clear(&config);
+      return "Python: Fatal error: Could not set venv executable";
+    }
+  }
   status = PyConfig_SetBytesString(&config, &config.program_name, argv0);
   if (PyStatus_Exception(status)) {
     PyConfig_Clear(&config);