Add utility to set filename attribute on all nodes

09392b77 · Jeremy Hylton · b329b713 · 09392b77 · 09392b77
Commit 09392b77 authored Sep 17, 2001 by Jeremy Hylton
Hide whitespace changes
Inline Side-by-side

Showing with 18 additions and 0 deletions

Lib/compiler/misc.py Lib/compiler/misc.py +9 -0

Tools/compiler/compiler/misc.py Tools/compiler/compiler/misc.py +9 -0

No files found.
--- 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())
+        
--- 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())
+