utils/database.py

from pymongo import MongoClient
import os

def get_db():
    MONGO_URI = os.getenv("MONGO_URI")
    MONGO_DB_NAME = os.getenv("MONGO_DB_NAME")
    client = MongoClient(MONGO_URI)
    db = client[MONGO_DB_NAME]
    return db