yuqei
17 lines
440 B
Python
17 lines
440 B
Python
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),
|
|
)
|