import motor.motor_asyncio
from pymongo.collection import Collection

from core.utils.mongo import get_mongo_url, get_mongo_db


def use_client(connection_name: str = None):
    return motor.motor_asyncio.AsyncIOMotorClient(get_mongo_url(connection_name))


def use_collection(collection: str, db: str = None, connection_name: str = None) -> Collection:
    if not db:
        db = get_mongo_db(connection_name)
    return use_client(connection_name)[db][collection]