Add test and fix for default argument processing in sub-functions.

MutatingWalker.py

@@ -83,7 +83,7 @@
 # 
 ##############################################################################
 
-__version__='$Revision: 1.2 $'[11:-2]
+__version__='$Revision: 1.3 $'[11:-2]
 
 from compiler import ast
 
@@ -97,8 +97,10 @@ def __init__(self, visitor):
         self.visitor = visitor
         self._cache = {}
 
-    def defaultVisitNode(self, node, walker=None):
+    def defaultVisitNode(self, node, walker=None, exclude=None):
         for name, child in node.__dict__.items():
+            if exclude is not None and name in exclude:
+                continue
             v = self.dispatchObject(child)
             if v is not child:
                 # Replace the node.

RestrictionMutator.py

@@ -87,7 +87,7 @@
 compiler.transformer.Transformer, restricting and enhancing the
 code in various ways before sending it to pycodegen.
 '''
-__version__='$Revision: 1.6 $'[11:-2]
+__version__='$Revision: 1.7 $'[11:-2]
 
 from compiler import ast
 from compiler.transformer import parse
@@ -206,10 +206,11 @@ def visitFunction(self, node, walker):
         self.checkName(node, node.name)
         for argname in node.argnames:
             self.checkName(node, argname)
+        walker.visitSequence(node.defaults)
 
         former_funcinfo = self.funcinfo
         self.funcinfo = FuncInfo()
-        node = walker.defaultVisitNode(node)
+        node = walker.defaultVisitNode(node, exclude=('defaults',))
         self.prepBody(node.code.nodes)
         self.funcinfo = former_funcinfo
         return node

tests/restricted_module.py

@@ -42,6 +42,11 @@ def allowed_read(ob):
     print len(ob)
     return printed
 
+def allowed_default_args(ob):
+    def f(a=ob.allowed, s=ob.s):
+        return a, s
+
+
 def allowed_simple():
     q = {'x':'a'}
     q['y'] = 'b'
@@ -72,6 +77,10 @@ def denied_getattr(ob):
     ob.disallowed = 1
     return ob.disallowed
 
+def denied_default_args(ob):
+    def f(d=ob.disallowed):
+        return d
+
 def denied_setattr(ob):
     ob.allowed = -1
 

tests/testRestrictions.py

@@ -204,6 +204,9 @@ def checkAllowedRead(self):
     def checkAllowedWrite(self):
         self.execFunc('allowed_write', RestrictedObject())
 
+    def checkAllowedArgs(self):
+        self.execFunc('allowed_default_args', RestrictedObject())
+
     def checkDenied(self):
         for k in rmodule.keys():
             if k[:6] == 'denied':