from beanie import init_beanie
from core.configs import get_database_settings
from motor.motor_asyncio import AsyncIOMotorClient

from db.models import Task, User, Video

MODELS = [User, Video, Task]


async def init_db() -> None:
    config = get_database_settings()
    client = AsyncIOMotorClient(config.URL.unicode_string())

    await init_beanie(database=client[config.NAME], document_models=MODELS)