The below code defines a Python class named MongoDB that provides a base wrapper for connecting to a MongoDB database.

mongodb-connection-python.py

# Third-party Imports
import pymongo
import motor


class MongoDB(object):
    """Base wrapper to connect to MongoDB"""

    def __init__(self):
        """Initialize the MongoDB connection"""

        self.url = "mongodb+srv://<username>:<password>@<url>/<db>?retryWrites=true&w=majority"
        self.client: pymongo.MongoClient = None

    def start_client(self):
        """This method creates an instance of the `AsyncIOMotorClient`"""

        self.client = motor.motor_asyncio.AsyncIOMotorClient(self.url)

    def get_client(self) -> pymongo.MongoClient:
        """Get the Mongo client"""

        if self.client is None:
            self.start_client()
        return self.client

    def stop_client(self):
        """Gracefully close the Mongo client"""

        self.client.close()
        self.client = None

    def get_db(self):
        """Get the <db_name> database"""
        
        db = self.get_client().get_database("<db_name>")

        if db is None:
            raise Exception({"message": "Database not found"})
        return db



# initialize the MongoDB connection
mongo = MongoDB()