-
Notifications
You must be signed in to change notification settings - Fork 4
/
database.py
65 lines (55 loc) · 2.09 KB
/
database.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
import sentry_sdk
import sqlite3
import sys
import keybase
def initial_tables(c):
c.execute('''create table if not exists reminders (
reminder_time int,
created_time int not null,
body text not null,
user text not null,
conv_id text not null)''')
c.execute('create index if not exists idx_reminder_time on reminders(reminder_time)')
c.execute('create index if not exists idx_reminder_conv on reminders(conv_id, reminder_time)')
c.execute('''create table if not exists users (
username text not null unique,
settings text not null)''')
c.execute('create index if not exists idx_user_name on users(username)')
c.execute('''create table if not exists conversations (
id text not null,
channel text not null,
is_team boolean not null,
topic text,
last_active_time int not null,
context int not null,
reminder_rowid int,
debug boolean not null)''')
c.execute('create unique index if not exists idx_conversations_id on conversations(id)')
def add_reminder_deleted(c):
c.execute('alter table reminders add deleted boolean not null default 0')
def add_reminder_repeating(c):
c.execute('alter table reminders add repetition_interval text')
c.execute('alter table reminders add repetition_nth int')
def add_reminder_errors(c):
c.execute('alter table reminders add errors int not null default 0')
def setup(db):
try:
c = sqlite3.connect(db)
except sqlite3.OperationalError as e:
sentry_sdk.capture_exception()
print >> sys.stderr, "FATAL: Error connecting to " + db + ": " + e.message
sys.exit(1)
cur = c.cursor()
cur.execute('pragma user_version')
db_version = cur.fetchone()[0]
migrations = [
initial_tables,
add_reminder_deleted,
add_reminder_repeating,
add_reminder_errors
]
for i, migration in enumerate(migrations, start=1):
if db_version < i:
migration(cur)
cur.execute('pragma user_version = ' + str(len(migrations)))
c.commit()