freeleaps-service-hub/apps/central_storage/webapi/providers/database.py
2024-10-30 07:22:26 -07:00

21 lines
591 B
Python

from webapi.config.site_settings import site_settings
from beanie import init_beanie
from motor.motor_asyncio import AsyncIOMotorClient
from backend.models import backend_models
def register(app):
app.debug = "mongo_debug"
app.title = "mongo_name"
@app.on_event("startup")
async def start_database():
await initiate_database()
async def initiate_database():
client = AsyncIOMotorClient(site_settings.MONGODB_URI, serverSelectionTimeoutMS=60000)
await init_beanie(
database=client[site_settings.MONGODB_NAME], document_models=backend_models
)