From 328b9157786005c8ddbd31764e5adaeddb37660a Mon Sep 17 00:00:00 2001 From: Jiri Vlasak Date: Thu, 2 Jan 2020 10:53:09 +0100 Subject: [PATCH] Add top tea drinkers getter method --- cbgetter.py | 39 +++++++++++++++++++++++++++++++++++++++ 1 file changed, 39 insertions(+) diff --git a/cbgetter.py b/cbgetter.py index 280ee03..668620d 100644 --- a/cbgetter.py +++ b/cbgetter.py @@ -109,6 +109,45 @@ class CoffeebotGetter: ) return + def load_top_tea_drinkers(self): + """Get list of top tea drinkers, update message.""" + self.msg = "*Top 5 tea drinkers*\n" + self.top = self.db.get_top_tea_drinkers() + self.computeEqualDrinkers() + tot = sum([c for (un, c) in self.top]) + i = 1 + for drinkers in self.topeq[:5]: + c = drinkers[0][1] + d = [un for (un, c) in drinkers] + self.msg += "{}{}".format(i, self.isuff[i]) + if len(d) > 1: + self.msg += " are " + else: + self.msg += " is " + for un in d: + if un is d[0]: + self.msg += "*{}*".format(un) + elif un is d[-1]: + self.msg += "" if len(d) == 2 else "," + self.msg += " and *{}*".format(un) + else: + self.msg += ", *{}*".format(un) + if len(d) > 1: + self.msg += " of whom each drank " + else: + self.msg += " who drank " + if c > 1: + self.msg += "`{:.2f}` teas".format(c) + else: + self.msg += "`{:.2f}` tea".format(c) + self.msg += "\n" + i += 1 + self.msg += "\n" + self.msg += "Oh, and together we drank *{:.2f}* teas {}".format( + tot, + self.getSomeSmile() + ) + def loadDrunkSum(self, *args, **kwargs): """Get list of drunken ``flavor`` from ``dtf`` to ``dtt``, update msg. -- 2.39.2