summaryrefslogtreecommitdiff
path: root/database/rebuild-database.py
blob: 6e10b6460c6c036fe538a1c4a951c28b515461a4 (plain)
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
import os
import sqlite3
import sys

sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)

try:
    BASE_DIR = directory_name=sys.argv[1]
except:
    BASE_DIR = os.path.dirname(os.path.abspath(__file__))
db_location = os.path.join(BASE_DIR, 'opendc.db')

print "Removing old database..."
os.remove(db_location)

print "Connecting to new database..."
conn = sqlite3.connect(db_location)
c = conn.cursor()

print "Importing schema..."
with open('schema.sql') as schema:
    c.executescript(schema.read())

print "Importing test data..."
with open('test.sql') as test:
    c.executescript(test.read())

conn.commit()
conn.close()

print "Done."