flavor = kwargs["flavor"]
dtf = kwargs["dtf"]
dtt = kwargs["dtt"]
+ flavors = flavor.split(";")
que = """
SELECT count(*), users.name FROM coffees
INNER JOIN users ON coffees.id = users.id
- WHERE flavor = '{}'
+ """
+ for f in flavors:
+ if f is flavors[0]:
+ que += "WHERE flavor = '{}'".format(f)
+ else:
+ que += "OR flavor = '{}'".format(f)
+ que += """
AND coffees.time BETWEEN
datetime('{}', 'localtime') AND
datetime('{}', 'localtime')
GROUP BY coffees.id
- """.format(flavor, dtf, dtt)
+ """.format(dtf, dtt)
drunk = []
for (cnt, un) in self.cur.execute(que):
drunk.append((un, cnt))