coffee_db.py

   1 import sqlite3
   2 import os
   3
   4 dbdir = os.path.dirname(__file__)
   5 dbdef = os.path.join(dbdir, "coffee_db.sql")
   6 dbfile = os.path.join(dbdir, "coffee.db")
   7
   8 def open_db():
   9     conn = sqlite3.connect(dbfile)
  10     c = conn.cursor()
  11     return conn, c
  12
  13 def close_db(conn):
  14     conn.commit()
  15     conn.close()
  16
  17 def init_db():
  18     conn, c = open_db()
  19     with open(dbdef, "r") as f:
  20         c.executescript(f.read())
  21     close_db(conn)
  22
  23 def add_user(uid):
  24     conn, c = open_db()
  25     c.execute("insert or ignore into users (id) values (?)", (uid,))
  26     close_db(conn)
  27
  28 def get_name(uid):
  29     conn, c = open_db()
  30     for name, in c.execute("select name from users where id = ?",(uid,)):
  31         close_db(conn)
  32         return name
  33     close_db(conn)
  34     return None
  35
  36
  37 def name_user(uid, name):
  38     conn, c = open_db()
  39     c.execute("update users set name = ? where id = ?", (name, uid))
  40     close_db(conn)
  41
  42 def list_users():
  43     conn, c = open_db()
  44     for row in c.execute("select * from users"):
  45         print(row)
  46     close_db(conn)
  47
  48
  49 def add_coffee(uid, flavor, time=None):
  50     conn, c = open_db()
  51     if time is None:
  52         c.execute("insert into coffees (id, flavor) values (?,?)", (uid,flavor))
  53     else:
  54         c.execute("insert into coffees (id, flavor, time) values (?,?,?)", (uid, flavor, time))
  55     close_db(conn)
  56
  57
  58 def list_coffees(uid=None):
  59     c = conn.cursor()
  60     if uid is None:
  61         for row in c.execute("select id, date(time, 'localtime'), flavor from coffees"):
  62             print(row)
  63     else:
  64         for row in c.execute("select date(time, 'localtime'), flavor from coffees where id = ?", (uid,)):
  65             print(row)
  66
  67 def flavors():
  68     conn, c = open_db()
  69     res = [row for row, in c.execute("select distinct name from flavors")]
  70     close_db(conn)
  71     return res
  72
  73 def coffee_flavors(uid=None):
  74     conn, c = open_db()
  75
  76     if uid is None:
  77         res = list(c.execute("""
  78             select f.name, count(c.flavor) from flavors f left join
  79             (select * from coffees) c
  80             on f.name=c.flavor group by f.name
  81             """))
  82     else:
  83         res = list(c.execute("""
  84             select f.name, count(c.flavor) from flavors f left join
  85             (select * from coffees where id = ?) c
  86             on f.name=c.flavor group by f.name
  87             """, (uid,)))
  88
  89     close_db(conn)
  90     return res
  91
  92
  93 def coffee_history(uid=None):
  94     conn, c = open_db()
  95
  96     if uid is None:
  97         res = list(c.execute("""
  98             select strftime('%s', ds.d),count(c.flavor),c.flavor from
  99             (select num,date('now', 'localtime', -num || ' days') as d from days) ds
 100             left join coffees c
 101             on d = date(c.time) group by d, c.flavor
 102             """))
 103     else:
 104         res = list(c.execute(
 105             """
 106             select strftime('%s', ds.d),count(c.flavor),c.flavor from
 107             (select num,date('now', 'localtime', -num || ' days') as d from days) ds
 108             left join
 109             (select date(time, 'localtime') as time,flavor from coffees where id = ?) c
 110             on d = date(c.time) group by d, c.flavor
 111             """
 112             , (uid,)))
 113
 114     close_db(conn)
 115     return res
 116
 117 def coffee_count(uid=None, start=None, stop=None):
 118     conn, c = open_db()
 119
 120     args = []
 121     clauses = []
 122
 123     if uid is not None:
 124         clauses.append("id = ?")
 125         args.append(uid)
 126
 127     if start is not None:
 128         clauses.append("date(time, 'localtime') >= date('now', 'localtime', '-%d days')" % int(start))
 129
 130     if stop is not None:
 131         clauses.append("date(time, 'localtime') <= date('now', 'localtime', '-%d days')" % int(stop))
 132
 133     for count, in c.execute(
 134             "select count(*) from coffees where " +
 135             " and ".join(clauses)
 136             , args):
 137         res = count
 138
 139     if not res:
 140         res = "0"
 141
 142     return res