diff options
| author | Jeremy Hylton <jeremy@alum.mit.edu> | 2001-09-17 18:02:21 (GMT) |
|---|---|---|
| committer | Jeremy Hylton <jeremy@alum.mit.edu> | 2001-09-17 18:02:21 (GMT) |
| commit | 09392b77a4d8be766e394177a466d51c2a64480f (patch) | |
| tree | c6e5c05ddc75b312f14f8703b004b94b907b7256 | |
| parent | b329b7134c15e6d1baafb5597e85de514b1eb588 (diff) | |
| download | cpython-09392b77a4d8be766e394177a466d51c2a64480f.zip cpython-09392b77a4d8be766e394177a466d51c2a64480f.tar.gz cpython-09392b77a4d8be766e394177a466d51c2a64480f.tar.bz2 | |
Add utility to set filename attribute on all nodes
| -rw-r--r-- | Lib/compiler/misc.py | 9 | ||||
| -rw-r--r-- | Tools/compiler/compiler/misc.py | 9 |
2 files changed, 18 insertions, 0 deletions
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 @@ 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 9cbd73f..b4bbe78 100644 --- 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()) + |