jaothan commited on
Commit
16a83c3
·
verified ·
1 Parent(s): 0994965

Update utils/database.py

Browse files
Files changed (1) hide show
  1. utils/database.py +9 -23
utils/database.py CHANGED
@@ -1,23 +1,9 @@
1
- from pymongo import MongoClient
2
- from pymongo.errors import ConnectionFailure
3
- import os
4
-
5
- client = None
6
- db = None
7
-
8
- async def connect_to_mongo():
9
- global client, db
10
- mongo_uri = os.getenv("MONGO_URI")
11
- db_name = os.getenv("MONGO_DB_NAME")
12
- client = MongoClient(mongo_uri)
13
- db = client[db_name]
14
- try:
15
- client.admin.command('ping')
16
- print("Pinged your deployment. You successfully connected to MongoDB!")
17
- except ConnectionFailure as e:
18
- print("Could not connect to MongoDB:", e)
19
-
20
- async def close_mongo_connection():
21
- global client
22
- if client:
23
- client.close()
 
1
+ from pymongo import MongoClient
2
+ import os
3
+
4
+ def get_db():
5
+ MONGO_URI = os.getenv("MONGO_URI")
6
+ MONGO_DB_NAME = os.getenv("MONGO_DB_NAME")
7
+ client = MongoClient(MONGO_URI)
8
+ db = client[MONGO_DB_NAME]
9
+ return db