Remove send top 3 drinks stats

[hubacji1/coffee-getter.git] / cbgetter.py

diff --git a/cbgetter.py b/cbgetter.py
index 4c4992ec2499d0a54550b3d86d6f553c98b33b5a..a08a2268de61e8eb5aaf5d065b1c1e9f745abdd6 100644 (file)
--- a/cbgetter.py
+++ b/cbgetter.py
@@ -34,7 +34,7 @@ class CoffeebotGetter:
         smiles = {
             "tea": ":tea:",
             "espresso": ":coffee:",
-            "espresso lungo": ":coffin:",
+            "espresso lungo": ":owl:",
             "cappuccino": ":flara:",
             "latte macchiato": ":baby_bottle:",
             "Club-Mate 0,5 l": ":club-mate:",
@@ -76,6 +76,7 @@ class CoffeebotGetter:
 
     def computeEqualDrinkers(self):
         """Compute equally drunken drinkers."""
+        self.topeq = []
         if len(self.top) <= 0:
             return
         lc = self.top[0][1] + 1
@@ -86,6 +87,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"