From: Jiri Vlasak <hubacji1@fel.cvut.cz>
Date: Thu, 2 Jan 2020 10:22:05 +0000 (+0100)
Subject: Add get top drinks from db
X-Git-Tag: v0.4.0~1^2~5
X-Git-Url: http://rtime.felk.cvut.cz/gitweb/hubacji1/coffee-getter.git/commitdiff_plain/5900f3ca435a23b7891eaf402a3f962b569d70ad?hp=6f7ef2c0a6bcc96b2777c7b65ed8ff2bca767661

Add get top drinks from db
---

diff --git a/cbdb.py b/cbdb.py
index 22e0058..2c53f41 100644
--- a/cbdb.py
+++ b/cbdb.py
@@ -22,6 +22,24 @@ class Db:
         if self.con:
             self.con.close()
 
+    def get_top_drinks(self):
+        """Return list of pairs of drink name and count."""
+        q = """
+
+        SELECT count(*), flavor
+        FROM coffees
+        WHERE time BETWEEN
+            datetime('now', 'localtime', '-7 days')
+            AND datetime('now', 'localtime')
+        GROUP BY flavor
+
+        """
+        top = []
+        for (cnt, dn) in self.cur.execute(q):
+            top.append((dn, cnt))
+        top.sort(key=lambda x: (x[1], x[0]), reverse=True)
+        return top
+
     def getTopMateDrinkers(self):
         """Return list of pairs of name, count for Mate drinkers."""
         users = {}