From 5990c005edb42a5e064224a03fed32522a66047e Mon Sep 17 00:00:00 2001 From: leonoverweel Date: Mon, 27 Feb 2017 00:08:58 +0100 Subject: Add Python database rebuilder This one allows you to specify the location of the database. --- database/rebuild-database.py | 31 +++++++++++++++++++++++++++++++ 1 file changed, 31 insertions(+) create mode 100644 database/rebuild-database.py (limited to 'database/rebuild-database.py') diff --git a/database/rebuild-database.py b/database/rebuild-database.py new file mode 100644 index 00000000..6e10b646 --- /dev/null +++ b/database/rebuild-database.py @@ -0,0 +1,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." -- cgit v1.2.3