mongo_client.py

from pymongo import MongoClient, server_api

from os import getenv
 
database_name=getenv("DATABASE_NAME")

def get_client():
    client = MongoClient(getenv("MONGODB_CONNECTION_STRING"), server_api=server_api.ServerApi(
        version="1", strict=True, deprecation_errors=True))
    
    return client

def get_collection(collection_name:str,client:MongoClient):
    database=client[database_name]
    return database[collection_name]