Merge pull request #352 from 0x0elliot/0x0elliot/weird_print_bug

Fixing #351: Overwriting the existing print() statement
Shuffle · Jul 3, 2023 · d3ae697 · d3ae697
2 parents 985e380 + b0dde8d
commit d3ae697
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/shuffle-tools/1.2.0/src/app.py b/shuffle-tools/1.2.0/src/app.py
@@ -529,10 +529,14 @@ def execute_python(self, code):
         # 2. Subprocess execute file?
         try:
             f = StringIO()
-            with redirect_stdout(f):
-                exec(code)  # nosec :(
+            def custom_print(*args, **kwargs):
+                return print(*args, file=f, **kwargs)
+
+            with redirect_stdout(f): # just in case
+                exec(code, {"print": custom_print})
 
             s = f.getvalue()
+            f.close() # why: https://www.youtube.com/watch?v=6SA6S9Ca5-U
 
             #try:
             #    s = s.encode("utf-8")