Move DB stuff inside a class for further refactoring.

1 files changed, 5 insertions, 2 deletions

diff --git a/app.py b/app.py
index 3d8d5c7..72becc3 100644
--- a/app.py
+++ b/app.py
@@ -1,7 +1,7 @@
 from secrets import token_hex
 from flask import Flask
 from os import getenv
-from db_actions import db
+#from db_actions import db
 
 app = Flask(__name__, static_url_path='')
 
@@ -15,7 +15,10 @@ if s_key := getenv("SECRET_KEY"):
 else:
     app.secret_key = token_hex()
     
-db.init_app(app)
+from db.db import DB
+D = DB()
+    
+#db.init_app(app)
 
 import routes.base
 import routes.answer