scripts/utils.py

from pymongo.mongo_client import MongoClient
from pymongo.server_api import ServerApi


def init_mongo_db(uri: str, db_name: str):
    """Initialize the mongodb database."""

    try:
        assert uri is not None, "No URI passed"
        client = MongoClient(uri, server_api=ServerApi("1"))
        database = client[db_name]
        print("Connected to MongoDB")
        return database
    except Exception as e:
        print("Something went wrong connecting to mongodb")
        return