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]