From 63cf08dba1d2c6a78e1c4be23bffc27888e7dd48 Mon Sep 17 00:00:00 2001 From: Jiri Vlasak Date: Thu, 2 Jan 2020 11:54:58 +0100 Subject: [PATCH] Add top drinks getter method --- cbgetter.py | 30 ++++++++++++++++++++++++++++++ 1 file changed, 30 insertions(+) diff --git a/cbgetter.py b/cbgetter.py index 4c4992e..ae611d4 100644 --- a/cbgetter.py +++ b/cbgetter.py @@ -86,6 +86,36 @@ class CoffeebotGetter: self.topeq[-1].append((un, c)) return + def load_top_drinks(self): + """Get list of top drinks, update message.""" + self.msg = "*Top drinks*\n" + self.top = self.db.get_top_drinks() + self.computeEqualDrinkers() + i = 1 + for drinks in self.topeq[:3]: + c = drinks[0][1] + d = [un for (un, c) in drinks] + 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(self.get_drink_smile(un)) + elif un is d[-1]: + self.msg += "" if len(d) == 2 else "," + self.msg += " and {}".format(self.get_drink_smile(un)) + else: + self.msg += ", *{}*".format(self.get_drink_smile(un)) + if len(d) > 1: + self.msg += " with each of " + else: + self.msg += " with " + self.msg += "`{}` drunk".format(c) + self.msg += "\n" + i += 1 + def loadTopMateDrinkers(self): """Get list of top Mate drinkers, update message.""" self.msg = "*Top 5 Club-Mate drinkers*\n" -- 2.39.2