Spaces:
Running
Running
File size: 1,416 Bytes
a8a2cf5 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 |
"""Async SQLModel database configuration for Supabase/PostgreSQL."""
import ssl
from typing import AsyncGenerator
from sqlalchemy.ext.asyncio import AsyncEngine, AsyncSession, create_async_engine
from sqlalchemy.orm import sessionmaker
from sqlmodel import SQLModel
from ..core.config.config import settings
# Supabase requires SSL but we need to handle certificate verification
# For Supabase connection pooler, use a more permissive SSL context
ssl_context = ssl.create_default_context()
ssl_context.check_hostname = False
ssl_context.verify_mode = ssl.CERT_NONE
engine: AsyncEngine = create_async_engine(
settings.DATABASE_URL,
echo=settings.DEBUG,
# Disable prepared statement cache for pgbouncer transaction/statement mode
connect_args={"ssl": ssl_context, "statement_cache_size": 0},
)
AsyncSessionLocal = sessionmaker(
bind=engine,
class_=AsyncSession,
expire_on_commit=False,
)
async def get_session() -> AsyncGenerator[AsyncSession, None]:
"""FastAPI dependency to provide an async database session."""
async with AsyncSessionLocal() as session:
yield session
async def init_db() -> None:
"""Create tables if they do not exist."""
async with engine.begin() as conn:
await conn.run_sync(SQLModel.metadata.create_all)
async def dispose_engine() -> None:
"""Dispose the async engine (called on shutdown)."""
await engine.dispose()
|