mirror of
https://github.com/furyhawk/home_stack.git
synced 2026-07-21 02:06:47 +00:00
furyhawk
3fa5720b5a
- Add database connection pooling (pool_size=20, max_overflow=10, pool_pre_ping) - Add LRU response caching for weather endpoints with configurable TTLs - Reuse httpx.AsyncClient across weather API requests - Add Docker build cache mounts for uv and npm package managers - Optimize query ordering in users and items endpoints - Apply ruff formatting fixes across backend
41 lines
1.3 KiB
Python
41 lines
1.3 KiB
Python
from sqlmodel import Session, create_engine, select
|
|
|
|
from app import crud
|
|
from app.core.config import settings
|
|
from app.models import User, UserCreate
|
|
|
|
engine = create_engine(
|
|
str(settings.SQLALCHEMY_DATABASE_URI),
|
|
pool_size=20,
|
|
max_overflow=10,
|
|
pool_timeout=30,
|
|
pool_recycle=3600,
|
|
pool_pre_ping=True,
|
|
)
|
|
|
|
|
|
# make sure all SQLModel models are imported (app.models) before initializing DB
|
|
# otherwise, SQLModel might fail to initialize relationships properly
|
|
# for more details: https://github.com/fastapi/full-stack-fastapi-template/issues/28
|
|
|
|
|
|
def init_db(session: Session) -> None:
|
|
# Tables should be created with Alembic migrations
|
|
# But if you don't want to use migrations, create
|
|
# the tables un-commenting the next lines
|
|
# from sqlmodel import SQLModel
|
|
|
|
# This works because the models are already imported and registered from app.models
|
|
# SQLModel.metadata.create_all(engine)
|
|
|
|
user = session.exec(
|
|
select(User).where(User.email == settings.FIRST_SUPERUSER)
|
|
).first()
|
|
if not user:
|
|
user_in = UserCreate(
|
|
email=settings.FIRST_SUPERUSER,
|
|
password=settings.FIRST_SUPERUSER_PASSWORD,
|
|
is_superuser=True,
|
|
)
|
|
user = crud.create_user(session=session, user_create=user_in)
|