Files
FabledCurator/backend/app/services/showcase_service.py
T

41 lines
1.3 KiB
Python

"""Random-sample query for the showcase.
Uses the tsm_system_rows TABLESAMPLE method (migration 0004) instead of
ORDER BY random(): sampling cost scales with the sample size, not the table,
so it stays fast as the collection grows. SYSTEM_ROWS(n) returns up to n
rows; an empty table yields none.
"""
from sqlalchemy import select, text
from sqlalchemy.ext.asyncio import AsyncSession
from ..models import ImageRecord
from .gallery_service import thumbnail_url
class ShowcaseService:
def __init__(self, session: AsyncSession):
self.session = session
async def random_sample(self, limit: int = 60) -> list[dict]:
if limit < 1 or limit > 200:
raise ValueError("limit must be between 1 and 200")
stmt = select(ImageRecord).from_statement(
text(
"SELECT * FROM image_record "
"TABLESAMPLE SYSTEM_ROWS(:n)"
).bindparams(n=limit)
)
rows = (await self.session.execute(stmt)).scalars().all()
return [
{
"id": r.id,
"sha256": r.sha256,
"mime": r.mime,
"width": r.width,
"height": r.height,
"thumbnail_url": thumbnail_url(r.sha256, r.mime),
}
for r in rows
]