Lib/compiler/misc.py

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