database.py

from pymongo import MongoClient

uri = "mongodb+srv://cluster0.tzvxqk3.mongodb.net/?authSource=%24external&authMechanism=MONGODB-X509&retryWrites=true&w=majority"
client = MongoClient(uri,
                     tls=True,
                     tlsCertificateKeyFile="X509-cert-2433626168795922546.pem")

db = client['testDB']
collection = db['testCol']


################################################

# from pymongo.mongo_client import MongoClient
# from pymongo.server_api import ServerApi
# from urllib.parse import quote_plus
# # import ssl
# from pymongo import MongoClient

# uri = "mongodb+srv://cluster0.tzvxqk3.mongodb.net/?authSource=%24external&authMechanism=MONGODB-X509&retryWrites=true&w=majority"
# client = MongoClient(uri,
#                      tls=True,
#                      tlsCertificateKeyFile='/Users/deemaalnuhait/research/hf/cidar_human_eval/X509-cert-2433626168795922546.pem')

# password = quote_plus('cidar@2024$')
# uri = f"mongodb+srv://deemaa2:{password}@cluster0.qjruxcb.mongodb.net/?retryWrites=true&w=majority"

# # Create a new client and connect to the server
# client = MongoClient(uri, server_api=ServerApi('1'))

# # Specify the database and collection
# db = client.your_database_name
# collection = db.your_collection_name

def save_response(response_data):
    """Saves a response to the MongoDB collection."""

    try:
        # Insert the response data into the collection
        print("right before insertion")
        collection.insert_one(response_data)
        print("right after insertion")
    except Exception as e:
        print(f"An error occurred: {e}")
        exit(0)

def read_responses(filter_query=None):
    """Reads responses from the MongoDB collection based on an optional filter."""
    try:
        if filter_query is None:
            filter_query = {}  # An empty query will return all documents
        responses = collection.find(filter_query)
        return list(responses)  # Convert cursor to list
    except Exception as e:
        print(f"An error occurred: {e}")
        return []