1 # -*- coding: utf-8 -*-
2 """Get data from database and publish them."""
4 from random import randint
5 from requests import post
6 from cbconf import Conf
10 """This class connects configuration, database access and publishing."""
13 self.db = Db(self.cfg.getCoffeeDbPath())
17 self.isuff = {1: "st", 2: "nd", 3: "rd", 4: "th", 5: "th"}
24 def sendMsgToSlack(self):
25 """Send message to Slack Bot."""
26 return post(self.cfg.getCoffeebotURL(), dumps({"text": self.msg}))
28 def get_drink_smile(self, drink):
29 """Return Slack smile for ``drink``.
32 drink -- Drink of smile.
36 "espresso": ":coffee:",
37 "espresso lungo": ":owl:",
38 "cappuccino": ":flara:",
39 "latte macchiato": ":baby_bottle:",
40 "Club-Mate 0,5 l": ":club-mate:",
41 "Club-Mate 0,33 l": ":club-mate_cola:",
45 def getHeroComment(self):
46 """Return comment for hero."""
50 " because couldn't resist",
54 " - superultramaxidrinker",
55 " and didn't sleep last week",
56 " in last 604800 seconds",
58 " and still drinking",
60 ri = randint(0, len(hsuff) - 1)
63 def getSomeSmile(self):
64 """Return some Slack smile."""
74 ri = randint(0, len(smiles) - 1)
77 def computeEqualDrinkers(self):
78 """Compute equally drunken drinkers."""
80 if len(self.top) <= 0:
82 lc = self.top[0][1] + 1
83 for (un, c) in self.top:
87 self.topeq[-1].append((un, c))
90 def load_top_drinks(self):
91 """Get list of top drinks, update message."""
92 self.msg = "*Top drinks*\n"
93 self.top = self.db.get_top_drinks()
94 self.computeEqualDrinkers()
96 for drinks in self.topeq[:3]:
98 d = [un for (un, c) in drinks]
99 self.msg += "{}{}".format(i, self.isuff[i])
106 self.msg += "{}".format(self.get_drink_smile(un))
108 self.msg += "" if len(d) == 2 else ","
109 self.msg += " and {}".format(self.get_drink_smile(un))
111 self.msg += ", *{}*".format(self.get_drink_smile(un))
113 self.msg += " with each of "
116 self.msg += "`{}` drunk".format(c)
120 def loadTopMateDrinkers(self):
121 """Get list of top Mate drinkers, update message."""
122 self.msg = "*Top 5 Club-Mate drinkers*\n"
123 self.top = self.db.getTopMateDrinkers()
124 self.computeEqualDrinkers()
125 totl = [c for (un, c) in self.top]
128 for drinkers in self.topeq[:5]:
130 d = [un for (un, c) in drinkers]
131 self.msg += "{}{}".format(i, self.isuff[i])
138 self.msg += "*{}*".format(un)
140 self.msg += "" if len(d) == 2 else ","
141 self.msg += " and *{}*".format(un)
143 self.msg += ", *{}*".format(un)
145 self.msg += " of whom each drank "
147 self.msg += " who drank "
148 self.msg += "`{:.2f}` liters".format(c)
150 self.msg += self.getHeroComment()
154 self.msg += "Oh, and together we drank *{:.2f}* liters {}".format(
160 def load_top_tea_drinkers(self):
161 """Get list of top tea drinkers, update message."""
162 self.msg = "*Top 5 tea drinkers*\n"
163 self.top = self.db.get_top_tea_drinkers()
164 self.computeEqualDrinkers()
165 tot = sum([c for (un, c) in self.top])
167 for drinkers in self.topeq[:5]:
169 d = [un for (un, c) in drinkers]
170 self.msg += "{}{}".format(i, self.isuff[i])
177 self.msg += "*{}*".format(un)
179 self.msg += "" if len(d) == 2 else ","
180 self.msg += " and *{}*".format(un)
182 self.msg += ", *{}*".format(un)
184 self.msg += " of whom each drank "
186 self.msg += " who drank "
188 self.msg += "`{}` teas".format(c)
190 self.msg += "`{}` tea".format(c)
194 self.msg += "Oh, and together we drank *{}* teas {}".format(
199 def loadDrunkSum(self, *args, **kwargs):
200 """Get list of drunken ``flavor`` from ``dtf`` to ``dtt``, update msg.
203 flavor -- The flavor of beverage.
204 dtf -- Date and time *from*.
205 dtt -- Date and time *to*.
207 self.top = self.db.getDrunkSum(*args, **kwargs)
213 flavor = kwargs["flavor"]
216 self.msg = "{} drunk from {} to {}:\n".format(flavor, dtf, dtt)
217 for (un, cnt) in self.top:
218 self.msg += "{}: {}\n".format(un, cnt)
222 def loadDrunkList(self, *args, **kwargs):
223 """Get list of drunken ``flavor`` from ``dtf`` to ``dtt``, update msg.
226 flavor -- The flavor of beverage.
227 dtf -- Date and time *from*.
228 dtt -- Date and time *to*.
230 self.top = self.db.getDrunkSum(*args, **kwargs)
231 d = self.db.getDrunkList(*args, **kwargs)
237 flavor = kwargs["flavor"]
240 self.msg = "{} drunk from {} to {}:\n".format(flavor, dtf, dtt)
241 for (un, l) in d.items():
242 self.msg += "{}: ".format(un)
244 self.msg += "{}, ".format(cnt)
245 self.msg += "{}\n".format(l[-1])