grab_bag.py

   1 import random
   2
   3
   4 class grab_bag(object):
   5     def __init__(self):
   6         self.contents = set()
   7
   8     def clear(self):
   9         self.contents.clear()
  10
  11     def add(self, item):
  12         if item not in self.contents:
  13             self.contents.add(item)
  14
  15     def add_all(self, collection):
  16         for x in collection:
  17             self.add(x)
  18
  19     def subset(self, count):
  20         if len(self.contents) < count:
  21             return None
  22         subset = random.sample(self.contents, count)
  23         return subset
  24
  25     def size(self):
  26         return len(self.contents)
  27
  28
  29 # x = grab_bag()
  30 # x.add_all([1, 2, 3, 4, 5, 6, 7, 8, 9, 10])
  31 # print x.subset(3)