'#43 adds suport for multiple parameter python function override.

sepinf-inc · Oct 18, 2023 · 6ad1a2e · 6ad1a2e
1 parent c9e9ea4
commit 6ad1a2e
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/iped-utils/src/main/java/iped/utils/pythonhook/PythonHook.java b/iped-utils/src/main/java/iped/utils/pythonhook/PythonHook.java
@@ -125,9 +125,16 @@ public void overrideModuleFunction(String module, String functionName, Method me
         String packageName = method.getDeclaringClass().getPackageName();
         String className = method.getDeclaringClass().getSimpleName();
         String methodName = method.getName();
+        StringBuffer params = new StringBuffer("");
+        for (int i = 0; i < method.getParameterCount(); i++) {
+            params.append("param" + i);
+            if (i < method.getParameterCount()) {
+                params.append(",");
+            }
+        }
         addModification(module, "def " + functionName, "def " + functionName
-                + "(str):\n    from " + packageName + " import " + className + "\n    " + className + "." + methodName
-                + "(str)\ndef " + functionName + "2");
+                + "(" + params + "):\n    from " + packageName + " import " + className + "\n    " + className + "."
+                + methodName + "(" + params + ")\ndef " + functionName + "2");
     }
 
     /*