Add equal drinkers list, method

diff --git a/cbgetter.py b/cbgetter.py
index 606bfef2427e34b81ee9af9cd8ed7c31207d6071..c02c11c838516f1b71fc9a848169852331fb81ae 100644 (file)
--- a/cbgetter.py
+++ b/cbgetter.py
@@ -13,6 +13,7 @@ class CoffeebotGetter:
         self.db = Db(self.cfg.getCoffeeDbPath())
         self.msg = ""
         self.top = []
+        self.topeq = []
         return None
 
     def getMsg(self):
@@ -55,6 +56,16 @@ class CoffeebotGetter:
         ri = randint(0, len(smiles) - 1)
         return smiles[ri]
 
+    def computeEqualDrinkers(self):
+        """Compute equally drunken drinkers."""
+        lc = self.top[0][1] + 1
+        for (un, c) in self.top:
+            if c < lc:
+                self.topeq.append([])
+                lc = c
+            self.topeq[-1].append((un, c))
+        return
+
     def loadTopMateDrinkers(self):
         """Get list of top Mate drinkers, update message."""
         self.msg = "*Top 5 Club-Mate drinkers*\n"