(fix) add python NameError support to dynamic polyglot context

rapiddweller · Jan 19, 2024 · b27c983 · b27c983
1 parent c7bf2b5
commit b27c983
Showing 1 changed file with 11 additions and 2 deletions.
diff --git a/src/main/java/com/rapiddweller/benerator/script/PolyglotContext.java b/src/main/java/com/rapiddweller/benerator/script/PolyglotContext.java
@@ -55,14 +55,23 @@ synchronized public Value evalScript(Context context, String text, String langua
             this.updatePolyglotLocalFromGlobal(context, language);
             returnValue = polyglotCtx.eval(language, text);
         } catch (org.graalvm.polyglot.PolyglotException e) {
-            if (e.getMessage().contains("is not defined")) {
+            if (e.getMessage().contains("ReferenceError: ")) {
                 String missingObject = e.getMessage().replace("ReferenceError: ", "").replace(" is not defined", "");
                 if (!Objects.equals(previousMissingObject, missingObject)) {
                     this.migrateBeneratorContext2GraalVM(context, language, missingObject);
                     returnValue = evalScript(context, text, language);
                     previousMissingObject = missingObject;
                 }
-            } else {
+            }
+            if (e.getMessage().contains(("NameError: "))) {
+                String missingObject = e.getMessage().replace("NameError: name '", "").replace("' is not defined", "");
+                if (!Objects.equals(previousMissingObject, missingObject)) {
+                    this.migrateBeneratorContext2GraalVM(context, language, missingObject);
+                    returnValue = evalScript(context, text, language);
+                    previousMissingObject = missingObject;
+                }
+            }
+            else {
                 throw new ScriptException(e.getMessage(), null);
             }
         }