From 4e4fa09a281f74e97be497c0b856e0bb56c53f58 Mon Sep 17 00:00:00 2001 From: Jiri Vlasak Date: Mon, 25 Mar 2019 15:11:43 +0100 Subject: [PATCH] Add equal drinkers list, method --- cbgetter.py | 11 +++++++++++ 1 file changed, 11 insertions(+) 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" -- 2.39.2