From 681349db67bb828ab926b53734b44c50249d5f65 Mon Sep 17 00:00:00 2001 From: stilbruch Date: Mon, 21 Mar 2022 18:09:50 -0500 Subject: Setup basic database connections --- app/database.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) create mode 100644 app/database.py (limited to 'app/database.py') 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 -- cgit v1.2.3