Add drunk list getter

diff --git a/cbgetter.py b/cbgetter.py
index fce31bfc2bb01cbd4bd6ba17cdeab6230bcac9de..b837f8910c817c487fa96180628b24186075fe6f 100644 (file)
--- a/cbgetter.py
+++ b/cbgetter.py
@@ -131,3 +131,30 @@ class CoffeebotGetter:
             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