from beanie import init_beanie
from motor.motor_asyncio import AsyncIOMotorClient
from pydantic import BaseSettings, Field

from models.request import Request


class Settings(BaseSettings):
    DATABASE_URL: str = Field(..., env="DATABASE_URL")
    AZURE_KEY: str = Field(..., env="AZURE_KEY")
    AZURE_ENDPOINT: str = Field(..., env="AZURE_ENDPOINT")
    AZURE_LOCATION: str = Field(..., env="AZURE_LOCATION")
    ENV: str = Field(..., env="ENV")

    class Config:
        env_file = ".env"
        orm_mode = True


async def initiate_database():
    client = AsyncIOMotorClient(Settings().DATABASE_URL)
    await init_beanie(database=client.get_default_database(), document_models=[Request])