From: Jiri Vlasak <hubacji1@fel.cvut.cz>
Date: Mon, 25 Mar 2019 14:11:43 +0000 (+0100)
Subject: Add equal drinkers list, method
X-Git-Tag: v0.2.0~3^2~2
X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/commitdiff_plain/4e4fa09a281f74e97be497c0b856e0bb56c53f58?hp=865bf773e68f2047f4a27b48ff92bbb5a45d7ce7

Add equal drinkers list, method
---

diff --git a/cbgetter.py b/cbgetter.py
index 606bfef..c02c11c 100644
--- 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"