| 1 | n/a | |
|---|
| 2 | 2 | def flatten(tup): |
|---|
| 3 | 0 | elts = [] |
|---|
| 4 | 0 | for elt in tup: |
|---|
| 5 | 0 | if isinstance(elt, tuple): |
|---|
| 6 | 0 | elts = elts + flatten(elt) |
|---|
| 7 | n/a | else: |
|---|
| 8 | 0 | elts.append(elt) |
|---|
| 9 | 0 | return elts |
|---|
| 10 | n/a | |
|---|
| 11 | 4 | class Set: |
|---|
| 12 | 2 | def __init__(self): |
|---|
| 13 | 3159 | self.elts = {} |
|---|
| 14 | 2 | def __len__(self): |
|---|
| 15 | 0 | return len(self.elts) |
|---|
| 16 | 2 | def __contains__(self, elt): |
|---|
| 17 | 0 | return elt in self.elts |
|---|
| 18 | 2 | def add(self, elt): |
|---|
| 19 | 6814 | self.elts[elt] = elt |
|---|
| 20 | 2 | def elements(self): |
|---|
| 21 | 2109 | return self.elts.keys() |
|---|
| 22 | 2 | def has_elt(self, elt): |
|---|
| 23 | 11 | return elt in self.elts |
|---|
| 24 | 2 | def remove(self, elt): |
|---|
| 25 | 11 | del self.elts[elt] |
|---|
| 26 | 2 | def copy(self): |
|---|
| 27 | 0 | c = Set() |
|---|
| 28 | 0 | c.elts.update(self.elts) |
|---|
| 29 | 0 | return c |
|---|
| 30 | n/a | |
|---|
| 31 | 4 | class Stack: |
|---|
| 32 | 2 | def __init__(self): |
|---|
| 33 | 2118 | self.stack = [] |
|---|
| 34 | 2118 | self.pop = self.stack.pop |
|---|
| 35 | 2 | def __len__(self): |
|---|
| 36 | 3 | return len(self.stack) |
|---|
| 37 | 2 | def push(self, elt): |
|---|
| 38 | 1327 | self.stack.append(elt) |
|---|
| 39 | 2 | def top(self): |
|---|
| 40 | 3 | return self.stack[-1] |
|---|
| 41 | 2 | def __getitem__(self, index): # needed by visitContinue() |
|---|
| 42 | 0 | return self.stack[index] |
|---|
| 43 | n/a | |
|---|
| 44 | 2 | MANGLE_LEN = 256 # magic constant from compile.c |
|---|
| 45 | n/a | |
|---|
| 46 | 2 | def mangle(name, klass): |
|---|
| 47 | 20712 | if not name.startswith('__'): |
|---|
| 48 | 18898 | return name |
|---|
| 49 | 1814 | if len(name) + 2 >= MANGLE_LEN: |
|---|
| 50 | 0 | return name |
|---|
| 51 | 1814 | if name.endswith('__'): |
|---|
| 52 | 1750 | return name |
|---|
| 53 | 64 | try: |
|---|
| 54 | 64 | i = 0 |
|---|
| 55 | 64 | while klass[i] == '_': |
|---|
| 56 | 0 | i = i + 1 |
|---|
| 57 | 0 | except IndexError: |
|---|
| 58 | 0 | return name |
|---|
| 59 | 64 | klass = klass[i:] |
|---|
| 60 | n/a | |
|---|
| 61 | 64 | tlen = len(klass) + len(name) |
|---|
| 62 | 64 | if tlen > MANGLE_LEN: |
|---|
| 63 | 0 | klass = klass[:MANGLE_LEN-tlen] |
|---|
| 64 | n/a | |
|---|
| 65 | 64 | return "_%s%s" % (klass, name) |
|---|
| 66 | n/a | |
|---|
| 67 | 2 | def set_filename(filename, tree): |
|---|
| 68 | n/a | """Set the filename attribute to filename on every node in tree""" |
|---|
| 69 | 31 | worklist = [tree] |
|---|
| 70 | 24869 | while worklist: |
|---|
| 71 | 24838 | node = worklist.pop(0) |
|---|
| 72 | 24838 | node.filename = filename |
|---|
| 73 | 24838 | worklist.extend(node.getChildNodes()) |
|---|
