add varargs and kwargs flags to Lambda nodes

diff --git a/Lib/compiler/ast.py b/Lib/compiler/ast.py
index 7a7a6549c7034360c3e91f3cbc17d18763cd816a..dfed562be614e80db50b811349816d0bdf6407b1 100644 (file)
--- a/Lib/compiler/ast.py
+++ b/Lib/compiler/ast.py
@@ -105,6 +105,11 @@ class Lambda(Node):
     self.flags = flags
     self.code = code
     self._children = ('lambda', argnames, defaults, flags, code)
+    self.varargs = self.kwargs = None
+    if flags & CO_VARARGS:
+      self.varargs = 1
+    if flags & CO_VARKEYWORDS:
+      self.kwargs = 1
 
   def __repr__(self):
     return "Lambda(%s,%s,%s,%s)" % self._children[1:]

diff --git a/Tools/compiler/compiler/ast.py b/Tools/compiler/compiler/ast.py
index 7a7a6549c7034360c3e91f3cbc17d18763cd816a..dfed562be614e80db50b811349816d0bdf6407b1 100644 (file)
--- a/Tools/compiler/compiler/ast.py
+++ b/Tools/compiler/compiler/ast.py
@@ -105,6 +105,11 @@ class Lambda(Node):
     self.flags = flags
     self.code = code
     self._children = ('lambda', argnames, defaults, flags, code)
+    self.varargs = self.kwargs = None
+    if flags & CO_VARARGS:
+      self.varargs = 1
+    if flags & CO_VARKEYWORDS:
+      self.kwargs = 1
 
   def __repr__(self):
     return "Lambda(%s,%s,%s,%s)" % self._children[1:]