add varargs and kwargs flags to Lambda nodes

7708d697 · Jeremy Hylton · 873bdc18 · 7708d697 · 7708d697
Commit 7708d697 authored Feb 17, 2000 by Jeremy Hylton
Hide whitespace changes
Inline Side-by-side

Showing with 10 additions and 0 deletions

Lib/compiler/ast.py Lib/compiler/ast.py +5 -0

Tools/compiler/compiler/ast.py Tools/compiler/compiler/ast.py +5 -0

No files found.
--- 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:]

--- 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:]