27 lines
694 B
Python
27 lines
694 B
Python
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
|