import os | |
from dotenv import load_dotenv | |
from pymongo import MongoClient | |
load_dotenv() | |
client = MongoClient(os.getenv("MONGO_CONNECTION_STRING")) | |
print("Connected to MongoDB") | |
db = client[os.getenv("MONGO_DB_NAME")] | |
env_type = os.getenv("ENV_TYPE") | |
print("env:", env_type) | |
if env_type == "dev": | |
collection_name = os.getenv("DEV_MONGO_COLLECTION") | |
print("Using dev collection") | |
else: | |
collection_name = os.getenv("PROD_MONGO_COLLECTION") | |
print("Using prod collection") | |
collection = db[collection_name] | |
print("Using collection:", collection_name) | |
def save_results(student_name, | |
class_name, | |
model, | |
book, | |
questions, | |
feedback, | |
rating, | |
question_correct, | |
text_coverage, | |
interesting_question): | |
print("Saving results") | |
collection.insert_one({ | |
"student_name": student_name, | |
"class": class_name, | |
"model": model, | |
"book": book, | |
"questions": questions, | |
"question_correct": question_correct, | |
"text_coverage": text_coverage, | |
"interesting_question": interesting_question, | |
"feedback": feedback, | |
"rating": rating, | |
"created_at": db.command("serverStatus")["localTime"] | |
}) | |
print("Saved results") | |