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
    )