misc.py 1.75 KB
Newer Older
1 2 3 4

def flatten(tup):
    elts = []
    for elt in tup:
5
        if isinstance(elt, tuple):
6 7 8 9 10
            elts = elts + flatten(elt)
        else:
            elts.append(elt)
    return elts

11 12
class Set:
    def __init__(self):
13
        self.elts = {}
14 15
    def __len__(self):
        return len(self.elts)
16
    def __contains__(self, elt):
17
        return elt in self.elts
18
    def add(self, elt):
19
        self.elts[elt] = elt
20
    def elements(self):
21
        return self.elts.keys()
22
    def has_elt(self, elt):
23
        return elt in self.elts
Jeremy Hylton's avatar
Jeremy Hylton committed
24 25
    def remove(self, elt):
        del self.elts[elt]
26 27 28 29
    def copy(self):
        c = Set()
        c.elts.update(self.elts)
        return c
30 31 32

class Stack:
    def __init__(self):
33 34
        self.stack = []
        self.pop = self.stack.pop
35 36
    def __len__(self):
        return len(self.stack)
37
    def push(self, elt):
38
        self.stack.append(elt)
39
    def top(self):
40
        return self.stack[-1]
41 42
    def __getitem__(self, index): # needed by visitContinue()
        return self.stack[index]
Jeremy Hylton's avatar
Jeremy Hylton committed
43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65

MANGLE_LEN = 256 # magic constant from compile.c

def mangle(name, klass):
    if not name.startswith('__'):
        return name
    if len(name) + 2 >= MANGLE_LEN:
        return name
    if name.endswith('__'):
        return name
    try:
        i = 0
        while klass[i] == '_':
            i = i + 1
    except IndexError:
        return name
    klass = klass[i:]

    tlen = len(klass) + len(name)
    if tlen > MANGLE_LEN:
        klass = klass[:MANGLE_LEN-tlen]

    return "_%s%s" % (klass, name)
66 67 68 69 70 71 72 73

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())