Make top drinkers list constant

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

diff --git a/cbgetter.py b/cbgetter.py
index 727cacab589da520f141239ebf3daf0ebe8fdbee..aa4ab281c44812a0d08129a8b0383332ea5bd376 100644 (file)
--- a/cbgetter.py
+++ b/cbgetter.py
@@ -12,6 +12,8 @@ class CoffeebotGetter:
         self.cfg = Conf()
         self.db = Db(self.cfg.getCoffeeDbPath())
         self.msg = ""
+        self.top = []
+        self.topeq = []
         return None
 
     def getMsg(self):
@@ -25,20 +27,17 @@ class CoffeebotGetter:
     def getHeroComment(self):
         """Return comment for hero."""
         hsuff = [
-            " - yes",
-            " - no",
-            " - maybe",
-            #", oops",
-            #" - real hero",
-            #" because couldn't resist",
-            #" - drunken master",
-            #", woohoo",
-            #" - unbeatable",
-            #" - superultramaxidrinker",
-            #" and didn't sleep last week",
-            #" in last 604800 seconds",
-            #", no idea how",
-            #" and still drinking",
+            ", oops",
+            " - real hero",
+            " because couldn't resist",
+            " - drunken master",
+            ", woohoo",
+            " - unbeatable",
+            " - superultramaxidrinker",
+            " and didn't sleep last week",
+            " in last 604800 seconds",
+            ", no idea how",
+            " and still drinking",
         ]
         ri = randint(0, len(hsuff) - 1)
         return hsuff[ri]
@@ -56,3 +55,54 @@ 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"
+        self.top = self.db.getTopMateDrinkers()
+        self.computeEqualDrinkers()
+        totl = [c for (un, c) in self.top]
+        tot = sum(totl)
+        isuff = {1: "st", 2: "nd", 3: "rd", 4: "th", 5: "th"}
+        i = 1
+        for drinkers in self.topeq[:5]:
+            c = drinkers[0][1]
+            d = [un for (un, c) in drinkers]
+            self.msg += "{}{}".format(i, 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 "
+            self.msg += "`{:.2f}` liters".format(c)
+            if i == 1:
+                self.msg += self.getHeroComment()
+            self.msg += "\n"
+            i += 1
+        self.msg += "\n"
+        self.msg += "Oh, and together we drank *{:.2f}* liters {}".format(
+            tot,
+            self.getSomeSmile()
+        )
+        return