# -*- coding: utf-8 -*-
"""Get data from database and publish them."""
from json import dumps
+from random import randint
from requests import post
from cbconf import Conf
from cbdb import Db
self.cfg = Conf()
self.db = Db(self.cfg.getCoffeeDbPath())
self.msg = ""
+ self.top = []
+ self.topeq = []
return None
def getMsg(self):
def sendMsgToSlack(self):
"""Send message to Slack Bot."""
return post(self.cfg.getCoffeebotURL(), dumps({"text": self.msg}))
+
+ def getHeroComment(self):
+ """Return comment for hero."""
+ hsuff = [
+ ", 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]
+
+ def getSomeSmile(self):
+ """Return some Slack smile."""
+ smiles = [
+ ":party_parrot:",
+ ":cat2:",
+ ":heavy_check_mark:",
+ ":baby_bottle:",
+ ":+1:",
+ ":aaw_yeah:",
+ ":better:",
+ ]
+ ri = randint(0, len(smiles) - 1)
+ return smiles[ri]
+
+ def computeEqualDrinkers(self):
+ """Compute equally drunken drinkers."""
+ if len(self.top) <= 0:
+ return
+ 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
+
+ def loadDrunkSum(self, *args, **kwargs):
+ """Get list of drunken ``flavor`` from ``dtf`` to ``dtt``, update msg.
+
+ Keyword arguments:
+ flavor -- The flavor of beverage.
+ dtf -- Date and time *from*.
+ dtt -- Date and time *to*.
+ """
+ self.top = self.db.getDrunkSum(*args, **kwargs)
+ if args:
+ flavor = args[0]
+ dtf = args[1]
+ dtt = args[2]
+ elif kwargs:
+ flavor = kwargs["flavor"]
+ dtf = kwargs["dtf"]
+ dtt = kwargs["dtt"]
+ self.msg = "{} drunk from {} to {}:\n".format(flavor, dtf, dtt)
+ for (un, cnt) in self.top:
+ self.msg += "{}: {}\n".format(un, cnt)
+ self.msg += "---"
+ return
+
+ def loadDrunkList(self, *args, **kwargs):
+ """Get list of drunken ``flavor`` from ``dtf`` to ``dtt``, update msg.
+
+ Keyword arguments:
+ flavor -- The flavor of beverage.
+ dtf -- Date and time *from*.
+ dtt -- Date and time *to*.
+ """
+ self.top = self.db.getDrunkSum(*args, **kwargs)
+ d = self.db.getDrunkList(*args, **kwargs)
+ if args:
+ flavor = args[0]
+ dtf = args[1]
+ dtt = args[2]
+ elif kwargs:
+ flavor = kwargs["flavor"]
+ dtf = kwargs["dtf"]
+ dtt = kwargs["dtt"]
+ self.msg = "{} drunk from {} to {}:\n".format(flavor, dtf, dtt)
+ for (un, l) in d.items():
+ self.msg += "{}: ".format(un)
+ for cnt in l[:-1]:
+ self.msg += "{}, ".format(cnt)
+ self.msg += "{}\n".format(l[-1])
+ self.msg += "---"
+ return