Initial commit

core/service/mongo_db.py

@@ -0,0 +1,14 @@
+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]