Modify db decorator

gp-sb · Dec 8, 2023 · ddef8e5 · ddef8e5
1 parent e5d4e2b
commit ddef8e5
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/Backend/db_access.py b/Backend/db_access.py
@@ -21,13 +21,17 @@
 load_dotenv()
 
 app = Flask(__name__)
+db_lock = 0
 
 
+@app.before_request
 def db_connect(DB_URL=None):
-    # MongoDB connection
-    mongo_uri = DB_URL or os.environ.get("MONGO_URI")
-    connect(alias="default", host=mongo_uri)
-    return True
+    if not db_lock:
+        # MongoDB connection
+        mongo_uri = DB_URL or os.environ.get("MONGO_URI")
+        connect(alias="default", host=mongo_uri)
+        db_lock = 1
+        return True
 
 
 class Response: