BIG BIG BIG BIG BIG UPDATES

Co-authored-by: Copilot <[email protected]>

backend/app/core/mongodb.py

@@ -0,0 +1,37 @@
+from core.config import settings
+from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
+
+
+class _MongoDB:
+    client: AsyncIOMotorClient | None = None
+
+    def get_database(self) -> AsyncIOMotorDatabase:
+        if self.client is None:
+            raise RuntimeError("MongoDB client has not been initialized")
+        return self.client[settings.MONGODB_DB_NAME]
+
+
+mongodb = _MongoDB()
+
+
+async def init_mongodb() -> None:
+    mongodb.client = AsyncIOMotorClient(settings.MONGODB_URI)
+    await ensure_mongodb_available()
+
+
+async def close_mongodb() -> None:
+    client = mongodb.client
+    mongodb.client = None
+    if client is not None:
+        client.close()
+
+
+async def get_database() -> AsyncIOMotorDatabase:
+    return mongodb.get_database()
+
+
+async def ensure_mongodb_available() -> None:
+    client = mongodb.client
+    if client is None:
+        raise RuntimeError("MongoDB client has not been initialized")
+    await client.admin.command("ping")