vladyslav commited on
Commit
d28c029
·
1 Parent(s): f86ba47

Added logs for db

Browse files
Files changed (1) hide show
  1. 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
- collection_name = os.getenv("DEV_MONGO_COLLECTION") if os.getenv("ENV_TYPE") == "dev" else os.getenv(
13
- "PROD_MONGO_COLLECTION")
 
 
 
 
 
 
 
 
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")