sakhteman/auth/utils/mongo.py

from pymongo.collection import Collection

from config import conf
from core.service.mongo_db import use_collection


def get_auth_db():
    return conf.server.is_prod and conf.database.get('mongo_auth').production.db or conf.database.get(
        'mongo_auth').dev.db


def use_auth_collection(collection: str) -> Collection:
    return use_collection(collection, get_auth_db(), connection_name='mongo_auth')