from sqlalchemy import create_engine

from src.config.settings import settings


def build_connect_args(database_url: str, connect_timeout: int) -> dict:
    if database_url.startswith("postgresql"):
        return {"connect_timeout": connect_timeout}
    return {}


engine = create_engine(
    settings.database_url,
    pool_pre_ping=True,
    connect_args=build_connect_args(settings.database_url, settings.database_connect_timeout),
)