Add utility to set filename attribute on all nodes

jeremyhylton · jeremyhylton · commit 09392b77a4d8 · 2001-09-17T18:02:21.000Z
diff --git 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
@@ -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())
+        
