commit 09392b77a4d8be766e394177a466d51c2a64480f
author Jeremy Hylton <jeremy@alum.mit.edu> Mon Sep 17 18:02:21 2001 +0000
committer Jeremy Hylton <jeremy@alum.mit.edu> Mon Sep 17 18:02:21 2001 +0000
tree c6e5c05ddc75b312f14f8703b004b94b907b7256
parent b329b7134c15e6d1baafb5597e85de514b1eb588

Add utility to set filename attribute on all nodes

Lib/compiler/misc.py

diff --git a/Lib/compiler/misc.py b/Lib/compiler/misc.py
index 9cbd73f..b4bbe78 100644
--- a/Lib/compiler/misc.py
+++ b/Lib/compiler/misc.py
@@ -64,3 +64,12 @@
         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())
+