utils.py

   1 #
   2 # Useful utilities.
   3 #
   4 #
   5 # A simple class for accumulating counts or lists
   6 #
   7 class accumulator:
   8     def __init__(self):
   9         self._data = { }
  10
  11     def incr(self, key, increment = 1):
  12         try:
  13             self._data[key] += increment
  14         except KeyError:
  15             self._data[key] = increment
  16
  17     def get(self, key, default = None):
  18         try:
  19             return self._data[key]
  20         except KeyError:
  21             return default
  22
  23     def append(self, key, item, unique = False):
  24         if unique and (key in self._data) and (item in self._data[key]):
  25             return
  26         try:
  27             self._data[key].append(item)
  28         except KeyError:
  29             self._data[key] = [item]
  30
  31     def keys(self):
  32         return self._data.keys()
  33
  34     def __getitem__(self, key):
  35         return self._data[key]
  36