SyncM8/syncm8

"""Client to connect mongoengine to the db."""
import os

from flask.app import Flask
from flask_mongoengine import MongoEngine


def connect_db(app: Flask) -> None:
    """Connect mongoengine to db. Only needs to be called once per thread."""
    mongo_user = os.environ.get("MONGO_USER")
    mongo_password = os.environ.get("MONGO_PASSWORD")
    mongo_host = os.environ.get("MONGO_HOST")
    app.config["MONGODB_SETTINGS"] = {
        "db": "main",
        "username": mongo_user,
        "password": mongo_password,
        "host": mongo_host,
        "w": "majority",
        "connect": False,  # necessary to make mongoengine play nicely w/ flask threads
    }
    MongoEngine(app)