[KOGITO-9354] Adding search path configuration

apache · Jun 22, 2023 · 9c09b5a · 9c09b5a
1 parent b10c23a
commit 9c09b5a
Showing 1 changed file with 15 additions and 2 deletions.
diff --git a/...e/src/main/java/org/kie/kogito/serverless/workflow/python/PythonWorkItemHandlerUtils.java b/...e/src/main/java/org/kie/kogito/serverless/workflow/python/PythonWorkItemHandlerUtils.java
@@ -15,6 +15,10 @@
  */
 package org.kie.kogito.serverless.workflow.python;
 
+import java.util.List;
+
+import org.kie.kogito.serverless.workflow.utils.ConfigResolverHolder;
+
 import jep.Interpreter;
 import jep.SharedInterpreter;
 
@@ -23,13 +27,16 @@ public class PythonWorkItemHandlerUtils {
     private PythonWorkItemHandlerUtils() {
     }
 
-    private static ThreadLocal<Interpreter> interpreter = new ThreadLocal<>();
+    private static final String PYTHON_SYS_PATH = "sys.path.append('%s')\n";
+
+    private static final ThreadLocal<Interpreter> interpreter = new ThreadLocal<>();
 
     protected static Interpreter interpreter() {
         Interpreter py = interpreter.get();
         if (py == null) {
             py = new SharedInterpreter();
             interpreter.set(py);
+            ConfigResolverHolder.getConfigResolver().getConfigProperty("sonata.python.searchPath", List.class).ifPresent(PythonWorkItemHandlerUtils::addToSearchPath);
         }
         return py;
     }
@@ -42,8 +49,14 @@ protected static void closeInterpreter() {
         }
     }
 
+    private static void addToSearchPath(List<String> paths) {
+        Interpreter py = interpreter.get();
+        StringBuilder sb = new StringBuilder("import sys\n");
+        paths.forEach(path -> sb.append(String.format(PYTHON_SYS_PATH, path)));
+        py.exec(sb.toString());
+    }
+
     protected static Object getValue(String key) {
         return interpreter().getValue(key);
     }
-
 }