commit 8e8a288ca5ca1592c54fd3c42263a13eb9c095d4 parent 8307bd21b7a54a2b77ef3167589efd2ff1a75213 Author: Nikolas Mazur <nikolas@pop-os.localdomain> Date: Fri, 23 Dec 2022 11:42:38 -0700 Add basic database setup Diffstat:
| A | db.py | | | 17 | +++++++++++++++++ |
1 file changed, 17 insertions(+), 0 deletions(-)
diff --git a/db.py b/db.py @@ -0,0 +1,17 @@ +import bson + +from flask import current_app, g +from werkzeug.local import LocalProxy +from flask_pymongo import PyMongo + +from pymongo.errors import DuplicateKeyError, OperationFailure +from bson.objectid import ObjectId +from bson.errors import InvalidId + +def get_db(): + db = getattr(g, "_database", None) + if db is None: + db = g._database = PyMongo(current_app).db + return db +db = LocalProxy(get_db) +