2026-06-04 20:30:04 -04:00
|
|
|
from sqlalchemy import MetaData, event, text
|
|
|
|
|
from sqlalchemy.ext.asyncio import AsyncSession, async_sessionmaker, create_async_engine
|
|
|
|
|
|
2026-07-07 17:50:51 -04:00
|
|
|
from config import DATABASE_URL
|
2026-06-04 20:30:04 -04:00
|
|
|
|
|
|
|
|
engine = create_async_engine(
|
|
|
|
|
DATABASE_URL, echo=False, pool_size=5, max_overflow=10, pool_recycle=300
|
|
|
|
|
)
|
|
|
|
|
async_session = async_sessionmaker(
|
|
|
|
|
engine, class_=AsyncSession, expire_on_commit=False
|
|
|
|
|
)
|
|
|
|
|
metadata = MetaData()
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def init_extensions():
|
|
|
|
|
"""Initialize PostGIS and TimescaleDB extensions on first connection."""
|
|
|
|
|
async with engine.connect() as conn:
|
|
|
|
|
await conn.execute(text("CREATE EXTENSION IF NOT EXISTS postgis"))
|
|
|
|
|
await conn.execute(text("CREATE EXTENSION IF NOT EXISTS timescaledb"))
|
|
|
|
|
await conn.commit()
|
