Allow a custom NodeTransformer.

zopefoundation · Sep 22, 2016 · 661fbf6 · 661fbf6
1 parent 7925823
commit 661fbf6
Showing 1 changed file with 8 additions and 3 deletions.
diff --git a/src/RestrictedPython/compiler.py b/src/RestrictedPython/compiler.py
@@ -3,8 +3,13 @@
 
 
 def compile_restricted(
-        source, filename='<unknown>', mode='exec', flags=0, dont_inherit=0):
-    """Replacement for the built-in compile() function."""
+        source, filename='<unknown>', mode='exec', flags=0, dont_inherit=0,
+        policy=RestrictingNodeTransformer):
+    """Replacement for the built-in compile() function.
+
+    policy ... `ast.NodeTransformer` class defining the restrictions.
+
+    """
     c_ast = ast.parse(source, filename, mode)
-    r_ast = RestrictingNodeTransformer().visit(c_ast)
+    r_ast = policy().visit(c_ast)
     return compile(r_ast, filename, mode, flags, dont_inherit)