Add utility to set filename attribute on all nodes

diff --git a/Lib/compiler/misc.py b/Lib/compiler/misc.py
index 9cbd73ff23547a1e7e110be10f14d0cdcb1bee4c..b4bbe78050c261b60ef44f51f10bba3422a8b3e4 100644 (file)
--- a/Lib/compiler/misc.py
+++ b/Lib/compiler/misc.py
@@ -64,3 +64,12 @@ def mangle(name, klass):
         klass = klass[:MANGLE_LEN-tlen]
 
     return "_%s%s" % (klass, name)
+
+def set_filename(filename, tree):
+    """Set the filename attribute to filename on every node in tree"""
+    worklist = [tree]
+    while worklist:
+        node = worklist.pop(0)
+        node.filename = filename
+        worklist.extend(node.getChildNodes())
+        

diff --git a/Tools/compiler/compiler/misc.py b/Tools/compiler/compiler/misc.py
index 9cbd73ff23547a1e7e110be10f14d0cdcb1bee4c..b4bbe78050c261b60ef44f51f10bba3422a8b3e4 100644 (file)
--- a/Tools/compiler/compiler/misc.py
+++ b/Tools/compiler/compiler/misc.py
@@ -64,3 +64,12 @@ def mangle(name, klass):
         klass = klass[:MANGLE_LEN-tlen]
 
     return "_%s%s" % (klass, name)
+
+def set_filename(filename, tree):
+    """Set the filename attribute to filename on every node in tree"""
+    worklist = [tree]
+    while worklist:
+        node = worklist.pop(0)
+        node.filename = filename
+        worklist.extend(node.getChildNodes())
+