Implement database initialisation

.gitignore

 *.swp
 config.py
 __pycache__
+*.sqlite

config.py

@@ -9,3 +9,5 @@ DEBUG = True
 
 SQLITE_DB = 'db.sqlite'
 DB_ENGINE = 'sqlite'
+SQLITE_INIT_SCHEMA = True
+SQLITE_INIT_DATA = True

server.py

@@ -2,11 +2,27 @@
 from flask import Flask, render_template, g
 import mysql.connector
 import sqlite3
+import os
 
 app = Flask(__name__)
 config = app.config
+config['DB_SCHEMA'] = 'db_schema.sql'
+config['DB_DATA'] = 'db_example.sql'
+config['SQLITE_INIT_SCHEMA'] = False
+config['SQLITE_INIT_DATA'] = False
 config.from_pyfile('config.py')
 
+if config['DB_ENGINE'] == 'sqlite':
+	created = not os.path.exists(config['SQLITE_DB'])
+	db = sqlite3.connect(config['SQLITE_DB'])
+	cur = db.cursor()
+	if config['SQLITE_INIT_SCHEMA']:
+		cur.executescript(open(config['DB_SCHEMA']).read())
+	if config['SQLITE_INIT_DATA'] and created:
+		cur.executescript(open(config['DB_DATA']).read())
+	db.commit()
+	db.close()
+
 # Row wrapper for sqlite
 def dict_factory(cursor, row):
 	d = {}