vladyslav
commited on
Commit
·
d28c029
1
Parent(s):
f86ba47
Added logs for db
Browse files- utils/db.py +14 -2
utils/db.py
CHANGED
@@ -6,15 +6,26 @@ from pymongo import MongoClient
|
|
6 |
load_dotenv()
|
7 |
|
8 |
client = MongoClient(os.getenv("MONGO_CONNECTION_STRING"))
|
|
|
9 |
|
10 |
db = client[os.getenv("MONGO_DB_NAME")]
|
11 |
|
12 |
-
|
13 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
14 |
collection = db[collection_name]
|
|
|
15 |
|
16 |
|
17 |
def save_results(student_name, class_name, model, book, questions, feedback, rating):
|
|
|
18 |
collection.insert_one({
|
19 |
"student_name": student_name,
|
20 |
"class": class_name,
|
@@ -24,3 +35,4 @@ def save_results(student_name, class_name, model, book, questions, feedback, rat
|
|
24 |
"feedback": feedback,
|
25 |
"rating": rating
|
26 |
})
|
|
|
|
6 |
load_dotenv()
|
7 |
|
8 |
client = MongoClient(os.getenv("MONGO_CONNECTION_STRING"))
|
9 |
+
print("Connected to MongoDB")
|
10 |
|
11 |
db = client[os.getenv("MONGO_DB_NAME")]
|
12 |
|
13 |
+
env_type = os.getenv("ENV_TYPE")
|
14 |
+
print("env:", env_type)
|
15 |
+
|
16 |
+
if env_type == "dev":
|
17 |
+
collection_name = os.getenv("DEV_MONGO_COLLECTION")
|
18 |
+
print("Using dev collection")
|
19 |
+
else:
|
20 |
+
collection_name = os.getenv("PROD_MONGO_COLLECTION")
|
21 |
+
print("Using prod collection")
|
22 |
+
|
23 |
collection = db[collection_name]
|
24 |
+
print("Using collection:", collection_name)
|
25 |
|
26 |
|
27 |
def save_results(student_name, class_name, model, book, questions, feedback, rating):
|
28 |
+
print("Saving results")
|
29 |
collection.insert_one({
|
30 |
"student_name": student_name,
|
31 |
"class": class_name,
|
|
|
35 |
"feedback": feedback,
|
36 |
"rating": rating
|
37 |
})
|
38 |
+
print("Saved results")
|