first commit
This commit is contained in:
26
backend/database.py
Normal file
26
backend/database.py
Normal file
@@ -0,0 +1,26 @@
|
||||
from motor.motor_asyncio import AsyncIOMotorClient, AsyncIOMotorDatabase
|
||||
from config import settings
|
||||
|
||||
client: AsyncIOMotorClient = None
|
||||
database: AsyncIOMotorDatabase = None
|
||||
|
||||
|
||||
async def connect_to_mongo():
|
||||
"""Connect to MongoDB"""
|
||||
global client, database
|
||||
client = AsyncIOMotorClient(settings.mongodb_url)
|
||||
database = client[settings.database_name]
|
||||
print(f"Connected to MongoDB at {settings.mongodb_url}")
|
||||
|
||||
|
||||
async def close_mongo_connection():
|
||||
"""Close MongoDB connection"""
|
||||
global client
|
||||
if client:
|
||||
client.close()
|
||||
print("Closed MongoDB connection")
|
||||
|
||||
|
||||
def get_database() -> AsyncIOMotorDatabase:
|
||||
"""Get database instance"""
|
||||
return database
|
||||
Reference in New Issue
Block a user