Source code for weaver.database

import logging
from typing import TYPE_CHECKING

from pyramid.settings import asbool

from weaver.utils import get_registry, get_settings

LOGGER = logging.getLogger(__name__)
if TYPE_CHECKING:
    from weaver.database.mongodb import MongoDatabase   # noqa: F401
    from weaver.typedefs import AnyDatabaseContainer    # noqa: F401


def get_db(container):
    # type: (AnyDatabaseContainer) -> MongoDatabase
    registry = get_registry(container)
    return registry.db


def includeme(config):
    settings = get_settings(config)
    if asbool(settings.get("weaver.build_docs", False)):
        LOGGER.info("Skipping database when building docs...")
        return

    LOGGER.info("Adding database...")
    from weaver.database.mongodb import MongoDatabase
    config.registry.db = MongoDatabase(config.registry)

    def _add_db(request):
        db = request.registry.db
        # if db_url.username and db_url.password:
        #     db.authenticate(db_url.username, db_url.password)
        return db
    config.add_request_method(_add_db, "db", reify=True)