Setup basic database connections

1 files changed, 19 insertions, 0 deletions

diff --git a/app/database.py b/app/database.py
new file mode 100644
index 0000000..e23aa5d
--- /dev/null
+++ b/app/database.py
@@ -0,0 +1,19 @@
+import sqlite3
+from app import app
+from flask import g
+
+# constants
+DATABASE_FILE = "../strengthy.db"
+
+# Called when an "appcontext" is closed, usually a request is finished
+@app.teardown_appcontext
+def close_db_conn(exception):
+    db = getattr(g, '_database', None)
+    if db is not None:
+        db.close()
+
+def database_get():
+    db = getattr(g, '_database', None)
+    if db is None:
+        db = g._database = sqlite3.connect(DATABASE_FILE)
+    return db