15 lines
467 B
Python
15 lines
467 B
Python
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]
|