first commit
This commit is contained in:
commit
6e354ba041
109 changed files with 12110 additions and 0 deletions
26
backend/app/mongo.py
Normal file
26
backend/app/mongo.py
Normal file
|
|
@ -0,0 +1,26 @@
|
|||
from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
|
||||
|
||||
from app.config import settings
|
||||
|
||||
mongo_client: AsyncIOMotorClient | None = None
|
||||
mongo_db: AsyncIOMotorDatabase | None = None
|
||||
|
||||
|
||||
async def connect_mongo() -> None:
|
||||
global mongo_client, mongo_db
|
||||
mongo_client = AsyncIOMotorClient(settings.mongodb_url)
|
||||
mongo_db = mongo_client[settings.mongodb_db]
|
||||
|
||||
|
||||
async def close_mongo() -> None:
|
||||
global mongo_client, mongo_db
|
||||
if mongo_client:
|
||||
mongo_client.close()
|
||||
mongo_client = None
|
||||
mongo_db = None
|
||||
|
||||
|
||||
def get_mongo_db() -> AsyncIOMotorDatabase:
|
||||
if mongo_db is None:
|
||||
raise RuntimeError("MongoDB is not connected")
|
||||
return mongo_db
|
||||
Loading…
Add table
Add a link
Reference in a new issue