Pymongo connection with a pinch of flavour

pymongo_connect.py

import os
import urllib

from pymongo import MongoClient

def get_conn_string() -> str:
    conn_string = "mongodb://{}:{}@{}/{}".format(
        os.getenv("DB_USER"),
        urllib.parse.quote(os.getenv("DB_PWD")), # escapes invalid chars
        os.getenv("DB_URI"),
        os.getenv("DB_NAME"),
    )

    return conn_string


def get_client() -> MongoClient:
    client = MongoClient(get_conn_string())
    return client

# usage
client = get_client()
db = client.get_database()

# from here on see http://api.mongodb.com/python/current/api/pymongo/database.html#pymongo.database.Database
# Example: push pandas dataframe
# client.get_database().COLLECTION_NAME.insert_many(df.to_dict(orient="records"))